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> |
| 25 | #include <utils/threads.h> |
| 26 | |
| 27 | #include "jni.h" |
| 28 | #include "JNIHelp.h" |
| 29 | #include "android_runtime/AndroidRuntime.h" |
Mike Lockwood | dad6927 | 2010-07-02 15:15:07 -0400 | [diff] [blame] | 30 | #include "private/android_filesystem_config.h" |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 31 | |
| 32 | #include "MtpServer.h" |
| 33 | |
Mike Lockwood | 81ea83d | 2010-06-30 17:49:41 -0400 | [diff] [blame] | 34 | using namespace android; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 35 | |
| 36 | // ---------------------------------------------------------------------------- |
| 37 | |
| 38 | static jfieldID field_context; |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 39 | static Mutex sMutex; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 40 | |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 41 | // in android_media_MtpDatabase.cpp |
| 42 | extern MtpDatabase* getMtpDatabase(JNIEnv *env, jobject database); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 43 | |
| 44 | // ---------------------------------------------------------------------------- |
| 45 | |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 46 | #ifdef HAVE_ANDROID_OS |
| 47 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 48 | static bool ExceptionCheck(void* env) |
| 49 | { |
| 50 | return ((JNIEnv *)env)->ExceptionCheck(); |
| 51 | } |
| 52 | |
| 53 | class MtpThread : public Thread { |
| 54 | private: |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 55 | MtpDatabase* mDatabase; |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 56 | MtpServer* mServer; |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 57 | String8 mStoragePath; |
| 58 | bool mDone; |
| 59 | jobject mJavaServer; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 60 | |
| 61 | public: |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 62 | MtpThread(MtpDatabase* database, const char* storagePath, jobject javaServer) |
| 63 | : mDatabase(database), |
| 64 | mServer(NULL), |
| 65 | mStoragePath(storagePath), |
| 66 | mDone(false), |
| 67 | mJavaServer(javaServer) |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 68 | { |
| 69 | } |
| 70 | |
| 71 | virtual bool threadLoop() { |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 72 | while (1) { |
| 73 | int fd = open("/dev/mtp_usb", O_RDWR); |
| 74 | printf("open returned %d\n", fd); |
| 75 | if (fd < 0) { |
| 76 | LOGE("could not open MTP driver\n"); |
| 77 | break; |
| 78 | } |
| 79 | |
| 80 | sMutex.lock(); |
| 81 | mServer = new MtpServer(fd, mDatabase, AID_SDCARD_RW, 0664, 0775); |
| 82 | mServer->addStorage(mStoragePath); |
| 83 | sMutex.unlock(); |
| 84 | |
| 85 | LOGD("MtpThread mServer->run"); |
| 86 | mServer->run(); |
| 87 | close(fd); |
| 88 | |
| 89 | sMutex.lock(); |
| 90 | delete mServer; |
| 91 | mServer = NULL; |
| 92 | if (mDone) |
| 93 | goto done; |
| 94 | sMutex.unlock(); |
| 95 | // wait a bit before retrying |
| 96 | sleep(1); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 97 | } |
| 98 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 99 | sMutex.lock(); |
| 100 | done: |
| 101 | JNIEnv* env = AndroidRuntime::getJNIEnv(); |
| 102 | env->SetIntField(mJavaServer, field_context, 0); |
| 103 | env->DeleteGlobalRef(mJavaServer); |
| 104 | sMutex.unlock(); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 105 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 106 | LOGD("threadLoop returning"); |
| 107 | return false; |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 108 | } |
| 109 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 110 | void setDone() { |
| 111 | LOGD("setDone"); |
| 112 | mDone = true; |
| 113 | } |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 114 | |
| 115 | void sendObjectAdded(MtpObjectHandle handle) { |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 116 | sMutex.lock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 117 | if (mServer) |
| 118 | mServer->sendObjectAdded(handle); |
| 119 | else |
| 120 | LOGE("sendObjectAdded called while disconnected\n"); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 121 | sMutex.unlock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 122 | } |
| 123 | |
| 124 | void sendObjectRemoved(MtpObjectHandle handle) { |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 125 | sMutex.lock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 126 | if (mServer) |
| 127 | mServer->sendObjectRemoved(handle); |
| 128 | else |
| 129 | LOGE("sendObjectRemoved called while disconnected\n"); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 130 | sMutex.unlock(); |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 131 | } |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 132 | }; |
| 133 | |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 134 | #endif // HAVE_ANDROID_OS |
| 135 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 136 | static void |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 137 | android_media_MtpServer_setup(JNIEnv *env, jobject thiz, jobject javaDatabase, jstring storagePath) |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 138 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 139 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 140 | LOGD("setup\n"); |
| 141 | |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 142 | MtpDatabase* database = getMtpDatabase(env, javaDatabase); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 143 | const char *storagePathStr = env->GetStringUTFChars(storagePath, NULL); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 144 | |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 145 | MtpThread* thread = new MtpThread(database, storagePathStr, env->NewGlobalRef(thiz)); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 146 | env->SetIntField(thiz, field_context, (int)thread); |
| 147 | |
| 148 | env->ReleaseStringUTFChars(storagePath, storagePathStr); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 149 | #endif |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 150 | } |
| 151 | |
| 152 | static void |
| 153 | android_media_MtpServer_finalize(JNIEnv *env, jobject thiz) |
| 154 | { |
| 155 | LOGD("finalize\n"); |
| 156 | } |
| 157 | |
| 158 | |
| 159 | static void |
| 160 | android_media_MtpServer_start(JNIEnv *env, jobject thiz) |
| 161 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 162 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 163 | LOGD("start\n"); |
| 164 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 165 | thread->run("MtpThread"); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 166 | #endif // HAVE_ANDROID_OS |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 167 | } |
| 168 | |
| 169 | static void |
| 170 | android_media_MtpServer_stop(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("stop\n"); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 174 | sMutex.lock(); |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 175 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 176 | if (thread) |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 177 | thread->setDone(); |
Mike Lockwood | 23ee42f | 2010-08-03 15:17:55 -0400 | [diff] [blame^] | 178 | sMutex.unlock(); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 179 | #endif |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 180 | } |
| 181 | |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 182 | static void |
| 183 | android_media_MtpServer_send_object_added(JNIEnv *env, jobject thiz, jint handle) |
| 184 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 185 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 186 | LOGD("send_object_added %d\n", handle); |
| 187 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 188 | if (thread) |
| 189 | thread->sendObjectAdded(handle); |
| 190 | else |
| 191 | LOGE("sendObjectAdded called while disconnected\n"); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 192 | #endif |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 193 | } |
| 194 | |
| 195 | static void |
| 196 | android_media_MtpServer_send_object_removed(JNIEnv *env, jobject thiz, jint handle) |
| 197 | { |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 198 | #ifdef HAVE_ANDROID_OS |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 199 | LOGD("send_object_removed %d\n", handle); |
| 200 | MtpThread *thread = (MtpThread *)env->GetIntField(thiz, field_context); |
| 201 | if (thread) |
| 202 | thread->sendObjectRemoved(handle); |
| 203 | else |
| 204 | LOGE("sendObjectRemoved called while disconnected\n"); |
Mike Lockwood | ff164a7 | 2010-07-15 15:01:17 -0400 | [diff] [blame] | 205 | #endif |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 206 | } |
| 207 | |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 208 | // ---------------------------------------------------------------------------- |
| 209 | |
| 210 | static JNINativeMethod gMethods[] = { |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 211 | {"native_setup", "(Landroid/media/MtpDatabase;Ljava/lang/String;)V", |
Mike Lockwood | d21eac9 | 2010-07-03 00:44:05 -0400 | [diff] [blame] | 212 | (void *)android_media_MtpServer_setup}, |
Mike Lockwood | be125a5 | 2010-07-12 18:54:16 -0400 | [diff] [blame] | 213 | {"native_finalize", "()V", (void *)android_media_MtpServer_finalize}, |
| 214 | {"native_start", "()V", (void *)android_media_MtpServer_start}, |
| 215 | {"native_stop", "()V", (void *)android_media_MtpServer_stop}, |
| 216 | {"native_send_object_added", "(I)V", (void *)android_media_MtpServer_send_object_added}, |
| 217 | {"native_send_object_removed", "(I)V", (void *)android_media_MtpServer_send_object_removed}, |
Mike Lockwood | 98ef64e | 2010-06-29 16:42:13 -0400 | [diff] [blame] | 218 | }; |
| 219 | |
| 220 | static const char* const kClassPathName = "android/media/MtpServer"; |
| 221 | |
| 222 | int register_android_media_MtpServer(JNIEnv *env) |
| 223 | { |
| 224 | jclass clazz; |
| 225 | |
| 226 | LOGD("register_android_media_MtpServer\n"); |
| 227 | |
| 228 | clazz = env->FindClass("android/media/MtpServer"); |
| 229 | if (clazz == NULL) { |
| 230 | LOGE("Can't find android/media/MtpServer"); |
| 231 | return -1; |
| 232 | } |
| 233 | field_context = env->GetFieldID(clazz, "mNativeContext", "I"); |
| 234 | if (field_context == NULL) { |
| 235 | LOGE("Can't find MtpServer.mNativeContext"); |
| 236 | return -1; |
| 237 | } |
| 238 | |
| 239 | return AndroidRuntime::registerNativeMethods(env, |
| 240 | "android/media/MtpServer", gMethods, NELEM(gMethods)); |
| 241 | } |