Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2010 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #define LOG_TAG "MtpServerJNI" |
| 18 | #include "utils/Log.h" |
| 19 | |
| 20 | #include <stdio.h> |
| 21 | #include <assert.h> |
| 22 | #include <limits.h> |
| 23 | #include <unistd.h> |
| 24 | #include <fcntl.h> |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 25 | #include <sys/ioctl.h> |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 26 | #include <utils/threads.h> |
| 27 | |
Mike Lockwood | 602bebd | 2010-09-02 09:58:48 -0400 | [diff] [blame] | 28 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 29 | #include <linux/usb/f_mtp.h> |
Mike Lockwood | 602bebd | 2010-09-02 09:58:48 -0400 | [diff] [blame] | 30 | #endif |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 31 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 32 | #include "jni.h" |
| 33 | #include "JNIHelp.h" |
| 34 | #include "android_runtime/AndroidRuntime.h" |
Mike Lockwood | dad6927 | 2010-07-02 15:15:07 -0400 | [diff] [blame] | 35 | #include "private/android_filesystem_config.h" |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 36 | |
| 37 | #include "MtpServer.h" |
| 38 | |
Mike Lockwood | 81ea83d | 2010-06-30 17:49:41 -0400 | [diff] [blame] | 39 | using namespace android; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 40 | |
| 41 | // ---------------------------------------------------------------------------- |
| 42 | |
| 43 | static jfieldID field_context; |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 44 | static Mutex sMutex; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 45 | |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 46 | // in android_media_MtpDatabase.cpp |
| 47 | extern MtpDatabase* getMtpDatabase(JNIEnv *env, jobject database); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 48 | |
| 49 | // ---------------------------------------------------------------------------- |
| 50 | |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 51 | #ifdef HAVE_ANDROID_OS |
| 52 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 53 | static bool ExceptionCheck(void* env) |
| 54 | { |
| 55 | return ((JNIEnv *)env)->ExceptionCheck(); |
| 56 | } |
| 57 | |
| 58 | class MtpThread : public Thread { |
| 59 | private: |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 60 | MtpDatabase* mDatabase; |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 61 | MtpServer* mServer; |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 62 | String8 mStoragePath; |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 63 | uint64_t mReserveSpace; |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 64 | jobject mJavaServer; |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 65 | int mFd; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 66 | |
| 67 | public: |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 68 | MtpThread(MtpDatabase* database, const char* storagePath, uint64_t reserveSpace, |
| 69 | jobject javaServer) |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 70 | : mDatabase(database), |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 71 | mServer(NULL), |
| 72 | mStoragePath(storagePath), |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 73 | mReserveSpace(reserveSpace), |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 74 | mJavaServer(javaServer), |
| 75 | mFd(-1) |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 76 | { |
| 77 | } |
| 78 | |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 79 | void setPtpMode(bool usePtp) { |
| 80 | sMutex.lock(); |
| 81 | if (mFd >= 0) { |
| 82 | ioctl(mFd, MTP_SET_INTERFACE_MODE, |
| 83 | (usePtp ? MTP_INTERFACE_MODE_PTP : MTP_INTERFACE_MODE_MTP)); |
| 84 | } else { |
| 85 | int fd = open("/dev/mtp_usb", O_RDWR); |
| 86 | if (fd >= 0) { |
| 87 | ioctl(fd, MTP_SET_INTERFACE_MODE, |
| 88 | (usePtp ? MTP_INTERFACE_MODE_PTP : MTP_INTERFACE_MODE_MTP)); |
| 89 | close(fd); |
| 90 | } |
| 91 | } |
| 92 | sMutex.unlock(); |
| 93 | } |
| 94 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 95 | virtual bool threadLoop() { |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 96 | sMutex.lock(); |
| 97 | mFd = open("/dev/mtp_usb", O_RDWR); |
| 98 | printf("open returned %d\n", mFd); |
| 99 | if (mFd < 0) { |
Mike Lockwood | 2b2bff5 | 2010-08-18 12:32:26 -0400 | [diff] [blame] | 100 | LOGE("could not open MTP driver\n"); |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 101 | sMutex.unlock(); |
Mike Lockwood | 2b2bff5 | 2010-08-18 12:32:26 -0400 | [diff] [blame] | 102 | return false; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 103 | } |
| 104 | |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 105 | mServer = new MtpServer(mFd, mDatabase, AID_SDCARD_RW, 0664, 0775); |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 106 | mServer->addStorage(mStoragePath, mReserveSpace); |
Mike Lockwood | 2b2bff5 | 2010-08-18 12:32:26 -0400 | [diff] [blame] | 107 | sMutex.unlock(); |
| 108 | |
| 109 | LOGD("MtpThread mServer->run"); |
| 110 | mServer->run(); |
Mike Lockwood | 2b2bff5 | 2010-08-18 12:32:26 -0400 | [diff] [blame] | 111 | |
| 112 | sMutex.lock(); |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 113 | close(mFd); |
| 114 | mFd = -1; |
Mike Lockwood | 2b2bff5 | 2010-08-18 12:32:26 -0400 | [diff] [blame] | 115 | delete mServer; |
| 116 | mServer = NULL; |
| 117 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 118 | JNIEnv* env = AndroidRuntime::getJNIEnv(); |
| 119 | env->SetIntField(mJavaServer, field_context, 0); |
| 120 | env->DeleteGlobalRef(mJavaServer); |
| 121 | sMutex.unlock(); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 122 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 123 | LOGD("threadLoop returning"); |
| 124 | return false; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 125 | } |
| 126 | |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 127 | void sendObjectAdded(MtpObjectHandle handle) { |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 128 | sMutex.lock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 129 | if (mServer) |
| 130 | mServer->sendObjectAdded(handle); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 131 | sMutex.unlock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 132 | } |
| 133 | |
| 134 | void sendObjectRemoved(MtpObjectHandle handle) { |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 135 | sMutex.lock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 136 | if (mServer) |
| 137 | mServer->sendObjectRemoved(handle); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame] | 138 | sMutex.unlock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 139 | } |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 140 | }; |
| 141 | |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 142 | #endif // HAVE_ANDROID_OS |
| 143 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 144 | static void |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 145 | android_media_MtpServer_setup(JNIEnv *env, jobject thiz, jobject javaDatabase, |
| 146 | jstring storagePath, jlong reserveSpace) |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 147 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 148 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 149 | LOGD("setup\n"); |
| 150 | |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 151 | MtpDatabase* database = getMtpDatabase(env, javaDatabase); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 152 | const char *storagePathStr = env->GetStringUTFChars(storagePath, NULL); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 153 | |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 154 | MtpThread* thread = new MtpThread(database, storagePathStr, |
| 155 | reserveSpace, env->NewGlobalRef(thiz)); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 156 | env->SetIntField(thiz, field_context, (int)thread); |
| 157 | |
| 158 | env->ReleaseStringUTFChars(storagePath, storagePathStr); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 159 | #endif |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 160 | } |
| 161 | |
| 162 | static void |
| 163 | android_media_MtpServer_finalize(JNIEnv *env, jobject thiz) |
| 164 | { |
| 165 | LOGD("finalize\n"); |
| 166 | } |
| 167 | |
| 168 | |
| 169 | static void |
| 170 | android_media_MtpServer_start(JNIEnv *env, jobject thiz) |
| 171 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 172 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 173 | LOGD("start\n"); |
| 174 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 175 | thread->run("MtpThread"); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 176 | #endif // HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | static void |
| 180 | android_media_MtpServer_stop(JNIEnv *env, jobject thiz) |
| 181 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 182 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 183 | LOGD("stop\n"); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 184 | #endif |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 185 | } |
| 186 | |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 187 | static void |
| 188 | android_media_MtpServer_send_object_added(JNIEnv *env, jobject thiz, jint handle) |
| 189 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 190 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 191 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 192 | if (thread) |
| 193 | thread->sendObjectAdded(handle); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 194 | #endif |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 195 | } |
| 196 | |
| 197 | static void |
| 198 | android_media_MtpServer_send_object_removed(JNIEnv *env, jobject thiz, jint handle) |
| 199 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 200 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 201 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 202 | if (thread) |
| 203 | thread->sendObjectRemoved(handle); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 204 | #endif |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 205 | } |
| 206 | |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 207 | static void |
| 208 | android_media_MtpServer_set_ptp_mode(JNIEnv *env, jobject thiz, jboolean usePtp) |
| 209 | { |
| 210 | #ifdef HAVE_ANDROID_OS |
| 211 | LOGD("set_ptp_mode\n"); |
| 212 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 213 | if (thread) |
| 214 | thread->setPtpMode(usePtp); |
| 215 | #endif |
| 216 | } |
| 217 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 218 | // ---------------------------------------------------------------------------- |
| 219 | |
| 220 | static JNINativeMethod gMethods[] = { |
Mike Lockwood | 7f36b19 | 2010-12-12 12:17:43 -0800 | [diff] [blame^] | 221 | {"native_setup", "(Landroid/media/MtpDatabase;Ljava/lang/String;J)V", |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 222 | (void *)android_media_MtpServer_setup}, |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 223 | {"native_finalize", "()V", (void *)android_media_MtpServer_finalize}, |
| 224 | {"native_start", "()V", (void *)android_media_MtpServer_start}, |
| 225 | {"native_stop", "()V", (void *)android_media_MtpServer_stop}, |
| 226 | {"native_send_object_added", "(I)V", (void *)android_media_MtpServer_send_object_added}, |
| 227 | {"native_send_object_removed", "(I)V", (void *)android_media_MtpServer_send_object_removed}, |
Mike Lockwood | eabe8bf | 2010-08-31 14:35:23 -0400 | [diff] [blame] | 228 | {"native_set_ptp_mode", "(Z)V", (void *)android_media_MtpServer_set_ptp_mode}, |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 229 | }; |
| 230 | |
| 231 | static const char* const kClassPathName = "android/media/MtpServer"; |
| 232 | |
| 233 | int register_android_media_MtpServer(JNIEnv *env) |
| 234 | { |
| 235 | jclass clazz; |
| 236 | |
| 237 | LOGD("register_android_media_MtpServer\n"); |
| 238 | |
| 239 | clazz = env->FindClass("android/media/MtpServer"); |
| 240 | if (clazz == NULL) { |
| 241 | LOGE("Can't find android/media/MtpServer"); |
| 242 | return -1; |
| 243 | } |
| 244 | field_context = env->GetFieldID(clazz, "mNativeContext", "I"); |
| 245 | if (field_context == NULL) { |
| 246 | LOGE("Can't find MtpServer.mNativeContext"); |
| 247 | return -1; |
| 248 | } |
| 249 | |
| 250 | return AndroidRuntime::registerNativeMethods(env, |
| 251 | "android/media/MtpServer", gMethods, NELEM(gMethods)); |
| 252 | } |