blob: d47b8fa0d20db8fa82990b9cd317e466d1d33040 [file] [log] [blame]
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001/*
2 * Copyright (C) 2014 TeamWin - bigbiff and Dees_Troy mtp database conversion to C++
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
18#include <string>
19#include <fcntl.h>
20#include <sys/wait.h>
21#include <pthread.h>
22#include <utils/Errors.h>
23#include <utils/threads.h>
24#include "MtpTypes.h"
25#include "MtpPacket.h"
26#include "MtpDataPacket.h"
27#include "MtpDatabase.h"
28#include "MtpRequestPacket.h"
29#include "MtpResponsePacket.h"
30#include "mtp_MtpDatabase.hpp"
31#include "mtp_MtpServer.hpp"
32#include "twrpMtp.hpp"
33#include "MtpDebug.h"
34
35#ifdef TWRPMTP
36static void usage(std::string prg) {
37 printf("Usage: %s <OPTIONS>\n", prg.c_str());
38 printf("Options:\n");
39 printf("\t-h, --help\t\tShow Usage\n");
40 printf("\t-s1, --storage1 /path/to/dir\t\tDestination to first storage directory\n");
41 printf("\t-s2, --storage2 /path/to/dir\t\tDestination to first storage directory\n");
42 printf("\t-sN, --storageN /path/to/dir\t\tDestination to first storage directory\n");
43}
44
45int main(int argc, char* argv[]) {
46 printf("argc: %d\n", argc);
47 if (argc < 2) {
48 usage(argv[0]);
49 return 1;
50 }
51
52 std::vector <std::string> storages;
53
54 for (int i = 1; i < argc; ++i) {
55 std::string arg = argv[i];
56 if ((arg == "-h") || (arg == "--help")) {
57 usage(argv[0]);
58 }
59 else {
60 storages.push_back(arg);
61 }
62 }
63 printf("starting\n");
64 twmtp_MtpServer* mtp = new twmtp_MtpServer();
65 mtp->set_storages(storages);
66 mtp->start();
67 return 0;
68}
69#endif //def TWRPMTP
70
Ethan Yonker6d154c42014-09-03 14:19:43 -050071twrpMtp::twrpMtp(int debug_enabled = 0) {
72 if (debug_enabled)
73 MtpDebug::enableDebug();
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040074 mtpstorages = new storages;
Ethan Yonker726a0202014-12-16 20:01:38 -060075 mtp_read_pipe = -1;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040076}
77
78int twrpMtp::start(void) {
79 MTPI("Starting MTP\n");
80 twmtp_MtpServer *mtp = new twmtp_MtpServer();
81 mtp->set_storages(mtpstorages);
Ethan Yonker726a0202014-12-16 20:01:38 -060082 mtp->set_read_pipe(mtp_read_pipe);
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040083 mtp->start();
84 return 0;
85}
86
Ethan Yonker8dfa7772014-09-04 21:48:41 -050087pthread_t twrpMtp::threadserver(void) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -040088 pthread_t thread;
89 ThreadPtr mtpptr = &twrpMtp::start;
90 PThreadPtr p = *(PThreadPtr*)&mtpptr;
91 pthread_create(&thread, NULL, p, this);
92 return thread;
93}
94
Ethan Yonker726a0202014-12-16 20:01:38 -060095pid_t twrpMtp::forkserver(int mtppipe[2]) {
Ethan Yonker8dfa7772014-09-04 21:48:41 -050096 pid_t pid;
97 if ((pid = fork()) == -1) {
98 MTPE("MTP fork failed.\n");
99 return 0;
100 }
101 if (pid == 0) {
102 // Child process
Ethan Yonker726a0202014-12-16 20:01:38 -0600103 close(mtppipe[1]); // Child closes write side
104 mtp_read_pipe = mtppipe[0];
Ethan Yonker8dfa7772014-09-04 21:48:41 -0500105 start();
Ethan Yonker8613dc02014-09-11 09:28:20 -0500106 MTPD("MTP child process exited.\n");
Ethan Yonker726a0202014-12-16 20:01:38 -0600107 close(mtppipe[0]);
Ethan Yonker8613dc02014-09-11 09:28:20 -0500108 _exit(0);
Ethan Yonker8dfa7772014-09-04 21:48:41 -0500109 } else {
110 return pid;
111 }
112 return 0;
113}
114
bigbiff7cb4c332014-11-26 20:36:07 -0500115void twrpMtp::addStorage(std::string display, std::string path, int mtpid, uint64_t maxFileSize) {
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400116 s = new storage;
117 s->display = display;
118 s->mount = path;
119 s->mtpid = mtpid;
bigbiff7cb4c332014-11-26 20:36:07 -0500120 s->maxFileSize = maxFileSize;
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -0400121 MTPD("twrpMtp mtpid: %d\n", s->mtpid);
122 mtpstorages->push_back(s);
123}