*
* 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 <string.h>
#include <signal.h>
#include <termios.h>
+#include <netdb.h>
#include <gweb/gweb.h>
}
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;
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;
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;
char *formdata;
};
-static void execute_login(struct wispr_session *wispr);
+static gboolean execute_login(gpointer user_data);
static struct {
const char *str;
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_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);
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);
{
struct wispr_msg *msg = user_data;
- msg->has_error = TRUE;
+ msg->has_error = true;
}
static const GMarkupParser wispr_parser = {
{
struct wispr_session *wispr = user_data;
GMarkupParseContext *context;
- gboolean result;
+ bool result;
//printf("%s\n", str);
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);
}
GString *str;
user_input_cb cb;
gpointer user_data;
- gboolean hidden;
+ bool hidden;
int fd;
struct termios saved_termios;
};
static void user_callback(struct user_input_data *data)
{
char *value;
- int len;
- if (data->hidden == TRUE)
+ if (data->hidden) {
+ ssize_t len;
+
len = write(data->fd, "\n", 1);
+ if (len < 0)
+ return;
+ }
tcsetattr(data->fd, TCSADRAIN, &data->saved_termios);
g_string_append_c(data->str, buf[0]);
- if (data->hidden == TRUE)
+ if (data->hidden)
len = write(data->fd, "*", 1);
return TRUE;
}
-static gboolean user_input(const char *label, gboolean hidden,
+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;
- int len;
+ ssize_t len;
data = g_try_new0(struct user_input_data, 1);
- if (data == NULL)
- return FALSE;
+ 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);
+ data->fd = open("/dev/tty", O_RDWR | O_NOCTTY | O_CLOEXEC);
if (data->fd < 0)
goto error;
}
new_termios = data->saved_termios;
- if (data->hidden == TRUE)
+ if (data->hidden)
new_termios.c_lflag &= ~(ICANON|ECHO);
else
new_termios.c_lflag &= ~ICANON;
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;
+ return true;
error:
g_string_free(data->str, TRUE);
g_free(data);
- return FALSE;
+ return false;
}
static void password_callback(const char *value, gpointer user_data)
g_free(wispr->username);
wispr->username = g_strdup(value);
- if (wispr->password == NULL) {
- user_input("Password", TRUE, password_callback, wispr);
+ if (!wispr->password) {
+ user_input("Password", true, password_callback, wispr);
return;
}
execute_login(wispr);
}
-static gboolean wispr_input(const guint8 **data, gsize *length,
+static bool wispr_input(const guint8 **data, gsize *length,
gpointer user_data)
{
struct wispr_session *wispr = user_data;
*data = (guint8 *) 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;
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);
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),
printf("Response code: %s (%d)\n",
response_code_to_string(wispr->msg.response_code),
wispr->msg.response_code);
- if (wispr->msg.access_procedure != NULL)
+ if (wispr->msg.access_procedure)
printf("Access procedure: %s\n", wispr->msg.access_procedure);
- if (wispr->msg.access_location != NULL)
+ if (wispr->msg.access_location)
printf("Access location: %s\n", wispr->msg.access_location);
- if (wispr->msg.location_name != NULL)
+ if (wispr->msg.location_name)
printf("Location name: %s\n", wispr->msg.location_name);
- if (wispr->msg.login_url != NULL)
+ 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) {
- if (wispr->username == NULL) {
- user_input("Username", FALSE, username_callback, wispr);
- return FALSE;
+ if (status != 200 && status != 302 && status != 404)
+ goto done;
+
+ if (wispr->msg.message_type == 100) {
+ if (!wispr->username) {
+ user_input("Username", false,
+ username_callback, wispr);
+ return false;
}
- if (wispr->password == NULL) {
- user_input("Password", TRUE, password_callback, wispr);
- return FALSE;
+ if (!wispr->password) {
+ user_input("Password", true, password_callback, wispr);
+ return false;
}
- execute_login(wispr);
- return FALSE;
- } else if (status == 200 && wispr->msg.message_type == 120) {
+ 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;
+ return false;
}
-static void execute_login(struct wispr_session *wispr)
+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;
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
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);
}
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;
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");