source: azure_iot_hub_riscv/trunk/app_iothub_client/src/envcmd.c@ 458

/*
 *  TOPPERS/ASP Kernel
 *      Toyohashi Open Platform for Embedded Real-Time Systems/
 *      Advanced Standard Profile Kernel
 * 
 *  Copyright (C) 2000-2003 by Embedded and Real-Time Systems Laboratory
 *                              Toyohashi Univ. of Technology, JAPAN
 *  Copyright (C) 2004-2012 by Embedded and Real-Time Systems Laboratory
 *              Graduate School of Information Science, Nagoya Univ., JAPAN
 * 
 *  上記著作権者は，以下の(1)～(4)の条件を満たす場合に限り，本ソフトウェ
 *  ア（本ソフトウェアを改変したものを含む．以下同じ）を使用・複製・改
 *  変・再配布（以下，利用と呼ぶ）することを無償で許諾する．
 *  (1) 本ソフトウェアをソースコードの形で利用する場合には，上記の著作
 *      権表示，この利用条件および下記の無保証規定が，そのままの形でソー
 *      スコード中に含まれていること．
 *  (2) 本ソフトウェアを，ライブラリ形式など，他のソフトウェア開発に使
 *      用できる形で再配布する場合には，再配布に伴うドキュメント（利用
 *      者マニュアルなど）に，上記の著作権表示，この利用条件および下記
 *      の無保証規定を掲載すること．
 *  (3) 本ソフトウェアを，機器に組み込むなど，他のソフトウェア開発に使
 *      用できない形で再配布する場合には，次のいずれかの条件を満たすこ
 *      と．
 *    (a) 再配布に伴うドキュメント（利用者マニュアルなど）に，上記の著
 *        作権表示，この利用条件および下記の無保証規定を掲載すること．
 *    (b) 再配布の形態を，別に定める方法によって，TOPPERSプロジェクトに
 *        報告すること．
 *  (4) 本ソフトウェアの利用により直接的または間接的に生じるいかなる損
 *      害からも，上記著作権者およびTOPPERSプロジェクトを免責すること．
 *      また，本ソフトウェアのユーザまたはエンドユーザからのいかなる理
 *      由に基づく請求からも，上記著作権者およびTOPPERSプロジェクトを
 *      免責すること．
 * 
 *  本ソフトウェアは，無保証で提供されているものである．上記著作権者お
 *  よびTOPPERSプロジェクトは，本ソフトウェアに関して，特定の使用目的
 *  に対する適合性も含めて，いかなる保証も行わない．また，本ソフトウェ
 *  アの利用により直接的または間接的に生じたいかなる損害に関しても，そ
 *  の責任を負わない．
 * 
 *  $Id$
 */
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include "azure_c_shared_utility/shared_util_options.h"

extern char* connectionString;

extern bool g_use_proxy;
extern HTTP_PROXY_OPTIONS g_proxy_options;

enum proxy_parse_state_t {
        proxy_parse_state_schema,
        proxy_parse_state_colonslash,
        proxy_parse_state_colonslashslash,
        proxy_parse_state_username,
        proxy_parse_state_password,
        proxy_parse_state_address,
        proxy_parse_state_port,
        proxy_parse_state_error,
};

