kore-jsonrtc/jsonrtc/src/jsonrtc.c

#include <kore/kore.h>
#include <kore/http.h>

#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <sys/ioctl.h>
#include <linux/rtc.h>
#include <cjson/cJSON.h>


int		page(struct http_request *);
int handle_time(struct http_request *);
int handle_wkalrm(struct http_request *);

int
page(struct http_request *req)
{
	http_response(req, 200, NULL, 0);
	return (KORE_RESULT_OK);
}

static char * read_body(struct http_request *req) {
	int ret;
	struct kore_buf * buf;
	u_int8_t data[128];
	char * body;
	buf = kore_buf_alloc(128);
	while(1) {
		ret = http_body_read(req, data, sizeof(data));
		if (ret == -1) {
			kore_buf_free(buf);
			return NULL;
                } else if (ret == 0) {
			break;
                }
			kore_buf_append(buf,data,ret);
	}
	body = kore_buf_stringify(buf, NULL);
        kore_buf_free(buf);
	return body;
}

int handle_time(struct http_request *req) {
	int fdesc;
	struct rtc_time time;
	int ret;
	char *result_string;
	cJSON *time_json, *json_min, *json_hour, *json_mday,
		*json_mon, *json_year, *json_sec, *json_wday;
	if(req->method == HTTP_METHOD_GET) {
		time_json = cJSON_CreateObject();
		fdesc = open("/dev/rtc", O_RDONLY);
		ret = ioctl(fdesc, RTC_RD_TIME, &time);
		close(fdesc);
		if (!ret) {
			json_sec = cJSON_CreateNumber(time.tm_sec);
			json_min = cJSON_CreateNumber(time.tm_min);
			json_hour = cJSON_CreateNumber(time.tm_hour);
			json_mday = cJSON_CreateNumber(time.tm_mday);
			json_mon = cJSON_CreateNumber(time.tm_mon);
			json_year = cJSON_CreateNumber(time.tm_year);
			json_wday = cJSON_CreateNumber(time.tm_wday);
			cJSON_AddItemToObject(time_json, "seconds", json_sec);
			cJSON_AddItemToObject(time_json, "minutes", json_min);
			cJSON_AddItemToObject(time_json, "hour", json_hour);
			cJSON_AddItemToObject(time_json, "mday", json_mday);
			cJSON_AddItemToObject(time_json, "month", json_mon);
			cJSON_AddItemToObject(time_json, "year", json_year);
			cJSON_AddItemToObject(time_json, "wday", json_wday);
			result_string = cJSON_Print(time_json);
			cJSON_Delete(time_json);
			http_response_header(req, "Content-Type", "application/json");
			http_response(req, 200, result_string, strlen(result_string));
			return KORE_RESULT_OK;
		}
		cJSON_Delete(time_json);
		result_string = "Error: could not get time from /dev/rtc\n";
		http_response(req, 500, result_string, strlen(result_string));
		return KORE_RESULT_OK;
	} else if (req->method == HTTP_METHOD_POST) {
		char * body;
		body = read_body(req);
		if (body == NULL) {
			result_string = "Error: could not read request body\n";
			http_response(req, 500, result_string, strlen(result_string));
			return (KORE_RESULT_OK);
		}
		time_json = cJSON_Parse(body);
		if (time_json == NULL) {
			result_string = "Error: could not parse json\n";
			http_response(req, 500, result_string, strlen(result_string));
			return (KORE_RESULT_OK);
		}
		json_sec = cJSON_GetObjectItemCaseSensitive(time_json, "seconds");
		json_min = cJSON_GetObjectItemCaseSensitive(time_json, "minutes");
		json_hour = cJSON_GetObjectItemCaseSensitive(time_json, "hour");
		json_mday = cJSON_GetObjectItemCaseSensitive(time_json, "mday");
		json_mon = cJSON_GetObjectItemCaseSensitive(time_json, "month");
		json_year = cJSON_GetObjectItemCaseSensitive(time_json, "year");
		json_wday = cJSON_GetObjectItemCaseSensitive(time_json, "wday");
		if (cJSON_IsNumber(json_sec) && cJSON_IsNumber(json_min) && cJSON_IsNumber(json_hour) && 
			cJSON_IsNumber(json_mday) && cJSON_IsNumber(json_mon) && 
			cJSON_IsNumber(json_year) && cJSON_IsNumber(json_wday)) {
			time.tm_sec = json_sec->valueint;
			time.tm_min = json_min->valueint;
			time.tm_hour = json_hour->valueint;
			time.tm_mday = json_mday->valueint;
			time.tm_mon = json_mon->valueint;
			time.tm_year = json_year->valueint;
			time.tm_wday = json_wday->valueint;
		} else {
			cJSON_Delete(time_json);
			result_string = "Error: invalid or missing values in json object\n";
			http_response(req, 500, result_string, strlen(result_string));
                        return (KORE_RESULT_OK);
		}
		cJSON_Delete(time_json);
		fdesc = open("/dev/rtc", O_WRONLY);
		ret = ioctl(fdesc, RTC_SET_TIME, &time);
		close(fdesc);
		if (ret) {
			result_string = "Error: Could not set time via /dev/rtc\n";
			http_response(req, 500, result_string, strlen(result_string));
                        return (KORE_RESULT_OK);
		}
                http_response(req, 200, NULL, 0);
                return KORE_RESULT_OK;
	} else {
		http_response(req, HTTP_STATUS_METHOD_NOT_ALLOWED, NULL, 0);
		return (KORE_RESULT_OK);
	}
}

