[iwmgmt] Add functions for common user-level wireless tasks
[people/oremanj/gpxe.git] / src / usr / iwmgmt.c
1 /*
2  * Copyright (C) 2009 Joshua Oreman <oremanj@rwcr.net>.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 FILE_LICENCE ( GPL2_OR_LATER );
20
21 #include <stdio.h>
22 #include <console.h>
23 #include <string.h>
24 #include <errno.h>
25 #include <gpxe/net80211.h>
26 #include <gpxe/ethernet.h>
27 #include <usr/ifmgmt.h>
28 #include <usr/iwmgmt.h>
29
30 /** @file
31  *
32  * Wireless network interface management
33  *
34  */
35
36 /**
37  * Print status of 802.11 device
38  *
39  * @v dev       802.11 device
40  */
41 void iwstat ( struct net80211_device *dev ) {
42
43         ifstat ( dev->netdev );
44
45         printf ( "  [802.11 ");
46         if ( dev->state & NET80211_ASSOCIATED ) {
47                 printf ( "SSID '%s', ", dev->essid );
48         } else {
49                 printf ( "not associated, " );
50         }
51         if ( dev->channel < dev->nr_channels && dev->rate < dev->nr_rates ) {
52                 printf ( "Ch:%d Sig:%d", dev->channels[dev->channel].channel_nr,
53                          dev->last_signal );
54                 switch ( dev->hw->signal_type ) {
55                 case NET80211_SIGNAL_NONE:
56                         printf ( "?" );
57                         break;
58                 case NET80211_SIGNAL_ARBITRARY:
59                         printf ( "/%d", dev->hw->signal_max );
60                         break;
61                 case NET80211_SIGNAL_DB:
62                         printf ( "/%d dB", dev->hw->signal_max );
63                         break;
64                 case NET80211_SIGNAL_DBM:
65                         printf ( " dBm" );
66                         break;
67                 }
68                 printf ( ", Rate:%d Mbps]\n",
69                          NET80211_RATE_VALUE ( dev->rates[dev->rate] ) / 10 );
70         } else {
71                 printf ( "antenna off]\n" );
72         }
73
74         if ( dev->state & NET80211_WORKING ) {
75                 printf ( "  [associating" );
76                 if ( dev->associating )
77                         printf ( " to '%s'", dev->associating->essid );
78                 printf ( "...]\n" );
79         } else if ( dev->assoc_rc ) {
80                 printf ( "  [association error: %s]\n",
81                          strerror ( dev->assoc_rc ) );
82         }
83 }
84
85
86 static const char *crypto_types[] = {
87         [NET80211_CRYPT_NONE] = "Open",
88         [NET80211_CRYPT_WEP] = "WEP",
89         [NET80211_CRYPT_TKIP] = "WPA",
90         [NET80211_CRYPT_CCMP] = "WPA2",
91 };
92
93 #define NR_CRYPTO_TYPES ( sizeof ( crypto_types ) / sizeof ( crypto_types[0] ) )
94
95 static const char *auth_types[] = {
96         [NET80211_SECPROT_NONE] = "",
97         [NET80211_SECPROT_PSK] = "PSK",
98         [NET80211_SECPROT_EAP] = "802.1X",
99 };
100
101 #define NR_AUTH_TYPES ( sizeof ( auth_types ) / sizeof ( auth_types[0] ) )
102
103 /**
104  * Scan for wireless networks using 802.11 device
105  *
106  * @v dev       802.11 device
107  * @v active    Whether to use active scanning
108  */
109 int iwlist ( struct net80211_device *dev, int active ) {
110         struct net80211_probe_ctx *ctx;
111         struct list_head *networks;
112         struct net80211_wlan *wlan;
113         char ssid_buf[22];
114         int rc;
115         unsigned i;
116         int was_opened = dev->netdev->state & NETDEV_OPEN;
117         int bands = dev->hw->bands;
118
119         if ( active )
120                 bands &= NET80211_BAND_2GHZ;
121
122         if ( ! was_opened ) {
123                 dev->state |= NET80211_NO_ASSOC;
124                 rc = netdev_open ( dev->netdev );
125                 if ( rc < 0 )
126                         goto err;
127         }
128
129         if ( dev->state & NET80211_WORKING ) {
130                 rc = -EINVAL;
131                 goto err_close_netdev;
132         }
133
134         rc = net80211_prepare_probe ( dev, dev->hw->bands, active );
135         if ( rc < 0 )
136                 goto err_close_netdev;
137
138         ctx = net80211_probe_start ( dev, "", active );
139         if ( ! ctx ) {
140                 rc = -ENOMEM;
141                 goto err_close_netdev;
142         }
143
144         while ( ! ( rc = net80211_probe_step ( ctx ) ) ) {
145                 step();
146         }
147
148         networks = net80211_probe_finish_all ( ctx );
149
150         if ( list_empty ( networks ) ) {
151                 goto err_free_networks;
152         }
153
154         printf ( "Networks on %s:\n\n", dev->netdev->name );
155
156         /* Output format:
157          * 0         1         2         3         4         5         6
158          * 0123456789012345678901234567890123456789012345678901234567890
159          * [Sig] SSID                  BSSID              Ch  Crypt/Auth
160          * -------------------------------------------------------------
161          * [ 15] abcdefghijklmnopqrst> 00:00:00:00:00:00  11  Open
162          *                                             ... or WPA   PSK etc.
163          */
164
165         /* Quoting the dashes and spaces verbatim uses less code space
166            than generating them programmatically. */
167         printf ( "[Sig] SSID                  BSSID              Ch  Crypt/Auth\n"
168                  "-------------------------------------------------------------\n" );
169
170         list_for_each_entry ( wlan, networks, list ) {
171
172                 /* Format SSID into 22-character string, space-padded,
173                    with '>' indicating truncation */
174
175                 snprintf ( ssid_buf, sizeof ( ssid_buf ), "%s", wlan->essid );
176                 for ( i = strlen ( ssid_buf ); i < sizeof ( ssid_buf ) - 1;
177                       i++ )
178                         ssid_buf[i] = ' ';
179                 if ( ssid_buf[sizeof ( ssid_buf ) - 2] != ' ' )
180                         ssid_buf[sizeof ( ssid_buf ) - 2] = '>';
181                 ssid_buf[sizeof ( ssid_buf ) - 1] = 0;
182
183                 /* Sanity check */
184                 if ( wlan->crypto >= NR_CRYPTO_TYPES ||
185                      wlan->handshaking >= NR_AUTH_TYPES )
186                         continue;
187
188                 printf ( "[%3d] %s %s  %2d  %s%s  %s\n",
189                          wlan->signal < 0 ? 100 + wlan->signal : wlan->signal,
190                          ssid_buf, eth_ntoa ( wlan->bssid ), wlan->channel,
191                          crypto_types[wlan->crypto],
192                          wlan->crypto == NET80211_CRYPT_TKIP? " " : "",
193                          auth_types[wlan->handshaking] );
194         }
195         printf ( "\n" );
196
197         net80211_free_wlanlist ( networks );
198
199         if ( ! was_opened ) {
200                 dev->state &= ~NET80211_NO_ASSOC;
201                 netdev_close ( dev->netdev );
202         }
203
204         return 0;
205
206  err_free_networks:
207         net80211_free_wlanlist ( networks );
208  err_close_netdev:
209         if ( ! was_opened ) {
210                 dev->state &= ~NET80211_NO_ASSOC;
211                 netdev_close ( dev->netdev );
212         }
213  err:
214         printf ( "Scanning for networks on %s: %s\n",
215                  dev->netdev->name, strerror ( rc ) );
216         return rc;
217 }
218
219 /**
220  * Cause reassociation of 802.11 device
221  *
222  * @v dev       802.11 device
223  */
224 void iwassoc ( struct net80211_device *dev ) {
225         net80211_autoassociate ( dev );
226 }