summaryrefslogtreecommitdiffstats
path: root/mtp/twrpMtp.cpp
blob: 9a7df5be1926bcdd2f5b183580f6408862471de9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
 * Copyright (C) 2014 TeamWin - bigbiff and Dees_Troy mtp database conversion to C++
 *
 * 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.
 */


#include <string>
#include <fcntl.h>
#include <sys/wait.h>
#include <pthread.h>
#include <utils/Errors.h>
#include <utils/threads.h>
#include "MtpTypes.h"
#include "MtpPacket.h"
#include "MtpDataPacket.h"
#include "MtpDatabase.h"
#include "MtpRequestPacket.h"
#include "MtpResponsePacket.h"
#include "mtp_MtpDatabase.hpp"
#include "mtp_MtpServer.hpp"
#include "twrpMtp.hpp"
#include "MtpDebug.h"

#ifdef TWRPMTP
static void usage(std::string prg) {
	printf("Usage: %s <OPTIONS>\n", prg.c_str());
	printf("Options:\n");
	printf("\t-h, --help\t\tShow Usage\n");
	printf("\t-s1, --storage1 /path/to/dir\t\tDestination to first storage directory\n");
	printf("\t-s2, --storage2 /path/to/dir\t\tDestination to first storage directory\n");
	printf("\t-sN, --storageN /path/to/dir\t\tDestination to first storage directory\n");
}

int main(int argc, char* argv[]) {
	printf("argc: %d\n", argc);
	if (argc < 2) {
		usage(argv[0]);
		return 1;
	}

	std::vector <std::string> storages;

	for (int i = 1; i < argc; ++i) {
		std::string arg = argv[i];
		if ((arg == "-h") || (arg == "--help")) {
			usage(argv[0]);
		}
		else {
			storages.push_back(arg);
		}
	}
	printf("starting\n");
	twmtp_MtpServer* mtp = new twmtp_MtpServer();
	mtp->set_storages(storages);
	mtp->start();
	return 0;
}
#endif //def TWRPMTP

twrpMtp::twrpMtp(int debug_enabled = 0) {
	if (debug_enabled)
		MtpDebug::enableDebug();
	mtpstorages = new storages;
}

int twrpMtp::start(void) {
	MTPI("Starting MTP\n");
	twmtp_MtpServer *mtp = new twmtp_MtpServer();
	mtp->set_storages(mtpstorages);
	mtp->start();
	return 0;
}

pthread_t twrpMtp::threadserver(void) {
	pthread_t thread;
	ThreadPtr mtpptr = &twrpMtp::start;
	PThreadPtr p = *(PThreadPtr*)&mtpptr;
	pthread_create(&thread, NULL, p, this);
	return thread;
}

pid_t twrpMtp::forkserver(void) {
	pid_t pid;
	if ((pid = fork()) == -1) {
		MTPE("MTP fork failed.\n");
		return 0;
	}
	if (pid == 0) {
		// Child process
		start();
		MTPD("MTP child process exited.\n");
		_exit(0);
	} else {
		return pid;
	}
	return 0;
}

void twrpMtp::addStorage(std::string display, std::string path, int mtpid) {
	s = new storage;
	s->display = display;
	s->mount = path;
	s->mtpid = mtpid;
	MTPD("twrpMtp mtpid: %d\n", s->mtpid);
	mtpstorages->push_back(s);
}