int handle_wkalrm(struct http_request *req) {
	char *result_string;
	struct rtc_wkalrm alarm;
	int fdesc, ret;
	cJSON *alarm_json, *json_min, *json_hour, *json_mday,
                *json_mon, *json_year, *json_sec, *json_wday,
		*json_enabled, *json_pending;
	if(req->method == HTTP_METHOD_GET) {
		alarm_json = cJSON_CreateObject();
		fdesc = open("/dev/rtc", O_RDONLY);
		ret = ioctl(fdesc, RTC_WKALM_RD, &alarm);
		close(fdesc);
		if (!ret) {
			json_sec = cJSON_CreateNumber(alarm.time.tm_sec);
                        json_min = cJSON_CreateNumber(alarm.time.tm_min);
                        json_hour = cJSON_CreateNumber(alarm.time.tm_hour);
                        json_mday = cJSON_CreateNumber(alarm.time.tm_mday);
                        json_mon = cJSON_CreateNumber(alarm.time.tm_mon);
                        json_year = cJSON_CreateNumber(alarm.time.tm_year);
                        json_wday = cJSON_CreateNumber(alarm.time.tm_wday);
			json_enabled = cJSON_CreateNumber(alarm.enabled);
			json_pending = cJSON_CreateNumber(alarm.pending);
			cJSON_AddItemToObject(alarm_json, "seconds", json_sec);
                        cJSON_AddItemToObject(alarm_json, "minutes", json_min);
                        cJSON_AddItemToObject(alarm_json, "hour", json_hour);
                        cJSON_AddItemToObject(alarm_json, "mday", json_mday);
                        cJSON_AddItemToObject(alarm_json, "month", json_mon);
                        cJSON_AddItemToObject(alarm_json, "year", json_year);
                        cJSON_AddItemToObject(alarm_json, "wday", json_wday);
			cJSON_AddItemToObject(alarm_json, "enabled", json_enabled);
			cJSON_AddItemToObject(alarm_json, "pending", json_pending);
			result_string = cJSON_Print(alarm_json);
			cJSON_Delete(alarm_json);
                        http_response_header(req, "Content-Type", "application/json");
                        http_response(req, 200, result_string, strlen(result_string));
                        return KORE_RESULT_OK;
		} 
		cJSON_Delete(alarm_json);
                result_string = "Error: could not get alarm from /dev/rtc\n";
                http_response(req, 500, result_string, strlen(result_string));
                return KORE_RESULT_OK;
	} else if (req->method == HTTP_METHOD_POST) {
		char * body = read_body(req);
		if (body == NULL) {
			result_string = "Error: could not read request body\n";
			http_response(req, 500, result_string, strlen(result_string));
			return (KORE_RESULT_OK);
                }
		alarm_json = cJSON_Parse(body);
		if (alarm_json == NULL) {
                        result_string = "Error: could not parse json\n";
                        http_response(req, 500, result_string, strlen(result_string));
                        return (KORE_RESULT_OK);
                }
		json_sec = cJSON_GetObjectItemCaseSensitive(alarm_json, "seconds");
                json_min = cJSON_GetObjectItemCaseSensitive(alarm_json, "minutes");
                json_hour = cJSON_GetObjectItemCaseSensitive(alarm_json, "hour");
                json_mday = cJSON_GetObjectItemCaseSensitive(alarm_json, "mday");
                json_mon = cJSON_GetObjectItemCaseSensitive(alarm_json, "month");
                json_year = cJSON_GetObjectItemCaseSensitive(alarm_json, "year");
                json_wday = cJSON_GetObjectItemCaseSensitive(alarm_json, "wday");
		json_enabled = cJSON_GetObjectItemCaseSensitive(alarm_json, "enabled");
		if (cJSON_IsNumber(json_sec) && cJSON_IsNumber(json_min) && cJSON_IsNumber(json_hour) &&
                        cJSON_IsNumber(json_mday) && cJSON_IsNumber(json_mon) &&
                        cJSON_IsNumber(json_year) && cJSON_IsNumber(json_wday) &&
			cJSON_IsNumber(json_enabled)) {
                        alarm.time.tm_sec = json_sec->valueint;
                        alarm.time.tm_min = json_min->valueint;
                        alarm.time.tm_hour = json_hour->valueint;
                        alarm.time.tm_mday = json_mday->valueint;
                        alarm.time.tm_mon = json_mon->valueint;
                        alarm.time.tm_year = json_year->valueint;
                        alarm.time.tm_wday = json_wday->valueint;
			alarm.enabled = json_enabled->valueint;
                } else {
                        cJSON_Delete(alarm_json);
                        result_string = "Error: invalid or missing values in json object\n";
                        http_response(req, 500, result_string, strlen(result_string));
                        return (KORE_RESULT_OK);
                }
		cJSON_Delete(alarm_json);
                fdesc = open("/dev/rtc", O_WRONLY);
                ret = ioctl(fdesc, RTC_WKALM_SET, &alarm);
                close(fdesc);
                if (ret) {
                        result_string = "Error: Could not set alarm via /dev/rtc\n";
                        http_response(req, 500, result_string, strlen(result_string));
                        return (KORE_RESULT_OK);
                }
                http_response(req, 200, NULL, 0);
                return KORE_RESULT_OK;
	} else {
                http_response(req, HTTP_STATUS_METHOD_NOT_ALLOWED, NULL, 0);
                return (KORE_RESULT_OK);
	}
}