c483837eff8a42d871f7be5e08f9a5c90cbd2f72
[dhcpcd-ui] / wpa.c
1 /*
2  * dhcpcd-gtk
3  * Copyright 2009 Roy Marples <roy@marples.name>
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26
27 #include "dhcpcd-gtk.h"
28 #include "wpa.h"
29
30 static gint
31 find_network(const char *ifname, const char *ssid)
32 {
33         GType otype;
34         GError *error;
35         gint id;
36         size_t i;
37         GPtrArray *array;
38         GValueArray *varray;
39         GValue *val;
40         const char *str;
41         char *nssid;
42
43         otype = dbus_g_type_get_struct("GValueArray",
44                         G_TYPE_INT,
45                         G_TYPE_STRING,
46                         G_TYPE_STRING,
47                         G_TYPE_STRING,
48                         G_TYPE_INVALID);
49         otype = dbus_g_type_get_collection("GPtrArray", otype);
50         error = NULL;
51         if (!dbus_g_proxy_call(dbus, "ListNetworks", &error,
52                                G_TYPE_STRING, ifname, G_TYPE_INVALID,
53                                otype, &array, G_TYPE_INVALID))
54         {
55                 g_warning("ListNetworks: %s", error->message);
56                 g_error_free(error);
57                 return -1;
58         }
59
60         for (i = 0; i < array->len; i++) {
61                 varray = g_ptr_array_index(array, i);
62                 val = g_value_array_get_nth(varray, 1);
63                 str = g_value_get_string(val);
64                 if (g_strcmp0(str, ssid) == 0) {
65                         val = g_value_array_get_nth(varray, 0);
66                         return g_value_get_int(val);
67                 }
68         }
69
70         if (!dbus_g_proxy_call(dbus, "AddNetwork", &error,
71                                G_TYPE_STRING, ifname, G_TYPE_INVALID,
72                                G_TYPE_INT, &id, G_TYPE_INVALID))
73         {
74                 g_warning("AddNetwork: %s", error->message);
75                 g_error_free(error);
76                 return -1;
77         }
78
79         nssid = g_strconcat("\"", ssid, "\"", NULL);
80         if (!dbus_g_proxy_call(dbus, "SetNetwork", &error,
81                                G_TYPE_STRING, ifname,
82                                G_TYPE_INT, id,
83                                G_TYPE_STRING, "ssid",
84                                G_TYPE_STRING, nssid,
85                                G_TYPE_INVALID,
86                                G_TYPE_INVALID))
87         {
88                 g_warning("SetNetwork: %s", error->message);
89                 g_free(nssid);
90                 g_error_free(error);
91                 return -1;
92         }
93         g_free(nssid);
94
95         return id;
96 }
97
98 static int
99 configure_network(const char *ifname, int id, const char *var, const char *val,
100                 gboolean quote)
101 {
102         GError *error;
103         char *str;
104         static gboolean warned = FALSE;
105         GtkWidget *dialog;
106
107         if (id == -1)
108                 return -1;
109
110         error = NULL;
111         if (quote)
112                 str = g_strconcat("\"", val, "\"", NULL);
113         else
114                 str = NULL;
115         if (!dbus_g_proxy_call(dbus, "SetNetwork", &error,
116                                G_TYPE_STRING, ifname,
117                                G_TYPE_INT, id,
118                                G_TYPE_STRING, var,
119                                G_TYPE_STRING, quote ? str : val,
120                                G_TYPE_INVALID,
121                                G_TYPE_INVALID))
122         {
123                 g_warning("SetNetwork: %s", error->message);
124                 g_free(str);
125                 g_error_free(error);
126                 return -1;
127         }
128         g_free(str);
129
130         if (!dbus_g_proxy_call(dbus, "EnableNetwork", &error,
131                                G_TYPE_STRING, ifname,
132                                G_TYPE_INT, id,
133                                G_TYPE_INVALID,
134                                G_TYPE_INVALID))
135         {
136                 g_warning("EnableNetwork: %s", error->message);
137                 g_error_free(error);
138                 return -1;
139         }
140
141         if (!dbus_g_proxy_call(dbus, "SaveConfig", &error,
142                                G_TYPE_STRING, ifname,
143                                G_TYPE_INVALID,
144                                G_TYPE_INVALID))
145         {
146                 g_warning("SaveConfig: %s", error->message);
147                 if (!warned) {
148                         warned = TRUE;
149                         dialog = gtk_message_dialog_new(NULL,
150                                                         GTK_DIALOG_MODAL,
151                                                         GTK_MESSAGE_ERROR,
152                                                         GTK_BUTTONS_OK,
153                                                         _("Failed to save wpa_supplicant configuration.\n\nYou should add update_config=1 to /etc/wpa_supplicant.conf.\nThis warning will not appear again until program restarted."));
154                         gtk_window_set_title(GTK_WINDOW(dialog),
155                                              _("Error saving configuration"));
156                         gtk_dialog_run(GTK_DIALOG(dialog));
157                         gtk_widget_destroy(dialog);
158                 }
159                 g_error_free(error);
160         }
161         return 0;
162 }
163
164 gboolean
165 wpa_configure(const struct if_ap *ifa)
166 {
167         GtkWidget *dialog, *label, *psk, *vbox, *hbox;
168         const char *var, *mgt;
169         gint result, id, retval;
170
171         dialog = gtk_dialog_new_with_buttons(ifa->ssid,
172                 NULL,
173                 GTK_DIALOG_MODAL,
174                 GTK_STOCK_OK, GTK_RESPONSE_ACCEPT,
175                 GTK_STOCK_CANCEL, GTK_RESPONSE_REJECT,
176                 NULL);
177         gtk_window_set_resizable(GTK_WINDOW(dialog), FALSE);
178         gtk_window_set_icon_name(GTK_WINDOW(dialog), "config-users");
179         vbox = GTK_DIALOG(dialog)->vbox;
180
181         hbox = gtk_hbox_new(FALSE, 2);
182         label = gtk_label_new(_("Pre Shared Key:"));
183         gtk_box_pack_start(GTK_BOX(hbox), label, FALSE, FALSE, 0);
184         psk = gtk_entry_new();
185         gtk_entry_set_max_length(GTK_ENTRY(psk), 130);
186         gtk_box_pack_start(GTK_BOX(hbox), psk, TRUE, TRUE, 0);
187         gtk_container_add(GTK_CONTAINER(vbox), hbox);
188
189         gtk_widget_show_all(dialog);
190         result = gtk_dialog_run(GTK_DIALOG(dialog));
191         
192         id = -1;
193         retval = -1;
194         if (result == GTK_RESPONSE_ACCEPT) {
195                 id = find_network(ifa->ifname, ifa->ssid);
196                 if (g_strcmp0(ifa->flags, "[WEP]") == 0) {
197                         mgt = "NONE";
198                         var = "wep_key0";
199                 } else {
200                         mgt = "WPA-PSK";
201                         var = "psk";
202                 }
203                 if (id != -1) {
204                         retval = configure_network(ifa->ifname, id, "key_mgmt", 
205                                                mgt, FALSE);
206                         retval |= configure_network(ifa->ifname, id, var,
207                                                gtk_entry_get_text(GTK_ENTRY(psk)), TRUE);
208                 }
209         }
210         gtk_widget_destroy(dialog);
211         return retval == -1 ? FALSE : TRUE;
212 }