Imported Upstream version 1.24
[platform/upstream/connman.git] / tools / wispr.c
index c2f77c4..d5f9341 100644 (file)
@@ -2,7 +2,7 @@
  *
  *  Connection Manager
  *
- *  Copyright (C) 2007-2010  Intel Corporation. All rights reserved.
+ *  Copyright (C) 2007-2012  Intel Corporation. All rights reserved.
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License version 2 as
 #include <config.h>
 #endif
 
+#define _GNU_SOURCE
 #include <stdio.h>
+#include <fcntl.h>
+#include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
 #include <signal.h>
+#include <termios.h>
+#include <netdb.h>
 
 #include <gweb/gweb.h>
 
@@ -95,17 +100,21 @@ static const char *response_code_to_string(int response_code)
 }
 
 struct wispr_msg {
-       gboolean has_error;
+       bool has_error;
        const char *current_element;
        int message_type;
        int response_code;
        char *login_url;
+       char *abort_login_url;
        char *logoff_url;
+       char *access_procedure;
+       char *access_location;
+       char *location_name;
 };
 
 static inline void wispr_msg_init(struct wispr_msg *msg)
 {
-       msg->has_error = FALSE;
+       msg->has_error = false;
        msg->current_element = NULL;
 
        msg->message_type = -1;
@@ -114,8 +123,20 @@ static inline void wispr_msg_init(struct wispr_msg *msg)
        g_free(msg->login_url);
        msg->login_url = NULL;
 
+       g_free(msg->abort_login_url);
+       msg->abort_login_url = NULL;
+
        g_free(msg->logoff_url);
        msg->logoff_url = NULL;
+
+       g_free(msg->access_procedure);
+       msg->access_procedure = NULL;
+
+       g_free(msg->access_location);
+       msg->access_location = NULL;
+
+       g_free(msg->location_name);
+       msg->location_name = NULL;
 }
 
 struct wispr_session {
@@ -123,12 +144,14 @@ struct wispr_session {
        GWebParser *parser;
        guint request;
        struct wispr_msg msg;
-       const char *username;
-       const char *password;
-       const char *originurl;
+       char *username;
+       char *password;
+       char *originurl;
        char *formdata;
 };
 
+static gboolean execute_login(gpointer user_data);
+
 static struct {
        const char *str;
        enum {
@@ -189,25 +212,34 @@ static void text_handler(GMarkupParseContext *context,
        struct wispr_msg *msg = user_data;
        int i;
 
-       if (msg->current_element == NULL)
+       if (!msg->current_element)
                return;
 
        for (i = 0; wispr_element_map[i].str; i++) {
-               if (g_str_equal(wispr_element_map[i].str,
-                                       msg->current_element) == FALSE)
+               if (!g_str_equal(wispr_element_map[i].str, msg->current_element))
                        continue;
 
                switch (wispr_element_map[i].element) {
                case WISPR_ELEMENT_NONE:
                case WISPR_ELEMENT_ACCESS_PROCEDURE:
+                       g_free(msg->access_procedure);
+                       msg->access_procedure = g_strdup(text);
+                       break;
                case WISPR_ELEMENT_ACCESS_LOCATION:
+                       g_free(msg->access_location);
+                       msg->access_location = g_strdup(text);
+                       break;
                case WISPR_ELEMENT_LOCATION_NAME:
+                       g_free(msg->location_name);
+                       msg->location_name = g_strdup(text);
                        break;
                case WISPR_ELEMENT_LOGIN_URL:
                        g_free(msg->login_url);
                        msg->login_url = g_strdup(text);
                        break;
                case WISPR_ELEMENT_ABORT_LOGIN_URL:
+                       g_free(msg->abort_login_url);
+                       msg->abort_login_url = g_strdup(text);
                        break;
                case WISPR_ELEMENT_MESSAGE_TYPE:
                        msg->message_type = atoi(text);
@@ -233,7 +265,7 @@ static void error_handler(GMarkupParseContext *context,
 {
        struct wispr_msg *msg = user_data;
 
-       msg->has_error = TRUE;
+       msg->has_error = true;
 }
 
 static const GMarkupParser wispr_parser = {
@@ -248,7 +280,7 @@ static void parser_callback(const char *str, gpointer user_data)
 {
        struct wispr_session *wispr = user_data;
        GMarkupParseContext *context;
-       gboolean result;
+       bool result;
 
        //printf("%s\n", str);
 
@@ -256,29 +288,212 @@ static void parser_callback(const char *str, gpointer user_data)
                        G_MARKUP_TREAT_CDATA_AS_TEXT, &wispr->msg, NULL);
 
        result = g_markup_parse_context_parse(context, str, strlen(str), NULL);
-       if (result == TRUE)
-               result = g_markup_parse_context_end_parse(context, NULL);
+       if (result)
+               g_markup_parse_context_end_parse(context, NULL);
 
        g_markup_parse_context_free(context);
 }
 
-static gboolean wispr_input(const guint8 **data, gsize *length,
+typedef void (*user_input_cb)(const char *value, gpointer user_data);
+
+struct user_input_data {
+       GString *str;
+       user_input_cb cb;
+       gpointer user_data;
+       bool hidden;
+       int fd;
+       struct termios saved_termios;
+};
+
+static void user_callback(struct user_input_data *data)
+{
+       char *value;
+
+       if (data->hidden) {
+               ssize_t len;
+
+               len = write(data->fd, "\n", 1);
+               if (len < 0)
+                       return;
+       }
+
+       tcsetattr(data->fd, TCSADRAIN, &data->saved_termios);
+
+       close(data->fd);
+
+       value = g_string_free(data->str, FALSE);
+
+       if (data->cb)
+               data->cb(value, data->user_data);
+
+       g_free(value);
+
+       g_free(data);
+}
+
+static gboolean keyboard_input(GIOChannel *channel, GIOCondition condition,
+                                                       gpointer user_data)
+{
+       struct user_input_data *data = user_data;
+       char buf[1];
+       int len;
+
+       len = read(data->fd, buf, 1);
+
+       if (len != 1)
+               return TRUE;
+
+       if (buf[0] == '\n') {
+               user_callback(data);
+               return FALSE;
+       }
+
+       g_string_append_c(data->str, buf[0]);
+
+       if (data->hidden)
+               len = write(data->fd, "*", 1);
+
+       return TRUE;
+}
+
+static bool user_input(const char *label, bool hidden,
+                               user_input_cb func, gpointer user_data)
+{
+       struct user_input_data *data;
+       struct termios new_termios;
+       GIOChannel *channel;
+       guint watch;
+       ssize_t len;
+
+       data = g_try_new0(struct user_input_data, 1);
+       if (!data)
+               return false;
+
+       data->str = g_string_sized_new(32);
+       data->cb = func;
+       data->user_data = user_data;
+       data->hidden = hidden;
+
+       data->fd = open("/dev/tty", O_RDWR | O_NOCTTY | O_CLOEXEC);
+       if (data->fd < 0)
+               goto error;
+
+       if (tcgetattr(data->fd, &data->saved_termios) < 0) {
+               close(data->fd);
+               goto error;
+       }
+
+       new_termios = data->saved_termios;
+       if (data->hidden)
+               new_termios.c_lflag &= ~(ICANON|ECHO);
+       else
+               new_termios.c_lflag &= ~ICANON;
+       new_termios.c_cc[VMIN] = 1;
+       new_termios.c_cc[VTIME] = 0;
+
+       tcsetattr(data->fd, TCSADRAIN, &new_termios);
+
+       channel = g_io_channel_unix_new(data->fd);
+       g_io_channel_set_encoding(channel, NULL, NULL);
+       g_io_channel_set_buffered(channel, FALSE);
+       watch = g_io_add_watch(channel, G_IO_IN, keyboard_input, data);
+       g_io_channel_unref(channel);
+
+       if (watch == 0)
+               goto error;
+
+       len = write(data->fd, label, strlen(label));
+       if (len < 0)
+               goto error;
+
+       len = write(data->fd, ": ", 2);
+       if (len < 0)
+               goto error;
+
+       return true;
+
+error:
+       g_string_free(data->str, TRUE);
+       g_free(data);
+
+       return false;
+}
+
+static void password_callback(const char *value, gpointer user_data)
+{
+       struct wispr_session *wispr = user_data;
+
+       g_free(wispr->password);
+       wispr->password = g_strdup(value);
+
+       printf("\n");
+
+       execute_login(wispr);
+}
+
+static void username_callback(const char *value, gpointer user_data)
+{
+       struct wispr_session *wispr = user_data;
+
+       g_free(wispr->username);
+       wispr->username = g_strdup(value);
+
+       if (!wispr->password) {
+               user_input("Password", true, password_callback, wispr);
+               return;
+       }
+
+       printf("\n");
+
+       execute_login(wispr);
+}
+
+static bool wispr_input(const guint8 **data, gsize *length,
                                                gpointer user_data)
 {
        struct wispr_session *wispr = user_data;
+       GString *buf;
+       gsize count;
+
+       buf = g_string_sized_new(100);
+
+       g_string_append(buf, "button=Login&UserName=");
+       g_string_append_uri_escaped(buf, wispr->username, NULL, FALSE);
+       g_string_append(buf, "&Password=");
+       g_string_append_uri_escaped(buf, wispr->password, NULL, FALSE);
+       g_string_append(buf, "&FNAME=0&OriginatingServer=");
+       g_string_append_uri_escaped(buf, wispr->originurl, NULL, FALSE);
+
+       count = buf->len;
 
        g_free(wispr->formdata);
-       wispr->formdata = g_strdup_printf("button=Login&UserName=%s&"
-                       "Password=%s&FNAME=0&OriginatingServer=%s",
-                       wispr->username, wispr->password, wispr->originurl);
+       wispr->formdata = g_string_free(buf, FALSE);
 
        *data = (guint8 *) wispr->formdata;
-       *length = strlen(wispr->formdata);
+       *length = count;
 
-       return FALSE;
+       return false;
 }
 
-static gboolean wispr_result(GWebResult *result, gpointer user_data)
+static bool wispr_route(const char *addr, int ai_family, int if_index,
+               gpointer user_data)
+{
+       char *family = "unknown";
+
+       if (ai_family == AF_INET)
+               family = "IPv4";
+       else if (ai_family == AF_INET6)
+               family = "IPv6";
+
+       printf("Route request: %s %s index %d\n", family, addr, if_index);
+
+       if (ai_family != AF_INET && ai_family != AF_INET6)
+               return false;
+
+       return true;
+}
+
+static bool wispr_result(GWebResult *result, gpointer user_data)
 {
        struct wispr_session *wispr = user_data;
        const guint8 *chunk;
@@ -291,7 +506,7 @@ static gboolean wispr_result(GWebResult *result, gpointer user_data)
        if (length > 0) {
                //printf("%s\n", (char *) chunk);
                g_web_parser_feed_data(wispr->parser, chunk, length);
-               return TRUE;
+               return true;
        }
 
        g_web_parser_end_data(wispr->parser);
@@ -304,8 +519,23 @@ static gboolean wispr_result(GWebResult *result, gpointer user_data)
 
        g_print("elapse: %f seconds\n", elapsed);
 
-       if (wispr->msg.message_type < 0)
-               goto done;
+       if (wispr->msg.message_type < 0) {
+               const char *redirect;
+
+               if (status != 302)
+                       goto done;
+
+               if (!g_web_result_get_header(result, "Location", &redirect))
+                       goto done;
+
+               printf("Redirect URL: %s\n", redirect);
+               printf("\n");
+
+               wispr->request = g_web_request_get(wispr->web, redirect,
+                               wispr_result, wispr_route, wispr);
+
+               return false;
+       }
 
        printf("Message type: %s (%d)\n",
                        message_type_to_string(wispr->msg.message_type),
@@ -313,34 +543,80 @@ static gboolean wispr_result(GWebResult *result, gpointer user_data)
        printf("Response code: %s (%d)\n",
                        response_code_to_string(wispr->msg.response_code),
                                                wispr->msg.response_code);
-       if (wispr->msg.login_url != NULL)
+       if (wispr->msg.access_procedure)
+               printf("Access procedure: %s\n", wispr->msg.access_procedure);
+       if (wispr->msg.access_location)
+               printf("Access location: %s\n", wispr->msg.access_location);
+       if (wispr->msg.location_name)
+               printf("Location name: %s\n", wispr->msg.location_name);
+       if (wispr->msg.login_url)
                printf("Login URL: %s\n", wispr->msg.login_url);
-       if (wispr->msg.logoff_url != NULL)
+       if (wispr->msg.abort_login_url)
+               printf("Abort login URL: %s\n", wispr->msg.abort_login_url);
+       if (wispr->msg.logoff_url)
                printf("Logoff URL: %s\n", wispr->msg.logoff_url);
        printf("\n");
 
-       if (status == 302 && wispr->msg.message_type == 100) {
-               wispr->request = g_web_request_post(wispr->web,
-                                       wispr->msg.login_url,
-                                       "application/x-www-form-urlencoded",
-                                       wispr_input, wispr_result, wispr);
+       if (status != 200 && status != 302 && status != 404)
+               goto done;
 
-               wispr_msg_init(&wispr->msg);
+       if (wispr->msg.message_type == 100) {
+               if (!wispr->username) {
+                       user_input("Username", false,
+                                  username_callback, wispr);
+                       return false;
+               }
 
-               return FALSE;
-       } else if (status == 200 && wispr->msg.message_type == 120) {
+               if (!wispr->password) {
+                       user_input("Password", true, password_callback, wispr);
+                       return false;
+               }
+
+               g_idle_add(execute_login, wispr);
+               return false;
+       } else if (wispr->msg.message_type == 120 ||
+                                       wispr->msg.message_type == 140) {
                int code = wispr->msg.response_code;
                printf("Login process: %s\n",
                                        code == 50 ? "SUCCESS" : "FAILURE");
        }
 
+       if (status == 302) {
+               const char *redirect;
+
+               if (!g_web_result_get_header(result, "Location", &redirect))
+                       goto done;
+
+               printf("\n");
+               printf("Redirect URL: %s\n", redirect);
+               printf("\n");
+
+               wispr->request = g_web_request_get(wispr->web, redirect,
+                               wispr_result, NULL, wispr);
+
+               return false;
+       }
+
 done:
        g_main_loop_quit(main_loop);
 
+       return false;
+}
+
+static gboolean execute_login(gpointer user_data)
+{
+       struct wispr_session *wispr = user_data;
+
+       wispr->request = g_web_request_post(wispr->web, wispr->msg.login_url,
+                                       "application/x-www-form-urlencoded",
+                                       wispr_input, wispr_result, wispr);
+
+       wispr_msg_init(&wispr->msg);
+
        return FALSE;
 }
 
-static gboolean option_debug = FALSE;
+static bool option_debug = false;
 static gchar *option_nameserver = NULL;
 static gchar *option_username = NULL;
 static gchar *option_password = NULL;
@@ -371,8 +647,8 @@ int main(int argc, char *argv[])
        context = g_option_context_new(NULL);
        g_option_context_add_main_entries(context, options, NULL);
 
-       if (g_option_context_parse(context, &argc, &argv, &error) == FALSE) {
-               if (error != NULL) {
+       if (!g_option_context_parse(context, &argc, &argv, &error)) {
+               if (error) {
                        g_printerr("%s\n", error->message);
                        g_error_free(error);
                } else
@@ -386,17 +662,17 @@ int main(int argc, char *argv[])
        wispr_msg_init(&wispr.msg);
 
        wispr.web = g_web_new(index);
-       if (wispr.web == NULL) {
+       if (!wispr.web) {
                fprintf(stderr, "Failed to create web service\n");
                return 1;
        }
 
-       if (option_debug == TRUE)
+       if (option_debug)
                g_web_set_debug(wispr.web, web_debug, "WEB");
 
        main_loop = g_main_loop_new(NULL, FALSE);
 
-       if (option_nameserver != NULL) {
+       if (option_nameserver) {
                g_web_add_nameserver(wispr.web, option_nameserver);
                g_free(option_nameserver);
        }
@@ -405,7 +681,7 @@ int main(int argc, char *argv[])
        g_web_set_user_agent(wispr.web, "SmartClient/%s wispr", VERSION);
        g_web_set_close_connection(wispr.web, TRUE);
 
-       if (option_url == NULL)
+       if (!option_url)
                option_url = g_strdup(DEFAULT_URL);
 
        wispr.username = option_username;
@@ -419,7 +695,7 @@ int main(int argc, char *argv[])
                                                parser_callback, &wispr);
 
        wispr.request = g_web_request_get(wispr.web, option_url,
-                                                       wispr_result, &wispr);
+                       wispr_result, wispr_route, &wispr);
 
        if (wispr.request == 0) {
                fprintf(stderr, "Failed to start request\n");
@@ -443,9 +719,9 @@ int main(int argc, char *argv[])
 
        g_main_loop_unref(main_loop);
 
-       g_free(option_username);
-       g_free(option_password);
-       g_free(option_url);
+       g_free(wispr.username);
+       g_free(wispr.password);
+       g_free(wispr.originurl);
 
        return 0;
 }