850216e27f6d998d8706932e93e30ac4e696c476
[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 *mgmt,
100                 const char *var, const char *val,
101                 gboolean quote)
102 {
103         GError *error;
104         char *str;
105         static gboolean warned = FALSE;
106         GtkWidget *dialog;
107
108         if (id == -1)
109                 return -1;
110
111         dbus_g_proxy_call(dbus, "SetNetwork", &error,
112                                G_TYPE_STRING, ifname,
113                                G_TYPE_INT, id,
114                                G_TYPE_STRING, "key_mgmt",
115                                G_TYPE_STRING, mgmt,
116                                G_TYPE_INVALID,
117                                G_TYPE_INVALID);
118
119         error = NULL;
120         if (quote)
121                 str = g_strconcat("\"", val, "\"", NULL);
122         else
123                 str = NULL;
124         if (!dbus_g_proxy_call(dbus, "SetNetwork", &error,
125                                G_TYPE_STRING, ifname,
126                                G_TYPE_INT, id,
127                                G_TYPE_STRING, var,
128                                G_TYPE_STRING, quote ? str : val,
129                                G_TYPE_INVALID,
130                                G_TYPE_INVALID))
131         {
132                 g_warning("SetNetwork: %s", error->message);
133                 g_free(str);
134                 g_error_free(error);
135                 dialog = gtk_message_dialog_new(NULL,
136                                 GTK_DIALOG_MODAL,
137                                 GTK_MESSAGE_ERROR,
138                                 GTK_BUTTONS_OK,
139                                 _("Failed to set password, probably too short."));
140                 gtk_window_set_title(GTK_WINDOW(dialog),
141                                 _("Error setting password"));
142                 gtk_dialog_run(GTK_DIALOG(dialog));
143                 gtk_widget_destroy(dialog);
144                 return -1;
145         }
146         g_free(str);
147
148         if (!dbus_g_proxy_call(dbus, "EnableNetwork", &error,
149                                G_TYPE_STRING, ifname,
150                                G_TYPE_INT, id,
151                                G_TYPE_INVALID,
152                                G_TYPE_INVALID))
153         {
154                 g_warning("EnableNetwork: %s", error->message);
155                 g_error_free(error);
156                 return -1;
157         }
158
159         if (!dbus_g_proxy_call(dbus, "SaveConfig", &error,
160                                G_TYPE_STRING, ifname,
161                                G_TYPE_INVALID,
162                                G_TYPE_INVALID))
163         {
164                 g_warning("SaveConfig: %s", error->message);
165                 if (!warned) {
166                         warned = TRUE;
167                         dialog = gtk_message_dialog_new(NULL,
168                                                         GTK_DIALOG_MODAL,
169                                                         GTK_MESSAGE_ERROR,
170                                                         GTK_BUTTONS_OK,
171                                                         _("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."));
172                         gtk_window_set_title(GTK_WINDOW(dialog),
173                                              _("Error saving configuration"));
174                         gtk_dialog_run(GTK_DIALOG(dialog));
175                         gtk_widget_destroy(dialog);
176                 }
177                 g_error_free(error);
178         }
179 /*
180         if (!dbus_g_proxy_call(dbus, "Disconnect", &error,
181                                G_TYPE_STRING, ifname,
182                                G_TYPE_INVALID,
183                                G_TYPE_INVALID))
184         {
185                 g_warning("Disconnect: %s", error->message);
186                 g_error_free(error);
187         }
188 */
189         if (!dbus_g_proxy_call(dbus, "Reassociate", &error,
190                                G_TYPE_STRING, ifname,
191                                G_TYPE_INVALID,
192                                G_TYPE_INVALID))
193         {
194                 g_warning("Reassociate: %s", error->message);
195                 g_error_free(error);
196         }
197         
198         return 0;
199 }
200
201 gboolean
202 wpa_configure(const struct if_ap *ifa)
203 {
204         GtkWidget *dialog, *label, *psk, *vbox, *hbox;
205         const char *var, *mgt;
206         gint result, id, retval;
207
208         dialog = gtk_dialog_new_with_buttons(ifa->ssid,
209                 NULL,
210                 GTK_DIALOG_MODAL,
211                 GTK_STOCK_OK, GTK_RESPONSE_ACCEPT,
212                 GTK_STOCK_CANCEL, GTK_RESPONSE_REJECT,
213                 NULL);
214         gtk_window_set_resizable(GTK_WINDOW(dialog), FALSE);
215         gtk_window_set_icon_name(GTK_WINDOW(dialog), "config-users");
216         vbox = GTK_DIALOG(dialog)->vbox;
217
218         hbox = gtk_hbox_new(FALSE, 2);
219         label = gtk_label_new(_("Pre Shared Key:"));
220         gtk_box_pack_start(GTK_BOX(hbox), label, FALSE, FALSE, 0);
221         psk = gtk_entry_new();
222         gtk_entry_set_max_length(GTK_ENTRY(psk), 130);
223         gtk_box_pack_start(GTK_BOX(hbox), psk, TRUE, TRUE, 0);
224         gtk_container_add(GTK_CONTAINER(vbox), hbox);
225
226         gtk_widget_show_all(dialog);
227         result = gtk_dialog_run(GTK_DIALOG(dialog));
228         
229         id = -1;
230         retval = -1;
231         if (result == GTK_RESPONSE_ACCEPT) {
232                 id = find_network(ifa->ifname, ifa->ssid);
233                 if (g_strcmp0(ifa->flags, "[WEP]") == 0) {
234                         mgt = "NONE";
235                         var = "wep_key0";
236                 } else {
237                         mgt = "WPA-PSK";
238                         var = "psk";
239                 }
240                 if (id != -1) {
241                         retval = configure_network(ifa->ifname, id, mgt, var,
242                                         gtk_entry_get_text(GTK_ENTRY(psk)), TRUE);
243                 }
244         }
245         gtk_widget_destroy(dialog);
246         return retval == -1 ? FALSE : TRUE;
247 }