[iwmgmt] Make iwlist preserve existing associations, and be non-active
[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 ) {
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 was_channel = dev->channels[dev->channel].channel_nr;
118
119         if ( ! was_opened ) {
120                 dev->state |= NET80211_NO_ASSOC;
121                 rc = netdev_open ( dev->netdev );
122                 if ( rc < 0 )
123                         goto err;
124         }
125
126         if ( dev->state & NET80211_WORKING ) {
127                 rc = -EINVAL;
128                 goto err_close_netdev;
129         }
130
131         if ( ! was_opened ) {
132                 rc = net80211_prepare_probe ( dev, dev->hw->bands, 0 );
133                 if ( rc < 0 )
134                         goto err_close_netdev;
135         }
136
137         ctx = net80211_probe_start ( dev, "", 0 );
138         if ( ! ctx ) {
139                 rc = -ENOMEM;
140                 goto err_close_netdev;
141         }
142
143         while ( ! ( rc = net80211_probe_step ( ctx ) ) ) {
144                 step();
145         }
146
147         networks = net80211_probe_finish_all ( ctx );
148
149         if ( list_empty ( networks ) ) {
150                 goto err_free_networks;
151         }
152
153         rc = 0;
154
155         printf ( "Networks on %s:\n\n", dev->netdev->name );
156
157         /* Output format:
158          * 0         1         2         3         4         5         6
159          * 0123456789012345678901234567890123456789012345678901234567890
160          * [Sig] SSID                  BSSID              Ch  Crypt/Auth
161          * -------------------------------------------------------------
162          * [ 15] abcdefghijklmnopqrst> 00:00:00:00:00:00  11  Open
163          *                                             ... or WPA   PSK etc.
164          */
165
166         /* Quoting the dashes and spaces verbatim uses less code space
167            than generating them programmatically. */
168         printf ( "[Sig] SSID                  BSSID              Ch  Crypt/Auth\n"
169                  "-------------------------------------------------------------\n" );
170
171         list_for_each_entry ( wlan, networks, list ) {
172
173                 /* Format SSID into 22-character string, space-padded,
174                    with '>' indicating truncation */
175
176                 snprintf ( ssid_buf, sizeof ( ssid_buf ), "%s", wlan->essid );
177                 for ( i = strlen ( ssid_buf ); i < sizeof ( ssid_buf ) - 1;
178                       i++ )
179                         ssid_buf[i] = ' ';
180                 if ( ssid_buf[sizeof ( ssid_buf ) - 2] != ' ' )
181                         ssid_buf[sizeof ( ssid_buf ) - 2] = '>';
182                 ssid_buf[sizeof ( ssid_buf ) - 1] = 0;
183
184                 /* Sanity check */
185                 if ( wlan->crypto >= NR_CRYPTO_TYPES ||
186                      wlan->handshaking >= NR_AUTH_TYPES )
187                         continue;
188
189                 printf ( "[%3d] %s %s  %2d  %s%s  %s\n",
190                          wlan->signal < 0 ? 100 + wlan->signal : wlan->signal,
191                          ssid_buf, eth_ntoa ( wlan->bssid ), wlan->channel,
192                          crypto_types[wlan->crypto],
193                          wlan->crypto == NET80211_CRYPT_TKIP? " " : "",
194                          auth_types[wlan->handshaking] );
195         }
196         printf ( "\n" );
197
198  err_free_networks:
199         net80211_free_wlanlist ( networks );
200
201  err_close_netdev:
202         if ( ! was_opened ) {
203                 dev->state &= ~NET80211_NO_ASSOC;
204                 netdev_close ( dev->netdev );
205         } else {
206                 net80211_change_channel ( dev, was_channel );
207         }
208
209         if ( ! rc )
210                 return 0;
211
212  err:
213         printf ( "Scanning for networks on %s: %s\n",
214                  dev->netdev->name, strerror ( rc ) );
215         return rc;
216 }
217
218 /**
219  * Cause reassociation of 802.11 device
220  *
221  * @v dev       802.11 device
222  */
223 void iwassoc ( struct net80211_device *dev ) {
224         net80211_autoassociate ( dev );
225 }