4 * Copyright (c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
8 * Jaewon Lim <jaewon81.lim@samsung.com>
9 * Woojin Jung <woojin2.jung@samsung.com>
10 * Juyoung Kim <j0.kim@samsung.com>
12 * Licensed under the Apache License, Version 2.0 (the "License");
13 * you may not use this file except in compliance with the License.
14 * You may obtain a copy of the License at
16 * http://www.apache.org/licenses/LICENSE-2.0
18 * Unless required by applicable law or agreed to in writing, software
19 * distributed under the License is distributed on an "AS IS" BASIS,
20 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
21 * See the License for the specific language governing permissions and
22 * limitations under the License.
30 #include <stdlib.h> // for realpath
31 #include <string.h> // for strtok, strcpy, strncpy
32 #include <limits.h> // for realpath
34 #include <errno.h> // for errno
35 #include <sys/types.h> // for accept, mkdir, opendir, readdir
36 #include <sys/socket.h> // for accept
37 #include <sys/stat.h> // for mkdir
38 #include <sys/eventfd.h> // for eventfd
39 #include <sys/epoll.h> // for epoll apis
40 #include <unistd.h> // for access, sleep
41 #include <attr/xattr.h> // for fsetxattr
47 #define DA_WORK_DIR "/home/developer/sdk_tools/da/"
48 #define DA_READELF_PATH "/home/developer/sdk_tools/da/readelf"
49 #define SCREENSHOT_DIR "/tmp/da"
52 #define MAX_CONNECT_SIZE 12
54 static void terminate_error(char* errstr, int sendtohost);
56 long long get_total_alloc_size()
59 long long allocsize = 0;
61 for(i = 0; i < MAX_TARGET_COUNT; i++)
63 if(manager.target[i].socket != -1 && manager.target[i].allocmem > 0)
64 allocsize += manager.target[i].allocmem;
69 static int getEmptyTargetSlot()
72 for(i = 0; i < MAX_TARGET_COUNT; i++)
74 if(manager.target[i].socket == -1)
81 static void setEmptyTargetSlot(int index)
83 if(index >= 0 && index < MAX_TARGET_COUNT)
85 manager.target[index].pid = -1;
86 manager.target[index].recv_thread = -1;
87 manager.target[index].allocmem = 0;
88 manager.target[index].starttime = 0;
89 if(manager.target[index].event_fd != -1)
90 close(manager.target[index].event_fd);
91 manager.target[index].event_fd = -1;
92 if(manager.target[index].socket != -1)
93 close(manager.target[index].socket);
94 manager.target[index].socket = -1;
98 // ======================================================================================
99 // send functions to host
100 // ======================================================================================
102 int sendDataToHost(msg_t* log)
104 if (manager.host.data_socket != -1)
106 char logstr[DA_MSG_MAX];
110 loglen = sprintf(logstr, "%d|%d|%s\n", log->type, log->length + 1, log->data);
112 loglen = sprintf(logstr, "%d|%d|\n", log->type, log->length + 1);
114 // loglen = sprintf(logstr, "%d|%s\n", log->type, log->data);
116 pthread_mutex_lock(&(manager.host.data_socket_mutex));
117 send(manager.host.data_socket, logstr, loglen, MSG_NOSIGNAL);
118 pthread_mutex_unlock(&(manager.host.data_socket_mutex));
125 // msgstr can be NULL
126 static int sendACKStrToHost(enum HostMessageType resp, char* msgstr)
128 if (manager.host.control_socket != -1)
130 char logstr[DA_MSG_MAX];
134 loglen = sprintf(logstr, "%d|%d|%s", (int)resp, strlen(msgstr), msgstr);
136 loglen = sprintf(logstr, "%d|0|", (int)resp);
138 send(manager.host.control_socket, logstr, loglen, MSG_NOSIGNAL);
145 static int sendACKCodeToHost(enum HostMessageType resp, int msgcode)
147 if (manager.host.control_socket != -1)
150 char logstr[DA_MSG_MAX];
153 codelen = sprintf(codestr, "%d", msgcode);
154 loglen = sprintf(logstr, "%d|%d|%s", (int)resp, codelen, codestr);
156 send(manager.host.control_socket, logstr, loglen, MSG_NOSIGNAL);
163 // ========================================================================================
164 // start and terminate control functions
165 // ========================================================================================
167 static int startProfiling(long launchflag)
169 char execPath[PATH_MAX];
171 // remove previous screen capture files
172 remove_indir(SCREENSHOT_DIR);
173 mkdir(SCREENSHOT_DIR, 0777);
175 manager.config_flag = launchflag;
177 #ifdef RUN_APP_LOADER
178 strcpy(execPath, manager.appPath);
180 get_executable(manager.appPath, execPath, PATH_MAX);
182 if(exec_app(execPath, get_app_type(manager.appPath)))
184 if(samplingStart() < 0)
188 LOGI("Timer Started\n");
196 // terminate single target
197 // just send stop message to target process
198 static void terminate_target(int index)
202 sendlog.type = MSG_STOP;
205 if(manager.target[index].socket != -1)
207 // result of sending to disconnected socket is not expected
208 sendlen = send(manager.target[index].socket, &sendlog, sizeof(sendlog.type) + sizeof(sendlog.length), MSG_NOSIGNAL);
211 LOGI("TERMINATE send exit msg (socket %d) by terminate_target()\n", manager.target[index].socket);
216 // just send stop message to all target process
217 static void terminate_all_target()
223 sendlog.type = MSG_STOP;
226 for (i = 0; i < MAX_TARGET_COUNT; i++)
228 if(manager.target[i].socket != -1)
230 sendlen = send(manager.target[i].socket, &sendlog, sizeof(sendlog.type) + sizeof(sendlog.length), MSG_NOSIGNAL);
233 LOGI("TERMINATE send exit msg (socket %d) by terminate_all_target()\n", manager.target[i].socket);
239 // terminate all target and wait for threads
240 static void terminate_all()
243 terminate_all_target();
246 // wait for all other thread exit
247 for(i = 0; i < MAX_TARGET_COUNT; i++)
249 if(manager.target[i].recv_thread != -1)
251 pthread_join(manager.target[i].recv_thread, NULL);
256 // terminate all profiling by critical error
257 static void terminate_error(char* errstr, int sendtohost)
261 LOGE("TERMINATE ERROR: %s\n", errstr);
264 log.type = MSG_ERROR;
265 log.length = sprintf(log.data, "%s", errstr);
266 sendDataToHost(&log);
272 // ===========================================================================================
273 // message parsing and handling functions
274 // ===========================================================================================
276 // return 0 for normal case
277 // return negative value for error case
278 static int parseHostMessage(msg_t* log, char* msg)
282 int ret = 0; // parsing success
284 if(log == NULL || msg == NULL)
287 // Host message looks like this
288 // MSG_TYPE|MSG_LENGTH|MSG_STRING
289 // MSG_TYPE is always 3 digits number
293 log->type = atoi(msg);
296 for(i = 0; msg[i] != '\0'; i++)
305 log->length = atoi(msg);
317 msglen = strlen(msg);
319 if(msglen == log->length)
321 strcpy(log->data, msg);
322 log->data[log->length] = '\0';
324 // else if(msglen > log->length)
326 // strncpy(log->data, msg, log->length);
327 // log->data[log->length] = '\0';
331 ret = -1; // parsing error
336 ret = -1; // parsing error
342 ret = -1; // parsing error
348 // return 0 if normal case
349 // return plus value if non critical error occur
350 // return minus value if critical error occur
351 static int hostMessageHandler(msg_t* log)
355 char *barloc, *tmploc;
356 char execPath[PATH_MAX];
364 if(strcmp(PROTOCOL_VERSION, log->data) != 0)
366 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_PROTOCOL_VERSION);
370 sendACKStrToHost(MSG_OK, NULL);
374 LOGI("MSG_START handling : %s\n", log->data);
377 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_DATA);
378 return -1; // wrong message format
381 // parsing for host start status
383 barloc = strchr(tmploc, '|');
386 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_FORMAT);
387 return -1; // wrong message format
390 // parsing for target launch option flag
392 barloc = strchr(tmploc, '|');
395 while(tmploc < barloc)
397 flag = (flag * 10) + (*tmploc - '0');
403 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_FORMAT);
404 return -1; // wrong message format
406 LOGI("launch flag : %lx\n", flag);
408 // parsing for application package name
410 strcpy(manager.appPath, tmploc);
412 get_executable(manager.appPath, execPath, PATH_MAX); // get exact app executable file name
413 LOGI("executable app path %s\n", manager.appPath);
415 #ifdef RUN_APP_LOADER
416 kill_app(manager.appPath);
422 char command[PATH_MAX];
424 //save app install path
425 mkdir(DA_WORK_DIR, 0775);
427 "%s -Wwi %s | grep DW_AT_comp_dir > %s", DA_READELF_PATH,
428 execPath, DA_INSTALL_PATH);
429 LOGI("appInstallCommand %s\n", command);
433 "%s -h %s | grep Type | cut -d\" \" -f33 > %s", DA_READELF_PATH,
434 execPath, DA_BUILD_OPTION);
435 LOGI("appInstallCommand %s\n", command);
438 if(startProfiling(flag) < 0)
440 sendACKCodeToHost(MSG_NOTOK, ERR_CANNOT_START_PROFILING);
444 sendACKStrToHost(MSG_OK, NULL);
447 LOGI("MSG_STOP handling\n");
448 sendACKStrToHost(MSG_OK, NULL);
456 manager.config_flag = atoi(log->data);
457 sendACKStrToHost(MSG_OK, NULL);
459 LOGI("MSG_OPTION : str(%s), flag(%x)\n", log->data, manager.config_flag);
461 sendlog.type = MSG_OPTION;
462 sendlog.length = sprintf(sendlog.data, "%u", manager.config_flag);
464 for(i = 0; i < MAX_TARGET_COUNT; i++)
466 if(manager.target[i].socket != -1)
468 send(manager.target[i].socket, &sendlog, sizeof(sendlog.type) + sizeof(sendlog.length) + sendlog.length, MSG_NOSIGNAL);
474 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_DATA);
479 sendACKStrToHost(MSG_OK, NULL);
482 LOGW("Unknown msg\n");
483 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_TYPE);
491 // ========================================================================================
492 // socket and event_fd handling functions
493 // ========================================================================================
495 // return 0 if normal case
496 // return plus value if non critical error occur
497 // return minus value if critical error occur
498 // return -11 if all target process closed
499 static int targetEventHandler(int epollfd, int index, uint64_t msg)
505 if(index == 0) // assume index 0 is main application process
508 char tempBuff[DA_MSG_MAX];
509 char tempBuff2[DA_MSG_MAX];
510 char tempPath[PATH_MAX];
512 get_executable(manager.appPath, tempPath, PATH_MAX);
513 if(realpath(tempPath, tempBuff) == NULL)
515 LOGW("Failed to get realpath of app\n");
516 strcpy(tempBuff, tempPath);
519 sprintf(tempPath, "/proc/%d/maps", manager.target[index].pid);
520 sprintf(tempBuff2, "cat %s | grep %s | cut -d\"-\" -f1 > %s",
521 tempPath, tempBuff, DA_BASE_ADDRESS);
522 LOGI("base address command is %s\n", tempBuff2);
525 if(access(tempPath, F_OK) != 0)
527 if(is_same_app_process(manager.appPath, manager.target[index].pid) == 0)
531 if(get_app_base_address(&base_address) == 1)
537 get_app_install_path(tempPath, PATH_MAX);
538 get_device_info(tempBuff, DA_MSG_MAX);
539 log.type = MSG_DEVICE;
540 if (strlen(tempPath) > 0)
542 get_executable(manager.appPath, tempBuff2, DA_MSG_MAX);
543 log.length = sprintf(log.data, "%s`,%d`,%Lu`,%d`,%u`,%d`,%s/%s", tempBuff,
544 manager.target[index].pid, manager.target[index].starttime,
545 is_app_built_pie(), base_address, get_app_type(manager.appPath),
546 tempPath, get_app_name(tempBuff2));
550 log.length = sprintf(log.data, "%s`,%d`,%Lu`,%d`,%u`,%d`,", tempBuff,
551 manager.target[index].pid, manager.target[index].starttime,
552 is_app_built_pie(), base_address, get_app_type(manager.appPath));
555 LOGI("%s\n", log.data);
560 log.length = sprintf(log.data, "%d`,%Lu", manager.target[index].pid, manager.target[index].starttime);
563 sendDataToHost(&log);
566 if(msg & EVENT_STOP || msg & EVENT_ERROR)
568 LOGI("target close, socket(%d), pid(%d) : (remaining %d target)\n",
569 manager.target[index].socket, manager.target[index].pid, manager.target_count - 1);
571 terminate_target(index);
572 epoll_ctl(epollfd, EPOLL_CTL_DEL, manager.target[index].event_fd, NULL);
573 setEmptyTargetSlot(index);
574 if (0 == __sync_sub_and_fetch(&manager.target_count, 1)) // all target client are closed
576 log.type = MSG_TERMINATE;
579 sendDataToHost(&log);
587 // return 0 if normal case
588 // return plus value if non critical error occur
589 // return minus value if critical error occur
590 static int targetServerHandler(int efd)
593 struct epoll_event ev;
595 int index = getEmptyTargetSlot();
596 if(index == MAX_TARGET_COUNT)
598 LOGW("Max target number(8) reached, no more target can connected\n");
602 manager.target[index].socket = accept(manager.target_server_socket, NULL, NULL);
604 if(manager.target[index].socket >= 0) // accept succeed
606 // set smack attribute for certification
607 fsetxattr(manager.target[index].socket, "security.SMACK64IPIN", "*", 1, 0);
608 fsetxattr(manager.target[index].socket, "security.SMACK64IPOUT", "*", 1, 0);
610 // send config message to target process
611 log.type = MSG_OPTION;
612 log.length = sprintf(log.data, "%u", manager.config_flag);
613 send(manager.target[index].socket, &log, sizeof(log.type) + sizeof(log.length) + log.length, MSG_NOSIGNAL);
616 manager.target[index].event_fd = eventfd(0, EFD_NONBLOCK);
617 if(manager.target[index].event_fd == -1)
619 // fail to make event fd
620 LOGE("fail to make event fd for socket (%d)\n", manager.target[index].socket);
621 goto TARGET_CONNECT_FAIL;
624 // add event fd to epoll list
626 ev.data.fd = manager.target[index].event_fd;
627 if(epoll_ctl(efd, EPOLL_CTL_ADD, manager.target[index].event_fd, &ev) < 0)
629 // fail to add event fd
630 LOGE("fail to add event fd to epoll list for socket (%d)\n", manager.target[index].socket);
631 goto TARGET_CONNECT_FAIL;
634 // make recv thread for target
635 if(makeRecvThread(index) != 0)
637 // fail to make recv thread
638 LOGE("fail to make recv thread for socket (%d)\n", manager.target[index].socket);
639 epoll_ctl(efd, EPOLL_CTL_DEL, manager.target[index].event_fd, NULL);
640 goto TARGET_CONNECT_FAIL;
643 LOGI("target connected = %d(running %d target)\n",
644 manager.target[index].socket, manager.target_count + 1);
646 manager.target_count++;
651 LOGE("Failed to accept at target server socket\n");
655 if(manager.target_count == 0) // if this connection is main connection
659 else // if this connection is not main connection then ignore process by error
661 setEmptyTargetSlot(index);
666 // return 0 if normal case
667 // return plus value if non critical error occur
668 // return minus value if critical error occur
669 static int hostServerHandler(int efd)
671 static int hostserverorder = 0;
673 struct epoll_event ev;
675 if(hostserverorder > 1) // control and data socket connected already
678 csocket = accept(manager.host_server_socket, NULL, NULL);
680 if(csocket >= 0) // accept succeed
683 ev.data.fd = csocket;
684 if(epoll_ctl(efd, EPOLL_CTL_ADD, csocket, &ev) < 0)
686 // consider as accept fail
687 LOGE("Failed to add socket fd to epoll list\n");
692 if(hostserverorder == 0)
694 manager.host.control_socket = csocket;
695 LOGI("host control socket connected = %d\n", csocket);
699 manager.host.data_socket = csocket;
700 LOGI("host data socket connected = %d\n", csocket);
708 LOGE("Failed to accept from host server socket\n");
713 // return 0 if normal case
714 // return plus value if non critical error occur
715 // return minus value if critical error occur
716 // return -11 if socket closed
717 static int controlSocketHandler()
720 char recvBuf[DA_MSG_MAX];
723 // host log format xxx|length|str
724 recvLen = recv(manager.host.control_socket, recvBuf, RECV_BUF_MAX, 0);
728 recvBuf[recvLen] = '\0';
729 LOGI("host sent control msg str(%s)\n", recvBuf);
731 if(parseHostMessage(&log, recvBuf) < 0)
733 // error to parse host message
734 sendACKCodeToHost(MSG_NOTOK, ERR_WRONG_MESSAGE_FORMAT);
738 // host msg command handling
739 return hostMessageHandler(&log);
741 else // close request from HOST
747 // return 0 for normal case
750 int ret = 0; // return value
754 struct epoll_event ev, *events;
756 int numevent; // number of occured events
758 // initialize epoll event pool
759 events = (struct epoll_event*) malloc(sizeof(struct epoll_event) * EPOLL_SIZE);
762 LOGE("Out of memory when allocate epoll event pool\n");
766 if((efd = epoll_create(MAX_CONNECT_SIZE)) < 0)
768 LOGE("epoll creation error\n");
773 // add server sockets to epoll event pool
775 ev.data.fd = manager.host_server_socket;
776 if(epoll_ctl(efd, EPOLL_CTL_ADD, manager.host_server_socket, &ev) < 0)
778 LOGE("Host server socket epoll_ctl error\n");
783 ev.data.fd = manager.target_server_socket;
784 if(epoll_ctl(efd, EPOLL_CTL_ADD, manager.target_server_socket, &ev) < 0)
786 LOGE("Target server socket epoll_ctl error\n");
794 numevent = epoll_wait(efd, events, EPOLL_SIZE, -1);
797 LOGE("Failed to epoll_wait : num of event(%d), errno(%d)\n", numevent, errno);
801 for(i = 0; i < numevent; i++)
803 // check for request from event fd
804 for(k = 0; k < MAX_TARGET_COUNT; k++)
806 if(manager.target[k].socket != -1 &&
807 events[i].data.fd == manager.target[k].event_fd)
810 recvLen = read(manager.target[k].event_fd, &u, sizeof(uint64_t));
811 if(recvLen != sizeof(uint64_t))
813 // maybe closed, but ignoring is more safe then removing fd from epoll list
817 if(-11 == targetEventHandler(efd, k, u))
819 LOGI("all target process is closed\n");
829 if(k != MAX_TARGET_COUNT)
832 // connect request from target
833 if(events[i].data.fd == manager.target_server_socket)
835 if(targetServerHandler(efd) < 0) // critical error
837 terminate_error("Internal DA framework error, Please re-run the profiling.", 1);
842 // connect request from host
843 else if(events[i].data.fd == manager.host_server_socket)
845 int result = hostServerHandler(efd);
848 terminate_error("Internal DA framework error, Please re-run the profiling.", 1);
853 // control message from host
854 else if(events[i].data.fd == manager.host.control_socket)
856 int result = controlSocketHandler();
857 if(result == -11) // socket close
859 // close target and host socket and quit
860 LOGI("host close = %d\n", manager.host.control_socket);
867 terminate_error("Internal DA framework error, Please re-run the profiling.", 1);
872 else if(events[i].data.fd == manager.host.data_socket)
874 LOGW("host message from data socket\n");
880 LOGW("Unknown socket fd (%d)\n", events[i].data.fd);