#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>

#include "magisk.h"
#include "db.h"

void INIT_DB_STRINGS(struct db_strings *str) {
	for (int i = 0; i < DB_STRING_NUM; ++i)
		str->s[i][0] = '\0';
}

static int policy_cb(void *v, int col_num, char **data, char **col_name) {
	struct su_access *su = v;
	for (int i = 0; i < col_num; i++) {
		if (strcmp(col_name[i], "policy") == 0)
			su->policy = (policy_t) atoi(data[i]);
		else if (strcmp(col_name[i], "logging") == 0)
			su->log = atoi(data[i]);
		else if (strcmp(col_name[i], "notification") == 0)
			su->notify = atoi(data[i]);
	}
	LOGD("su_db: query policy=[%d] log=[%d] notify=[%d]\n", su->policy, su->log, su->notify);
	return 0;
}

static int settings_cb(void *v, int col_num, char **data, char **col_name) {
	struct db_settings *dbs = v;
	int key = -1, value;
	for (int i = 0; i < col_num; ++i) {
		if (strcmp(col_name[i], "key") == 0) {
			for (int k = 0; k < DB_SETTINGS_NUM; ++k) {
				if (strcmp(data[i], DB_SETTING_KEYS[k]) == 0)
					key = k;
			}
		} else if (strcmp(col_name[i], "value") == 0) {
			value = atoi(data[i]);
		}
	}
	if (key >= 0) {
		dbs->v[key] = value;
		LOGD("su_db: query %s=[%d]\n", DB_SETTING_KEYS[key], value);
	}
	return 0;
}

static int strings_cb(void *v, int col_num, char **data, char **col_name) {
	struct db_strings *dbs = v;
	int key = -1;
	char *value;
	for (int i = 0; i < col_num; ++i) {
		if (strcmp(col_name[i], "key") == 0) {
			for (int k = 0; k < DB_STRING_NUM; ++k) {
				if (strcmp(data[i], DB_STRING_KEYS[k]) == 0)
					key = k;
			}
		} else if (strcmp(col_name[i], "value") == 0) {
			value = data[i];
		}
	}
	if (key >= 0) {
		strcpy(dbs->s[key], value);
		LOGD("su_db: query %s=[%s]\n", DB_STRING_KEYS[key], value);
	}
	return 0;
}

sqlite3 *get_magiskdb() {
	sqlite3 *db = NULL;
	if (access(MAGISKDB, R_OK) == 0) {
		// Open database
		int ret = sqlite3_open_v2(MAGISKDB, &db, SQLITE_OPEN_READONLY, NULL);
		if (ret) {
			LOGE("sqlite3 open failure: %s\n", sqlite3_errstr(ret));
			sqlite3_close(db);
			db = NULL;
		}
	}
	return db;
}

int get_db_settings(sqlite3 *db, int key, struct db_settings *dbs) {
	char *err;
	if (key > 0) {
		char query[128];
		sprintf(query, "SELECT key, value FROM settings WHERE key=%d", key);
		sqlite3_exec(db, query, settings_cb, dbs, &err);
	} else {
		sqlite3_exec(db, "SELECT key, value FROM settings", settings_cb, dbs, &err);
	}
	if (err) {
		LOGE("sqlite3_exec: %s\n", err);
		return 1;
	}
	return 0;
}

int get_db_strings(sqlite3 *db, int key, struct db_strings *str) {
	char *err;
	if (key > 0) {
		char query[128];
		sprintf(query, "SELECT key, value FROM strings WHERE key=%d", key);
		sqlite3_exec(db, query, strings_cb, str, &err);
	} else {
		sqlite3_exec(db, "SELECT key, value FROM strings", strings_cb, str, &err);
	}
	if (err) {
		LOGE("sqlite3_exec: %s\n", err);
		return 1;
	}
	return 0;
}

int get_uid_policy(sqlite3 *db, int uid, struct su_access *su) {
	char query[256], *err;
	sprintf(query, "SELECT policy, logging, notification FROM policies "
			"WHERE uid=%d AND (until=0 OR until>%li)", uid, time(NULL));
	sqlite3_exec(db, query, policy_cb, su, &err);
	if (err) {
		LOGE("sqlite3_exec: %s\n", err);
		return 1;
	}
	return 0;
}

int validate_manager(char *pkg, int userid, struct stat *st) {
	if (st == NULL) {
		struct stat stat;
		st = &stat;
	}
	// Prefer DE storage
	const char *base = access("/data/user_de", F_OK) == 0 ? "/data/user_de" : "/data/user";
	char app_path[128];
	sprintf(app_path, "%s/%d/%s", base, userid, pkg[0] ? pkg : "xxx");
	if (stat(app_path, st)) {
		// Check the official package name
		sprintf(app_path, "%s/%d/"JAVA_PACKAGE_NAME, base, userid);
		if (stat(app_path, st)) {
			LOGE("su: cannot find manager");
			memset(st, 0, sizeof(*st));
			pkg[0] = '\0';
			return 1;
		} else {
			// Switch to official package if exists
			strcpy(pkg, JAVA_PACKAGE_NAME);
		}
	}
	return 0;
}