Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 726a0206 authored by Ethan Yonker's avatar Ethan Yonker
Browse files

MTP add/remove storage instead of disabling MTP 

Implement a pipe between TWRP and MTP to allow TWRP to tell MTP
to remove storage partitions as they become unavailable (e.g.
during a wipe, unmount, etc) instead of disabling MTP completely.
This includes some fixes and improvements in destructors to
properly remove / delete various items. This also means that we
will not be toggling adb off and on quite as often.

I do not like that we had to add another thread, but we were
unable to use select() on the mtp_usb character device because
this device does not support polling. Select always returned
indicating that the mtp file descriptor was ready to be read and
the resulting read would block. The read block prevented us from
being able to include reading of the pipe between TWRP and MTP in
the main MTP thread.

We might want to add a return pipe letting TWRP know if the
removal of the storage device was successful, but I am not sure
how we want to implement this. It would invovle timeouts in both
TWRP and MTP to ensure that we returned a failure indicator in a
timely manner to TWRP and prevent deleting the storage device in
the case of a failure. Right now we make no attempt to ensure that
an MTP operation is underway like a large file transfer, but we
were not doing anything like this in the past. In some respects we
have limited control over what happens. If the user installs a
zip that unmounts a storage partition, we will not know about the
change in storage status anyway. Regular Android does not have
these troubles because partitions rarely get unmounted like in
recovery. At some point, we have to hold the user accountable for
performing actions that may remove a storage partition while they
are using MTP anyway.

Ideally we do not want to toggle the USB IDs and thus toggle adb
off and on during early boot, but I am not sure what the best way
to handle that at this time.

Change-Id: I9343e5396bf6023d3b994de1bf01ed91d129bc14
parent 6cb35aab

gui/action.cpp

+1 −0
Original line number Diff line number Diff line
@@ -462,6 +462,7 @@ int GUIAction::doAction(Action action, int isThreaded /* = 0 */) 
				gui_print("Simulating actions...\n");

		} else if (!simulate) {

			PartitionManager.Mount_By_Path(arg, true);

			PartitionManager.Add_MTP_Storage(arg);

		} else

			gui_print("Simulating actions...\n");

		return 0;

gui/partitionlist.cpp

+1 −0
Original line number Diff line number Diff line
@@ -749,6 +749,7 @@ int GUIPartitionList::NotifyTouch(TOUCH_STATE state, int x, int y) 
				if (!mList.at(actualSelection).selected) {

					if (PartitionManager.Mount_By_Path(mList.at(actualSelection).Mount_Point, true)) {

						mList.at(actualSelection).selected = 1;

						PartitionManager.Add_MTP_Storage(mList.at(actualSelection).Mount_Point);

						mUpdate = 1;

					}

				} else {

mtp/MtpDatabase.h

+1 −0
Original line number Diff line number Diff line
@@ -61,6 +61,7 @@ public: 
    virtual MtpDevicePropertyList*  getSupportedDeviceProperties() = 0;



	virtual void 					createDB(MtpStorage* storage, MtpStorageID storageID) = 0;

	virtual void 					destroyDB(MtpStorageID storageID) = 0;



    virtual MtpResponseCode         getObjectPropertyValue(MtpObjectHandle handle,

                                            MtpObjectProperty property,

mtp/MtpMessage.hpp

0 → 100644
+33 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2010 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *	  http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 * Copyright (C) 2014 TeamWin - bigbiff and Dees_Troy mtp database conversion to C++

 */



#ifndef _MTPMESSAGE_HPP

#define _MTPMESSAGE_HPP



#define MTP_MESSAGE_ADD_STORAGE    1

#define MTP_MESSAGE_REMOVE_STORAGE 2



struct mtpmsg {

	int message_type; // 1 is add, 2 is remove, see above

	unsigned int storage_id;

	const char* display;

	const char* path;

	uint64_t maxFileSize;

};



#endif //_MTPMESSAGE_HPP

mtp/MtpServer.cpp

+27 −2
Original line number Diff line number Diff line
@@ -116,9 +116,13 @@ MtpServer::~MtpServer() { 
}



void MtpServer::addStorage(MtpStorage* storage) {

	android::Mutex::Autolock autoLock(mMutex);

	MTPD("addStorage(): storage: %x\n", storage);

	if (getStorage(storage->getStorageID()) != NULL) {

		MTPE("MtpServer::addStorage Storage for storage ID %i already exists.\n", storage->getStorageID());

		return;

	}

	mDatabase->createDB(storage, storage->getStorageID());

	android::Mutex::Autolock autoLock(mMutex);

	mStorages.push(storage);

	sendStoreAdded(storage->getStorageID());

}
@@ -128,11 +132,31 @@ void MtpServer::removeStorage(MtpStorage* storage) { 


	for (size_t i = 0; i < mStorages.size(); i++) {

		if (mStorages[i] == storage) {

			MTPD("MtpServer::removeStorage calling sendStoreRemoved\n");

			// First lock the mutex so that the inotify thread and main

			// thread do not do anything while we remove the storage

			// item, and to make sure we don't remove the item while an

			// operation is in progress

			mDatabase->lockMutex();

			// Grab the storage ID before we delete the item from the

			// database

			MtpStorageID storageID = storage->getStorageID();

			// Remove the item from the mStorages from the vector. At

			// this point the main thread will no longer be able to find

			// this storage item anymore.

			mStorages.removeAt(i);

			sendStoreRemoved(storage->getStorageID());

			// Destroy the storage item, free up all the memory, kill

			// the inotify thread.

			mDatabase->destroyDB(storageID);

			// Tell the host OS that the storage item is gone.

			sendStoreRemoved(storageID);

			// Unlock any remaining mutexes on other storage devices.

			// If no storage devices exist anymore this will do nothing.

			mDatabase->unlockMutex();

			break;

		}

	}

	MTPD("MtpServer::removeStorage DONE\n");

}



MtpStorage* MtpServer::getStorage(MtpStorageID id) {
@@ -275,6 +299,7 @@ void MtpServer::sendStoreAdded(MtpStorageID id) { 
void MtpServer::sendStoreRemoved(MtpStorageID id) {

	MTPD("sendStoreRemoved %08X\n", id);

	sendEvent(MTP_EVENT_STORE_REMOVED, id);

	MTPD("MtpServer::sendStoreRemoved done\n");

}



void MtpServer::sendEvent(MtpEventCode code, uint32_t param1) {