233 lines
6.5 KiB
C++
233 lines
6.5 KiB
C++
/**
|
|
* Copyright (C) ARM Limited 2014. All rights reserved.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation.
|
|
*/
|
|
|
|
#include "Setup.h"
|
|
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <sys/utsname.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
#include "Config.h"
|
|
#include "DynBuf.h"
|
|
#include "Logging.h"
|
|
|
|
bool getLinuxVersion(int version[3]) {
|
|
// Check the kernel version
|
|
struct utsname utsname;
|
|
if (uname(&utsname) != 0) {
|
|
logg->logMessage("%s(%s:%i): uname failed", __FUNCTION__, __FILE__, __LINE__);
|
|
return false;
|
|
}
|
|
|
|
version[0] = 0;
|
|
version[1] = 0;
|
|
version[2] = 0;
|
|
|
|
int part = 0;
|
|
char *ch = utsname.release;
|
|
while (*ch >= '0' && *ch <= '9' && part < 3) {
|
|
version[part] = 10*version[part] + *ch - '0';
|
|
|
|
++ch;
|
|
if (*ch == '.') {
|
|
++part;
|
|
++ch;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static int pgrep_gator(DynBuf *const printb) {
|
|
DynBuf b;
|
|
|
|
DIR *proc = opendir("/proc");
|
|
if (proc == NULL) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: opendir failed");
|
|
handleException();
|
|
}
|
|
|
|
int self = getpid();
|
|
|
|
struct dirent *dirent;
|
|
while ((dirent = readdir(proc)) != NULL) {
|
|
char *endptr;
|
|
const int pid = strtol(dirent->d_name, &endptr, 10);
|
|
if (*endptr != '\0' || (pid == self)) {
|
|
// Ignore proc items that are not integers like ., cpuinfo, etc...
|
|
continue;
|
|
}
|
|
|
|
if (!printb->printf("/proc/%i/stat", pid)) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: DynBuf::printf failed");
|
|
handleException();
|
|
}
|
|
|
|
if (!b.read(printb->getBuf())) {
|
|
// This is not a fatal error - the thread just doesn't exist any more
|
|
continue;
|
|
}
|
|
|
|
char *comm = strchr(b.getBuf(), '(');
|
|
if (comm == NULL) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: parsing stat begin failed");
|
|
handleException();
|
|
}
|
|
++comm;
|
|
char *const str = strrchr(comm, ')');
|
|
if (str == NULL) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: parsing stat end failed");
|
|
handleException();
|
|
}
|
|
*str = '\0';
|
|
|
|
if (strncmp(comm, "gator", 5) == 0) {
|
|
// Assume there is only one gator process
|
|
return pid;
|
|
}
|
|
}
|
|
|
|
closedir(proc);
|
|
|
|
return -1;
|
|
}
|
|
|
|
int update(const char *const gatorPath) {
|
|
printf("gator: starting\n");
|
|
|
|
int version[3];
|
|
if (!getLinuxVersion(version)) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: getLinuxVersion failed");
|
|
handleException();
|
|
}
|
|
|
|
if (KERNEL_VERSION(version[0], version[1], version[2]) < KERNEL_VERSION(2, 6, 32)) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: Streamline can't automatically setup gator as this kernel version is not supported. Please upgrade the kernel on your device.");
|
|
handleException();
|
|
}
|
|
|
|
if (KERNEL_VERSION(version[0], version[1], version[2]) < KERNEL_VERSION(3, 4, 0)) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: Streamline can't automatically setup gator as gator.ko is required for this version of Linux. Please build gator.ko and gatord and install them on your device.");
|
|
handleException();
|
|
}
|
|
|
|
if (access("/sys/module/gator", F_OK) == 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: Streamline has detected that the gator kernel module is loaded on your device. Please build an updated version of gator.ko and gatord and install them on your device.");
|
|
handleException();
|
|
}
|
|
|
|
if (geteuid() != 0) {
|
|
printf("gator: trying sudo\n");
|
|
execlp("sudo", "sudo", gatorPath, "-u", NULL);
|
|
// Streamline will provide the password if needed
|
|
|
|
printf("gator: trying su\n");
|
|
char buf[1<<10];
|
|
snprintf(buf, sizeof(buf), "%s -u", gatorPath);
|
|
execlp("su", "su", "-", "-c", buf, NULL);
|
|
// Streamline will provide the password if needed
|
|
|
|
logg->logError(__FILE__, __LINE__, "gator: error: Streamline was unable to sudo to root on your device. Please double check passwords, ensure sudo or su work with this user or try a different username.");
|
|
handleException();
|
|
}
|
|
printf("gator: now root\n");
|
|
|
|
// setenforce 0 not needed for userspace gator
|
|
|
|
// Kill existing gator
|
|
DynBuf gatorStatPath;
|
|
int gator_main = pgrep_gator(&gatorStatPath);
|
|
if (gator_main > 0) {
|
|
if (kill(gator_main, SIGTERM) != 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: kill SIGTERM failed");
|
|
handleException();
|
|
}
|
|
for (int i = 0; ; ++i) {
|
|
if (access(gatorStatPath.getBuf(), F_OK) != 0) {
|
|
break;
|
|
}
|
|
if (i == 5) {
|
|
if (kill(gator_main, SIGKILL) != 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: kill SIGKILL failed");
|
|
handleException();
|
|
}
|
|
} else if (i >= 10) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: unable to kill running gator");
|
|
handleException();
|
|
}
|
|
sleep(1);
|
|
}
|
|
}
|
|
printf("gator: no gatord running\n");
|
|
|
|
rename("gatord", "gatord.old");
|
|
rename("gator.ko", "gator.ko.old");
|
|
|
|
// Rename gatord.YYYYMMDDHHMMSSMMMM to gatord
|
|
char *newGatorPath = strdup(gatorPath);
|
|
char *dot = strrchr(newGatorPath, '.');
|
|
if (dot != NULL) {
|
|
*dot = '\0';
|
|
if (rename(gatorPath, newGatorPath) != 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: rename failed");
|
|
handleException();
|
|
}
|
|
}
|
|
|
|
// Fork and start gatord (redirect stdout and stderr)
|
|
int child = fork();
|
|
if (child < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: fork failed");
|
|
handleException();
|
|
} else if (child == 0) {
|
|
int inFd = open("/dev/null", O_RDONLY | O_CLOEXEC);
|
|
if (inFd < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: open of /dev/null failed");
|
|
handleException();
|
|
}
|
|
int outFd = open("gatord.out", O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0600);
|
|
if (outFd < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: open of gatord.out failed");
|
|
handleException();
|
|
}
|
|
int errFd = open("gatord.err", O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0600);
|
|
if (errFd < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: open of gatord.err failed");
|
|
handleException();
|
|
}
|
|
if (dup2(inFd, STDIN_FILENO) < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: dup2 for stdin failed");
|
|
handleException();
|
|
}
|
|
if (dup2(outFd, STDOUT_FILENO) < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: dup2 for stdout failed");
|
|
handleException();
|
|
}
|
|
if (dup2(errFd, STDERR_FILENO) < 0) {
|
|
logg->logError(__FILE__, __LINE__, "gator: error: dup2 for stderr failed");
|
|
handleException();
|
|
}
|
|
execlp(newGatorPath, newGatorPath, "-a", NULL);
|
|
logg->logError(__FILE__, __LINE__, "gator: error: execlp failed");
|
|
handleException();
|
|
}
|
|
|
|
printf("gator: done\n");
|
|
|
|
return 0;
|
|
}
|