tplinkController/hs100.c

#include <stdio.h>
#include <inttypes.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include "version.h"
#include "comms.h"

char *handler_associate(int argc, char *argv[])
{
	if(argc < 6) {
		fprintf(stderr, "not enough arguments\n");
		exit(1);
	}
	char *plug_addr = argv[1];
	char *ssid = argv[3];
	char *password = argv[4];
	char *key_type = argv[5];

	errno = 0;
	char *endptr;
	int key_type_num = (int)strtol(key_type, &endptr, 10);
	if(errno || endptr == key_type) {
		fprintf(stderr, "invalid key type: %s\n", key_type);
		exit(1);
	}

	const char *template =
		"{\"netif\":{\"set_stainfo\":{\"ssid\":\"%s\",\"password\":"
		"\"%s\",\"key_type\":%d}}}";

	size_t len = snprintf(NULL, 0, template, ssid, password,
				key_type_num);
	len++;	// snprintf does not count the null terminator

	char *msg = calloc(1, len);
	snprintf(msg, len, template, ssid, password, key_type_num);

	char *response = hs100_send(plug_addr, msg);

	return response;
}

char *handler_set_server(int argc, char *argv[])
{
	if(argc < 4) {
		fprintf(stderr, "not enough arguments\n");
		exit(1);
	}
	char *plug_addr = argv[1];
	char *server = argv[3];

	const char *template =
		"{\"cnCloud\":{\"set_server_url\":{\"server\":\"%s\"}}}";

	size_t len = snprintf(NULL, 0, template, server);
	len++;	// snprintf does not count the null terminator

	char *msg = calloc(1, len);
	snprintf(msg, len, template, server);

	char *response = hs100_send(plug_addr, msg);

	return response;
}


struct cmd_s {
	char *command;
	char *help;
	char *json;
	char *(*handler)(int argc, char *argv[]);
	int end;
};
struct cmd_s cmds[] = {
	{
		.command = "associate",
		.help = "associate <ssid> <key> <key_type>\n"
			"\t\t\tset wifi AP to connect to",
		.handler = handler_associate,
	},
	{
		.command = "factory-reset",
		.help = "factory-reset\treset the plug to factory settings",
		.json = "{\"system\":{\"reset\":{\"delay\":0}}}",
	},
	{
		.command = "info",
		.help = "info\t\tget device info",
		.json = "{\"system\":{\"get_sysinfo\":{}}}",
	},
	{
		.command = "off",
		.help = "off\t\tturn the plug on",
		.json = "{\"system\":{\"set_relay_state\":{\"state\":0}}}",
	},
	{
		.command = "on",
		.help = "on\t\tturn the plug off",
		.json = "{\"system\":{\"set_relay_state\":{\"state\":1}}}",
	},
	{
		.command = "reboot",
		.help = "reboot\t\treboot the plug",
		.json = "{\"system\":{\"reboot\":{\"delay\":0}}}",
	},
	{
		.command = "scan",
		.help = "scan\t\tscan for nearby wifi APs (probably only 2.4"
			" GHz ones)",
		.json = "{\"netif\":{\"get_scaninfo\":{\"refresh\":1}}}",
	},
	{
		.command = "set_server",
		.help = "set_server <url>\n"
			"\t\t\tset cloud server to <url> instead of tplink's",
		.handler = handler_set_server,
	},
	{
		.end = 1,
	},
};

struct cmd_s *get_cmd_from_name(char *needle)
{
	int cmds_index = 0;
	while(!cmds[cmds_index].end)
	{
		if(!strcmp(cmds[cmds_index].command, needle))
			return &cmds[cmds_index];
		cmds_index++;
	}
	return NULL;
}

void print_usage()
{
	fprintf(stderr, "hs100 version " VERSION_STRING ", Copyright (C) 2018 Jason Benaim.\n"
			"A tool for using certain wifi smart plugs.\n"
			"\n"
			"usage: hs100 <ip> <command>\n"
			"\n"
			"Commands:\n"
	);
	int cmds_index = 0;
	while(!cmds[cmds_index].end)
	{
		fprintf(stderr, "\t%s\n\n", cmds[cmds_index].help);
		cmds_index++;
	}
	fprintf(stderr, "Report bugs to https://github.com/jkbenaim/hs100\n");
}

int main(int argc, char *argv[])
{
	if(argc < 3) {
		print_usage();
		return 1;
	}
	char *plug_addr = argv[1];
	char *cmd_string = argv[2];
	char *response = NULL;

	struct cmd_s *cmd = get_cmd_from_name(cmd_string);
	if(cmd != NULL) {
		if(cmd->handler != NULL)
			response = cmd->handler(argc, argv);
		else if(cmd->json != NULL)
			response = hs100_send(plug_addr, cmd->json);
	} else {
		// command not recognized, so send it to the plug raw
		response = hs100_send(plug_addr, cmd_string);
	}
	
	if(response == NULL) {
		fprintf(stderr, "failed to send command\n");
		return 1;
	}

	printf("%s\n", response);

	free(response);
	return 0;
}