icwmp/cwmp_json.c

/*
 *	This program is free software: you can redistribute it and/or modify
 *	it under the terms of the GNU General Public License as published by
 *	the Free Software Foundation, either version 2 of the License, or
 *	(at your option) any later version.
 *
 *	Copyright (C) 2013-2019 iopsys Software Solutions AB
 *	  Author Mohamed Kallel <mohamed.kallel@pivasoftware.com>
 *	  Author Ahmed Zribi <ahmed.zribi@pivasoftware.com>
 *	Copyright (C) 2012 Luka Perkov <freecwmp@lukaperkov.net>
 */

#include <unistd.h>
#include <libubus.h>

#include <json-c/json.h>

#include "external.h"
#include "cwmp_json.h"
#include "log.h"

static json_object *jshn_obj = NULL;

static int cwmp_json_message_parse(char **policy, int size, char **tb, char *msg)
{
	int i;
	json_object *obj;

	jshn_obj = json_tokener_parse(msg);
	if (jshn_obj == NULL || json_object_get_type(jshn_obj) != json_type_object)
	{
		jshn_obj = NULL;
		return -1;
	}
	for (i=0; i<size; i++)
	{
		json_object_object_get_ex(jshn_obj, policy[i], &obj);
		if (obj == NULL || json_object_get_type(obj) != json_type_string)
			continue;
		tb[i] = (char *)json_object_get_string(obj);
	}
	return 0;
}

static inline void cwmp_json_message_delete()
{
	if(jshn_obj != NULL)
		json_object_put(jshn_obj);
}
enum download_fault {
	DOWNLOAD_FAULT,
	__DOWNLOAD_MAX
};

char *download_fault_policy[] = {
	[DOWNLOAD_FAULT] = "fault_code"
};

int
cwmp_handle_downloadFault(char *msg)
{
	char *tb[__DOWNLOAD_MAX] = {0};

	cwmp_json_message_parse(download_fault_policy, ARRAYSIZEOF(download_fault_policy), tb, msg);

	if (!tb[DOWNLOAD_FAULT])
		goto error;

	DD(INFO,"triggered handle download fault %s", tb[DOWNLOAD_FAULT]);

	external_downloadFaultResp (tb[DOWNLOAD_FAULT]);

	cwmp_json_message_delete();
	return 0;

error:
	cwmp_json_message_delete();
	return -1;
}

enum upload_fault {
	UPLOAD_FAULT,
	__UPLOAD_MAX
};

char *upload_fault_policy[] = {
	[UPLOAD_FAULT] = "fault_code"
};

int cwmp_handle_uploadFault(char *msg)
{
	char *tb[__UPLOAD_MAX] = {0};

	cwmp_json_message_parse(upload_fault_policy, ARRAYSIZEOF(upload_fault_policy), tb, msg);

	if (!tb[UPLOAD_FAULT])
		goto error;

	DD(INFO,"triggered handle upload fault %s", tb[UPLOAD_FAULT]);

	external_uploadFaultResp (tb[UPLOAD_FAULT]);

	cwmp_json_message_delete();
	return 0;

error:
	cwmp_json_message_delete();
	return -1;
}

enum dustatechange_fault {
	DUState_Change_FAULT,
	DUState_Change_VERSION,
	DUState_Change_NAME,
	DUState_Change_UUID,
	DUState_Change_ENV,
	__DUSTATE_MAX
};

char *dustatechange_fault_policy[] = {
	[DUState_Change_FAULT] 		= "fault_code",
	[DUState_Change_VERSION] 	= "package_version",
	[DUState_Change_NAME] 		= "package_name",
	[DUState_Change_UUID] 		= "package_uuid",
	[DUState_Change_ENV] 		= "package_env"
};

int cwmp_handle_dustate_changeFault(char *msg)
{
	char *tb[__DUSTATE_MAX] = {0};

	cwmp_json_message_parse(dustatechange_fault_policy, ARRAYSIZEOF(dustatechange_fault_policy), tb, msg);

	if (!tb[DUState_Change_FAULT])
		goto error;

	DD(INFO,"triggered handle dustate_change fault:%s version:%s name:%s", tb[DUState_Change_FAULT], tb[DUState_Change_VERSION], tb[DUState_Change_NAME], tb[DUState_Change_UUID], tb[DUState_Change_ENV]);

	external_du_change_stateFaultResp (tb[DUState_Change_FAULT], tb[DUState_Change_VERSION], tb[DUState_Change_NAME], tb[DUState_Change_UUID], tb[DUState_Change_ENV]);

	cwmp_json_message_delete();
	return 0;

error:
	cwmp_json_message_delete();
	return -1;
}

enum uninstall_fault {
	UNINSTALL_FAULT,
	__UNINSTALL_MAX
};

char *uninstall_fault_policy[] = {
	[UNINSTALL_FAULT] = "fault_code"
};

int cwmp_handle_uninstallFault(char *msg)
{
	char *tb[__UNINSTALL_MAX] = {0};

	cwmp_json_message_parse(uninstall_fault_policy, ARRAYSIZEOF(uninstall_fault_policy), tb, msg);

	if (!tb[UNINSTALL_FAULT])
		goto error;

	DD(INFO,"triggered handle upload fault %s", tb[UNINSTALL_FAULT]);

	external_uninstallFaultResp (tb[UNINSTALL_FAULT]);

	cwmp_json_message_delete();
	return 0;

error:
	cwmp_json_message_delete();
	return -1;
}

void cwmp_json_obj_init(char *str, json_object **obj)
{
	if (*obj) {
		json_object_put(*obj);
		*obj = NULL;
	}
	*obj = json_tokener_parse(str);
}

void cwmp_json_obj_clean(json_object **obj)
{
	if (*obj) {
		json_object_put(*obj);
		*obj = NULL;
	}
}
void cwmp_add_json_obj(json_object *json_obj_out, char *object, char *string)
{
	if (object != NULL && string != NULL) {
		json_object *json_obj_tmp = json_object_new_string(string);
		json_object_object_add(json_obj_out, object, json_obj_tmp);
	}
}

void cwmp_json_fprintf(FILE *fp, int argc, struct cwmp_json_arg cwmp_arg[])
{
	int i;
	char *arg;
	json_object *json_obj_out = json_object_new_object();
	if (json_obj_out == NULL)
		return;

	if (argc) {
		for (i = 0; i < argc; i++) {
			cwmp_add_json_obj(json_obj_out, cwmp_arg[i].key, cwmp_arg[i].val);
		}
		arg = (char *)json_object_to_json_string(json_obj_out);
		fprintf(fp, "%s\n", arg);
	}

	json_object_put(json_obj_out);
}

void cwmp_json_get_string(json_object *obj, char* key, char** value)
{
	json_object *key_obj = NULL;
	json_object_object_get_ex(obj, key, &key_obj);
	*value = strdup(key_obj?(char*)json_object_get_string(key_obj):"");
	if (key_obj) {
		json_object_put(key_obj);
		key_obj = NULL;
	}
}