int set_proxy(const char *proxy, bool silent)
{
        const char *schema = NULL, *address = NULL, *port = NULL, *username = NULL, *password = NULL;
        int schema_len = 0, address_len = 0, port_len = 0, username_len = 0, password_len = 0;

        enum proxy_parse_state_t state = proxy_parse_state_schema;

        const char *pos = proxy;
        schema = pos;
        for (char c = *pos; c != '\0'; pos++, c = *pos) {
                switch (state) {
                case proxy_parse_state_schema:
                        if (c == ':') {
                                schema_len = (int)pos - (int)schema;
                                username = pos + 3;
                                state = proxy_parse_state_colonslash;
                        }
                        break;
                case proxy_parse_state_colonslash:
                        if (c == '/') {
                                state = proxy_parse_state_colonslashslash;
                        }
                        else {
                                username = schema;
                                username_len = schema_len;
                                schema = NULL;
                                schema_len = 0;
                                password = pos;
                                state = proxy_parse_state_password;
                                goto case_proxy_parse_state_password;
                        }
                        break;
                case proxy_parse_state_colonslashslash:
                        if (c == '/') {
                                state = proxy_parse_state_username;
                        }
                        else {
                                state = proxy_parse_state_error;
                        }
                        break;
                case proxy_parse_state_username:
                        if (c == ':') {
                                username_len = (int)pos - (int)username;
                                password = pos + 1;
                                state = proxy_parse_state_password;
                        }
                        else if (c == '@') {
                                username_len = (int)pos - (int)username;
                                address = pos + 1;
                                state = proxy_parse_state_address;
                        }
                        break;
                case proxy_parse_state_password:
                case_proxy_parse_state_password:
                        if (c == '@') {
                                password_len = (int)pos - (int)password;
                                address = pos + 1;
                                state = proxy_parse_state_address;
                        }
                        break;
                case proxy_parse_state_address:
                        if (c == ':') {
                                address_len = (int)pos - (int)address;
                                port = pos + 1;
                                state = proxy_parse_state_port;
                        }
                        break;
                case proxy_parse_state_port:
                        if ((c < '0') && (c > '9')) {
                                state = proxy_parse_state_error;
                        }
                        break;
                }

                if (state == proxy_parse_state_error)
                        break;
        }

        switch (state)
        {
        case proxy_parse_state_schema:
        case proxy_parse_state_colonslash:
        case proxy_parse_state_colonslashslash:
                goto error;
        case proxy_parse_state_username:
                username_len = (int)pos - (int)username;
                address = username;
                address_len = username_len;
                username = NULL;
                username_len = 0;
                break;
        case proxy_parse_state_password:
                password_len = (int)pos - (int)password;
                address = username;
                address_len = username_len;
                username = NULL;
                username_len = 0;
                port = password;
                port_len = password_len;
                password = NULL;
                password_len = 0;
                break;
        case proxy_parse_state_address:
                address_len = (int)pos - (int)address;
                break;
        case proxy_parse_state_port:
                port_len = (int)pos - (int)port;
                break;
        case proxy_parse_state_error:
        default:
                goto error;
        }

        if (schema_len > 0) {
                if (strncmp(schema, "http", schema_len) != 0)
                        goto error;
        }

        if (address_len == 0)
                goto error;

        char *buf = malloc(address_len + port_len + username_len + password_len + 4);
        if (buf == NULL) {
                if (!silent)
                        printf("Memory error.\n");
                return -1;
        }

        free((char *)g_proxy_options.host_address);

        memcpy(buf, address, address_len);
        buf[address_len] = '\0';
        g_proxy_options.host_address = buf;

        buf = buf + address_len + 1;
        memcpy(buf, port, port_len);
        buf[port_len] = '\0';
        g_proxy_options.port = port_len == 0 ? 8080 : atoi(buf);

        buf = buf + port_len + 1;
        memcpy(buf, username, username_len);
        buf[username_len] = '\0';
        g_proxy_options.username = (username_len == 0) && (password_len == 0) ? NULL : buf;

        buf = buf + username_len + 1;
        memcpy(buf, password, password_len);
        buf[password_len] = '\0';
        g_proxy_options.password = (username_len == 0) && (password_len == 0) ? NULL : buf;

        return 0;
error:
        if (!silent)
                printf("proxy string parse error. 'http://[<user>:<password>@]<address>:<port>'\n");
        return -1;
}

int set_cs_main(int argc, char **argv)
{
        if (argc < 2) {
                printf("usage:\n%s <connection string>\n", argv[0]);
                return 0;
        }

        int len = strlen(argv[1]);
        if (len < 70) {
                printf("String containing Hostname, Device Id & Device Key in the format:\n");
                printf("  HostName=<host_name>;DeviceId=<device_id>;SharedAccessKey=<device_key>\n");
                printf("  HostName=<host_name>;DeviceId=<device_id>;SharedAccessSignature=<device_sas_token>\n");
                return 0;
        }

        char *conn_str = (char *)malloc(len + 1);
        if (conn_str == NULL) {
                printf("OOM while creating connection string\n");
                return 1;
        }

        memcpy(conn_str, argv[1], len);
        conn_str[len] = 0;
        free(connectionString);
        connectionString = conn_str;
        printf("Connection String:\n%s\n", connectionString);

        return 0;
}

int set_proxy_main(int argc, char **argv)
{
        int result;

        if (argc < 2) {
                if (g_proxy_options.username != NULL) {
                        printf("HTTP_PROXY=http://%s:%s@%s:%d\n",
                                g_proxy_options.username,
                                g_proxy_options.password,
                                g_proxy_options.host_address,
                                g_proxy_options.port);
                }
                else if (g_proxy_options.host_address != NULL) {
                        printf("HTTP_PROXY=http://%s:%d\n",
                                g_proxy_options.host_address,
                                g_proxy_options.port);
                }
                else {
                        printf("HTTP_PROXY=\n");
                }
                return 0;
        }

        if ((result = set_proxy(argv[1], false)) == 0) {
                g_use_proxy = true;
        }
        else {
                g_use_proxy = false;
        }

        return 0;
}

int clear_proxy_main(int argc, char **argv)
{
        free((char *)g_proxy_options.host_address);
        g_proxy_options.host_address = NULL;
        g_proxy_options.port = 0;
        g_proxy_options.username = NULL;
        g_proxy_options.password = NULL;

        g_use_proxy = false;
        return 0;
}

int set_wifi_main(int argc, char **argv)
{
        if (argc < 3) {
                return 0;
        }

        set_ssid_pwd(argv[1], argv[2]);

        prepare_esp_at();
}