flash.c

/*
  Copyright (C) CurlyMo

  This Source Code Form is subject to the terms of the Mozilla Public
  License, v. 2.0. If a copy of the MPL was not distributed with this
  file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <unistd.h>
#include <fcntl.h>
#include <limits.h>
#include <errno.h>
#include <time.h>
#include <math.h>
#include <string.h>
#ifndef _WIN32
#include <wiringx.h>
#endif
#include <assert.h>

#include "libs/pilight/core/threads.h"
#include "libs/pilight/core/pilight.h"
#include "libs/pilight/core/gc.h"
#include "libs/pilight/core/common.h"
#include "libs/pilight/core/log.h"
#include "libs/pilight/core/options.h"
#include "libs/pilight/core/firmware.h"
#include "libs/pilight/lua_c/lua.h"

#include "libs/pilight/events/events.h"

static char *lua_root = LUA_ROOT;

int main(int argc, char **argv) {
	// memtrack();

	atomicinit();
	log_shell_enable();
	log_file_disable();

	struct options_t *options = NULL;
	char *configtmp = CONFIG_FILE;
	char *fwfile = NULL;
	char *comport = NULL;
	int help = 0;

	if((progname = MALLOC(15)) == NULL) {
		fprintf(stderr, "out of memory\n");
		exit(EXIT_FAILURE);
	}
	strcpy(progname, "pilight-flash");

	options_add(&options, "H", "help", OPTION_NO_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "V", "version", OPTION_NO_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "C", "config", OPTION_HAS_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "f", "file", OPTION_HAS_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "p", "comport", OPTION_HAS_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "Ls", "storage-root", OPTION_HAS_VALUE, 0, JSON_NULL, NULL, NULL);
	options_add(&options, "Ll", "lua-root", OPTION_HAS_VALUE, 0, JSON_NULL, NULL, NULL);

	if(argc == 1) {
		printf("Usage: %s -f pilight_firmware_tX5_v3.hex\n", progname);
		goto close;
	}

	if(options_parse(options, argc, argv, 1) == -1) {
		help = 1;
	}

	if(options_exists(options, "H") == 0 || help == 1) {
		printf("Usage: %s [options]\n", progname);
		printf("\t -H --help\t\tdisplay usage summary\n");
		printf("\t -V --version\t\tdisplay version\n");
		printf("\t -C --config\t\tconfig file\n");
		printf("\t -p --comport\t\tserial COM port\n");
		printf("\t -f --file=firmware\tfirmware file\n");
		printf("\t -Ls --storage-root=xxxx\tlocation of storage lua modules\n");
		printf("\t -Ll --lua-root=xxxx\t\tlocation of the plain lua modules\n");
		goto close;
	}

	if(options_exists(options, "V") == 0) {
		printf("%s v%s\n", progname, PILIGHT_VERSION);
		goto close;
	}

	if(options_exists(options, "C") == 0) {
		options_get_string(options, "C", &configtmp);
	}

	if(options_exists(options, "Ls") == 0) {
		char *arg = NULL;
		options_get_string(options, "Ls", &arg);
		if(config_root(arg) == -1) {
			logprintf(LOG_ERR, "%s is not valid storage lua modules path", arg);
			goto close;
		}
	}

	if(options_exists(options, "Ll") == 0) {
		options_get_string(options, "Ll", &lua_root);
	}

	{
		int len = strlen(lua_root)+strlen("lua/?/?.lua")+1;
		char *lua_path = MALLOC(len);

		if(lua_path == NULL) {
			OUT_OF_MEMORY
		}

		plua_init();

		memset(lua_path, '\0', len);
		snprintf(lua_path, len, "%s/?/?.lua", lua_root);
		plua_package_path(lua_path);

		memset(lua_path, '\0', len);
		snprintf(lua_path, len, "%s/?.lua", lua_root);
		plua_package_path(lua_path);

		FREE(lua_path);
	}

	if(options_exists(options, "p") == 0) {
		options_get_string(options, "p", &comport);
	}

	if(options_exists(options, "f") == 0) {
		options_get_string(options, "f", &fwfile);
		if(access(fwfile, F_OK) == -1) {
			fprintf(stderr, "%s: the firmware file %s does not exist\n", progname, fwfile);
			goto close;
		}
	}

	if(config_set_file(configtmp) == EXIT_FAILURE) {
		goto close;
	}

	protocol_init();
	config_init();

	struct lua_state_t *state = plua_get_free_state();
	if(config_read(state->L, CONFIG_ALL) != EXIT_SUCCESS) {
		assert(plua_check_stack(state->L, 0) == 0);
		plua_clear_state(state);
		goto close;
	}
	assert(plua_check_stack(state->L, 0) == 0);
	plua_clear_state(state);

#ifdef _WIN32
	if(fwfile == NULL || strlen(fwfile) == 0 || strlen(comport) == 0) {
		printf("Usage: %s -f pilight_firmware_XXX_vX.hex -p comX\n", progname);
		goto close;
	}
#else
	if(fwfile == NULL || strlen(fwfile) == 0) {
		printf("Usage: %s -f pilight_firmware_XXX_vX.hex\n", progname);
		goto close;
	}
#endif

	log_level_set(LOG_DEBUG);
	firmware.version = 0;
	logprintf(LOG_INFO, "**** START UPD. FW ****");
	firmware_getmp(comport);

	if(firmware_update(fwfile, comport) != 0) {
		logprintf(LOG_INFO, "**** FAILED UPD. FW ****");
	} else {
		logprintf(LOG_INFO, "**** DONE UPD. FW ****");
	}

close:
	log_shell_disable();
	options_delete(options);

	log_shell_disable();
	log_level_set(LOG_ERR);
	config_gc();
	protocol_gc();
	options_gc();
#ifdef EVENTS
	events_gc();
#endif
#ifndef _WIN32
	wiringXGC();
#endif
	log_gc();
	threads_gc();
	gc_clear();
	FREE(progname);
	xfree();
	return (EXIT_SUCCESS);
}