contrib/package/iwinfo/src/iwinfo_lualib.c

   1 /*
   2  * iwinfo - Wireless Information Library - Lua Bindings
   3  *
   4  *   Copyright (C) 2009 Jo-Philipp Wich <xm@subsignal.org>
   5  *
   6  * The iwinfo library is free software: you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License version 2
   8  * as published by the Free Software Foundation.
   9  *
  10  * The iwinfo library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * See the GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with the iwinfo library. If not, see http://www.gnu.org/licenses/.
  17  */
  18
  19 #include "iwinfo_lualib.h"
  20
  21
  22 /*
  23  * ISO3166 country labels
  24  */
  25
  26 struct iwinfo_iso3166_label ISO3166_Names[] = {
  27         { 0x3030 /* 00 */, "World" },
  28         { 0x4144 /* AD */, "Andorra" },
  29         { 0x4145 /* AE */, "United Arab Emirates" },
  30         { 0x4146 /* AF */, "Afghanistan" },
  31         { 0x4147 /* AG */, "Antigua and Barbuda" },
  32         { 0x4149 /* AI */, "Anguilla" },
  33         { 0x414C /* AL */, "Albania" },
  34         { 0x414D /* AM */, "Armenia" },
  35         { 0x414E /* AN */, "Netherlands Antilles" },
  36         { 0x414F /* AO */, "Angola" },
  37         { 0x4151 /* AQ */, "Antarctica" },
  38         { 0x4152 /* AR */, "Argentina" },
  39         { 0x4153 /* AS */, "American Samoa" },
  40         { 0x4154 /* AT */, "Austria" },
  41         { 0x4155 /* AU */, "Australia" },
  42         { 0x4157 /* AW */, "Aruba" },
  43         { 0x4158 /* AX */, "Aland Islands" },
  44         { 0x415A /* AZ */, "Azerbaijan" },
  45         { 0x4241 /* BA */, "Bosnia and Herzegovina" },
  46         { 0x4242 /* BB */, "Barbados" },
  47         { 0x4244 /* BD */, "Bangladesh" },
  48         { 0x4245 /* BE */, "Belgium" },
  49         { 0x4246 /* BF */, "Burkina Faso" },
  50         { 0x4247 /* BG */, "Bulgaria" },
  51         { 0x4248 /* BH */, "Bahrain" },
  52         { 0x4249 /* BI */, "Burundi" },
  53         { 0x424A /* BJ */, "Benin" },
  54         { 0x424C /* BL */, "Saint Barthelemy" },
  55         { 0x424D /* BM */, "Bermuda" },
  56         { 0x424E /* BN */, "Brunei Darussalam" },
  57         { 0x424F /* BO */, "Bolivia" },
  58         { 0x4252 /* BR */, "Brazil" },
  59         { 0x4253 /* BS */, "Bahamas" },
  60         { 0x4254 /* BT */, "Bhutan" },
  61         { 0x4256 /* BV */, "Bouvet Island" },
  62         { 0x4257 /* BW */, "Botswana" },
  63         { 0x4259 /* BY */, "Belarus" },
  64         { 0x425A /* BZ */, "Belize" },
  65         { 0x4341 /* CA */, "Canada" },
  66         { 0x4343 /* CC */, "Cocos (Keeling) Islands" },
  67         { 0x4344 /* CD */, "Congo" },
  68         { 0x4346 /* CF */, "Central African Republic" },
  69         { 0x4347 /* CG */, "Congo" },
  70         { 0x4348 /* CH */, "Switzerland" },
  71         { 0x4349 /* CI */, "Cote d'Ivoire" },
  72         { 0x434B /* CK */, "Cook Islands" },
  73         { 0x434C /* CL */, "Chile" },
  74         { 0x434D /* CM */, "Cameroon" },
  75         { 0x434E /* CN */, "China" },
  76         { 0x434F /* CO */, "Colombia" },
  77         { 0x4352 /* CR */, "Costa Rica" },
  78         { 0x4355 /* CU */, "Cuba" },
  79         { 0x4356 /* CV */, "Cape Verde" },
  80         { 0x4358 /* CX */, "Christmas Island" },
  81         { 0x4359 /* CY */, "Cyprus" },
  82         { 0x435A /* CZ */, "Czech Republic" },
  83         { 0x4445 /* DE */, "Germany" },
  84         { 0x444A /* DJ */, "Djibouti" },
  85         { 0x444B /* DK */, "Denmark" },
  86         { 0x444D /* DM */, "Dominica" },
  87         { 0x444F /* DO */, "Dominican Republic" },
  88         { 0x445A /* DZ */, "Algeria" },
  89         { 0x4543 /* EC */, "Ecuador" },
  90         { 0x4545 /* EE */, "Estonia" },
  91         { 0x4547 /* EG */, "Egypt" },
  92         { 0x4548 /* EH */, "Western Sahara" },
  93         { 0x4552 /* ER */, "Eritrea" },
  94         { 0x4553 /* ES */, "Spain" },
  95         { 0x4554 /* ET */, "Ethiopia" },
  96         { 0x4649 /* FI */, "Finland" },
  97         { 0x464A /* FJ */, "Fiji" },
  98         { 0x464B /* FK */, "Falkland Islands" },
  99         { 0x464D /* FM */, "Micronesia" },
 100         { 0x464F /* FO */, "Faroe Islands" },
 101         { 0x4652 /* FR */, "France" },
 102         { 0x4741 /* GA */, "Gabon" },
 103         { 0x4742 /* GB */, "United Kingdom" },
 104         { 0x4744 /* GD */, "Grenada" },
 105         { 0x4745 /* GE */, "Georgia" },
 106         { 0x4746 /* GF */, "French Guiana" },
 107         { 0x4747 /* GG */, "Guernsey" },
 108         { 0x4748 /* GH */, "Ghana" },
 109         { 0x4749 /* GI */, "Gibraltar" },
 110         { 0x474C /* GL */, "Greenland" },
 111         { 0x474D /* GM */, "Gambia" },
 112         { 0x474E /* GN */, "Guinea" },
 113         { 0x4750 /* GP */, "Guadeloupe" },
 114         { 0x4751 /* GQ */, "Equatorial Guinea" },
 115         { 0x4752 /* GR */, "Greece" },
 116         { 0x4753 /* GS */, "South Georgia" },
 117         { 0x4754 /* GT */, "Guatemala" },
 118         { 0x4755 /* GU */, "Guam" },
 119         { 0x4757 /* GW */, "Guinea-Bissau" },
 120         { 0x4759 /* GY */, "Guyana" },
 121         { 0x484B /* HK */, "Hong Kong" },
 122         { 0x484D /* HM */, "Heard and McDonald Islands" },
 123         { 0x484E /* HN */, "Honduras" },
 124         { 0x4852 /* HR */, "Croatia" },
 125         { 0x4854 /* HT */, "Haiti" },
 126         { 0x4855 /* HU */, "Hungary" },
 127         { 0x4944 /* ID */, "Indonesia" },
 128         { 0x4945 /* IE */, "Ireland" },
 129         { 0x494C /* IL */, "Israel" },
 130         { 0x494D /* IM */, "Isle of Man" },
 131         { 0x494E /* IN */, "India" },
 132         { 0x494F /* IO */, "Chagos Islands" },
 133         { 0x4951 /* IQ */, "Iraq" },
 134         { 0x4952 /* IR */, "Iran" },
 135         { 0x4953 /* IS */, "Iceland" },
 136         { 0x4954 /* IT */, "Italy" },
 137         { 0x4A45 /* JE */, "Jersey" },
 138         { 0x4A4D /* JM */, "Jamaica" },
 139         { 0x4A4F /* JO */, "Jordan" },
 140         { 0x4A50 /* JP */, "Japan" },
 141         { 0x4B45 /* KE */, "Kenya" },
 142         { 0x4B47 /* KG */, "Kyrgyzstan" },
 143         { 0x4B48 /* KH */, "Cambodia" },
 144         { 0x4B49 /* KI */, "Kiribati" },
 145         { 0x4B4D /* KM */, "Comoros" },
 146         { 0x4B4E /* KN */, "Saint Kitts and Nevis" },
 147         { 0x4B50 /* KP */, "North Korea" },
 148         { 0x4B52 /* KR */, "South Korea" },
 149         { 0x4B57 /* KW */, "Kuwait" },
 150         { 0x4B59 /* KY */, "Cayman Islands" },
 151         { 0x4B5A /* KZ */, "Kazakhstan" },
 152         { 0x4C41 /* LA */, "Laos" },
 153         { 0x4C42 /* LB */, "Lebanon" },
 154         { 0x4C43 /* LC */, "Saint Lucia" },
 155         { 0x4C49 /* LI */, "Liechtenstein" },
 156         { 0x4C4B /* LK */, "Sri Lanka" },
 157         { 0x4C52 /* LR */, "Liberia" },
 158         { 0x4C53 /* LS */, "Lesotho" },
 159         { 0x4C54 /* LT */, "Lithuania" },
 160         { 0x4C55 /* LU */, "Luxembourg" },
 161         { 0x4C56 /* LV */, "Latvia" },
 162         { 0x4C59 /* LY */, "Libyan Arab Jamahiriya" },
 163         { 0x4D41 /* MA */, "Morocco" },
 164         { 0x4D43 /* MC */, "Monaco" },
 165         { 0x4D44 /* MD */, "Moldova" },
 166         { 0x4D45 /* ME */, "Montenegro" },
 167         { 0x4D46 /* MF */, "Saint Martin (French part)" },
 168         { 0x4D47 /* MG */, "Madagascar" },
 169         { 0x4D48 /* MH */, "Marshall Islands" },
 170         { 0x4D4B /* MK */, "Macedonia" },
 171         { 0x4D4C /* ML */, "Mali" },
 172         { 0x4D4D /* MM */, "Myanmar" },
 173         { 0x4D4E /* MN */, "Mongolia" },
 174         { 0x4D4F /* MO */, "Macao" },
 175         { 0x4D50 /* MP */, "Northern Mariana Islands" },
 176         { 0x4D51 /* MQ */, "Martinique" },
 177         { 0x4D52 /* MR */, "Mauritania" },
 178         { 0x4D53 /* MS */, "Montserrat" },
 179         { 0x4D54 /* MT */, "Malta" },
 180         { 0x4D55 /* MU */, "Mauritius" },
 181         { 0x4D56 /* MV */, "Maldives" },
 182         { 0x4D57 /* MW */, "Malawi" },
 183         { 0x4D58 /* MX */, "Mexico" },
 184         { 0x4D59 /* MY */, "Malaysia" },
 185         { 0x4D5A /* MZ */, "Mozambique" },
 186         { 0x4E41 /* NA */, "Namibia" },
 187         { 0x4E43 /* NC */, "New Caledonia" },
 188         { 0x4E45 /* NE */, "Niger" },
 189         { 0x4E46 /* NF */, "Norfolk Island" },
 190         { 0x4E47 /* NG */, "Nigeria" },
 191         { 0x4E49 /* NI */, "Nicaragua" },
 192         { 0x4E4C /* NL */, "Netherlands" },
 193         { 0x4E4F /* NO */, "Norway" },
 194         { 0x4E50 /* NP */, "Nepal" },
 195         { 0x4E52 /* NR */, "Nauru" },
 196         { 0x4E55 /* NU */, "Niue" },
 197         { 0x4E5A /* NZ */, "New Zealand" },
 198         { 0x4F4D /* OM */, "Oman" },
 199         { 0x5041 /* PA */, "Panama" },
 200         { 0x5045 /* PE */, "Peru" },
 201         { 0x5046 /* PF */, "French Polynesia" },
 202         { 0x5047 /* PG */, "Papua New Guinea" },
 203         { 0x5048 /* PH */, "Philippines" },
 204         { 0x504B /* PK */, "Pakistan" },
 205         { 0x504C /* PL */, "Poland" },
 206         { 0x504D /* PM */, "Saint Pierre and Miquelon" },
 207         { 0x504E /* PN */, "Pitcairn" },
 208         { 0x5052 /* PR */, "Puerto Rico" },
 209         { 0x5053 /* PS */, "Palestinian Territory" },
 210         { 0x5054 /* PT */, "Portugal" },
 211         { 0x5057 /* PW */, "Palau" },
 212         { 0x5059 /* PY */, "Paraguay" },
 213         { 0x5141 /* QA */, "Qatar" },
 214         { 0x5245 /* RE */, "Reunion" },
 215         { 0x524F /* RO */, "Romania" },
 216         { 0x5253 /* RS */, "Serbia" },
 217         { 0x5255 /* RU */, "Russian Federation" },
 218         { 0x5257 /* RW */, "Rwanda" },
 219         { 0x5341 /* SA */, "Saudi Arabia" },
 220         { 0x5342 /* SB */, "Solomon Islands" },
 221         { 0x5343 /* SC */, "Seychelles" },
 222         { 0x5344 /* SD */, "Sudan" },
 223         { 0x5345 /* SE */, "Sweden" },
 224         { 0x5347 /* SG */, "Singapore" },
 225         { 0x5348 /* SH */, "St. Helena and Dependencies" },
 226         { 0x5349 /* SI */, "Slovenia" },
 227         { 0x534A /* SJ */, "Svalbard and Jan Mayen" },
 228         { 0x534B /* SK */, "Slovakia" },
 229         { 0x534C /* SL */, "Sierra Leone" },
 230         { 0x534D /* SM */, "San Marino" },
 231         { 0x534E /* SN */, "Senegal" },
 232         { 0x534F /* SO */, "Somalia" },
 233         { 0x5352 /* SR */, "Suriname" },
 234         { 0x5354 /* ST */, "Sao Tome and Principe" },
 235         { 0x5356 /* SV */, "El Salvador" },
 236         { 0x5359 /* SY */, "Syrian Arab Republic" },
 237         { 0x535A /* SZ */, "Swaziland" },
 238         { 0x5443 /* TC */, "Turks and Caicos Islands" },
 239         { 0x5444 /* TD */, "Chad" },
 240         { 0x5446 /* TF */, "French Southern Territories" },
 241         { 0x5447 /* TG */, "Togo" },
 242         { 0x5448 /* TH */, "Thailand" },
 243         { 0x544A /* TJ */, "Tajikistan" },
 244         { 0x544B /* TK */, "Tokelau" },
 245         { 0x544C /* TL */, "Timor-Leste" },
 246         { 0x544D /* TM */, "Turkmenistan" },
 247         { 0x544E /* TN */, "Tunisia" },
 248         { 0x544F /* TO */, "Tonga" },
 249         { 0x5452 /* TR */, "Turkey" },
 250         { 0x5454 /* TT */, "Trinidad and Tobago" },
 251         { 0x5456 /* TV */, "Tuvalu" },
 252         { 0x5457 /* TW */, "Taiwan" },
 253         { 0x545A /* TZ */, "Tanzania" },
 254         { 0x5541 /* UA */, "Ukraine" },
 255         { 0x5547 /* UG */, "Uganda" },
 256         { 0x554D /* UM */, "U.S. Minor Outlying Islands" },
 257         { 0x5553 /* US */, "United States" },
 258         { 0x5559 /* UY */, "Uruguay" },
 259         { 0x555A /* UZ */, "Uzbekistan" },
 260         { 0x5641 /* VA */, "Vatican City State" },
 261         { 0x5643 /* VC */, "St. Vincent and Grenadines" },
 262         { 0x5645 /* VE */, "Venezuela" },
 263         { 0x5647 /* VG */, "Virgin Islands, British" },
 264         { 0x5649 /* VI */, "Virgin Islands, U.S." },
 265         { 0x564E /* VN */, "Viet Nam" },
 266         { 0x5655 /* VU */, "Vanuatu" },
 267         { 0x5746 /* WF */, "Wallis and Futuna" },
 268         { 0x5753 /* WS */, "Samoa" },
 269         { 0x5945 /* YE */, "Yemen" },
 270         { 0x5954 /* YT */, "Mayotte" },
 271         { 0x5A41 /* ZA */, "South Africa" },
 272         { 0x5A4D /* ZM */, "Zambia" },
 273         { 0x5A57 /* ZW */, "Zimbabwe" },
 274         { 0,               "" }
 275 };
 276
 277
 278
 279 /* Determine type */
 280 static int iwinfo_L_type(lua_State *L)
 281 {
 282         const char *ifname = luaL_checkstring(L, 1);
 283
 284 #ifdef USE_NL80211
 285         if( nl80211_probe(ifname) )
 286                 lua_pushstring(L, "nl80211");
 287         else
 288 #endif
 289
 290 #ifdef USE_MADWIFI
 291         if( madwifi_probe(ifname) )
 292                 lua_pushstring(L, "madwifi");
 293         else
 294 #endif
 295
 296 #ifdef USE_WL
 297         if( wl_probe(ifname) )
 298                 lua_pushstring(L, "wl");
 299         else
 300 #endif
 301
 302         if( wext_probe(ifname) )
 303                 lua_pushstring(L, "wext");
 304
 305         else
 306                 lua_pushnil(L);
 307
 308         return 1;
 309 }
 310
 311 /* Shutdown backends */
 312 static int iwinfo_L__gc(lua_State *L)
 313 {
 314 #ifdef USE_WL
 315         wl_close();
 316 #endif
 317 #ifdef USE_MADWIFI
 318         madwifi_close();
 319 #endif
 320 #ifdef USE_NL80211
 321         nl80211_close();
 322 #endif
 323         wext_close();
 324 }
 325
 326 /*
 327  * Build a short textual description of the crypto info
 328  */
 329
 330 static char * iwinfo_crypto_print_ciphers(int ciphers)
 331 {
 332         static char str[128] = { 0 };
 333         char *pos = str;
 334
 335         if( ciphers & IWINFO_CIPHER_WEP40 )
 336                 pos += sprintf(pos, "WEP-40, ");
 337
 338         if( ciphers & IWINFO_CIPHER_WEP104 )
 339                 pos += sprintf(pos, "WEP-104, ");
 340
 341         if( ciphers & IWINFO_CIPHER_TKIP )
 342                 pos += sprintf(pos, "TKIP, ");
 343
 344         if( ciphers & IWINFO_CIPHER_CCMP )
 345                 pos += sprintf(pos, "CCMP, ");
 346
 347         if( ciphers & IWINFO_CIPHER_WRAP )
 348                 pos += sprintf(pos, "WRAP, ");
 349
 350         if( ciphers & IWINFO_CIPHER_AESOCB )
 351                 pos += sprintf(pos, "AES-OCB, ");
 352
 353         if( ciphers & IWINFO_CIPHER_CKIP )
 354                 pos += sprintf(pos, "CKIP, ");
 355
 356         if( !ciphers || (ciphers & IWINFO_CIPHER_NONE) )
 357                 pos += sprintf(pos, "NONE, ");
 358
 359         *(pos - 2) = 0;
 360
 361         return str;
 362 }
 363
 364 static char * iwinfo_crypto_print_suites(int suites)
 365 {
 366         static char str[64] = { 0 };
 367         char *pos = str;
 368
 369         if( suites & IWINFO_KMGMT_PSK )
 370                 pos += sprintf(pos, "PSK/");
 371
 372         if( suites & IWINFO_KMGMT_8021x )
 373                 pos += sprintf(pos, "802.1X/");
 374
 375         if( !suites || (suites & IWINFO_KMGMT_NONE) )
 376                 pos += sprintf(pos, "NONE/");
 377
 378         *(pos - 1) = 0;
 379
 380         return str;
 381 }
 382
 383 static char * iwinfo_crypto_desc(struct iwinfo_crypto_entry *c)
 384 {
 385         static char desc[512] = { 0 };
 386
 387         if( c )
 388         {
 389                 if( c->enabled )
 390                 {
 391                         /* WEP */
 392                         if( c->auth_algs && !c->wpa_version )
 393                         {
 394                                 if( (c->auth_algs & IWINFO_AUTH_OPEN) &&
 395                                     (c->auth_algs & IWINFO_AUTH_SHARED) )
 396                                 {
 397                                         sprintf(desc, "WEP Open/Shared (%s)",
 398                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 399                                 }
 400                                 else if( c->auth_algs & IWINFO_AUTH_OPEN )
 401                                 {
 402                                         sprintf(desc, "WEP Open System (%s)",
 403                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 404                                 }
 405                                 else if( c->auth_algs & IWINFO_AUTH_SHARED )
 406                                 {
 407                                         sprintf(desc, "WEP Shared Auth (%s)",
 408                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 409                                 }
 410                         }
 411
 412                         /* WPA */
 413                         else if( c->wpa_version )
 414                         {
 415                                 switch(c->wpa_version) {
 416                                         case 3:
 417                                                 sprintf(desc, "mixed WPA/WPA2 %s (%s)",
 418                                                         iwinfo_crypto_print_suites(c->auth_suites),
 419                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 420                                                 break;
 421
 422                                         case 2:
 423                                                 sprintf(desc, "WPA2 %s (%s)",
 424                                                         iwinfo_crypto_print_suites(c->auth_suites),
 425                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 426                                                 break;
 427
 428                                         case 1:
 429                                                 sprintf(desc, "WPA %s (%s)",
 430                                                         iwinfo_crypto_print_suites(c->auth_suites),
 431                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 432                                                 break;
 433                                 }
 434                         }
 435                 }
 436                 else
 437                 {
 438                         sprintf(desc, "None");
 439                 }
 440         }
 441         else
 442         {
 443                 sprintf(desc, "Unknown");
 444         }
 445
 446         return desc;
 447 }
 448
 449 /* Build Lua table from crypto data */
 450 static void iwinfo_L_cryptotable(lua_State *L, struct iwinfo_crypto_entry *c)
 451 {
 452         int i, j;
 453
 454         lua_newtable(L);
 455
 456         lua_pushboolean(L, c->enabled);
 457         lua_setfield(L, -2, "enabled");
 458
 459         lua_pushstring(L, iwinfo_crypto_desc(c));
 460         lua_setfield(L, -2, "description");
 461
 462         lua_pushboolean(L, (c->enabled && !c->wpa_version));
 463         lua_setfield(L, -2, "wep");
 464
 465         lua_pushinteger(L, c->wpa_version);
 466         lua_setfield(L, -2, "wpa");
 467
 468         lua_newtable(L);
 469         for( i = 0, j = 1; i < 8; i++ )
 470         {
 471                 if( c->pair_ciphers & (1 << i) )
 472                 {
 473                         lua_pushstring(L, IWINFO_CIPHER_NAMES[i]);
 474                         lua_rawseti(L, -2, j++);
 475                 }
 476         }
 477         lua_setfield(L, -2, "pair_ciphers");
 478
 479         lua_newtable(L);
 480         for( i = 0, j = 1; i < 8; i++ )
 481         {
 482                 if( c->group_ciphers & (1 << i) )
 483                 {
 484                         lua_pushstring(L, IWINFO_CIPHER_NAMES[i]);
 485                         lua_rawseti(L, -2, j++);
 486                 }
 487         }
 488         lua_setfield(L, -2, "group_ciphers");
 489
 490         lua_newtable(L);
 491         for( i = 0, j = 1; i < 8; i++ )
 492         {
 493                 if( c->auth_suites & (1 << i) )
 494                 {
 495                         lua_pushstring(L, IWINFO_KMGMT_NAMES[i]);
 496                         lua_rawseti(L, -2, j++);
 497                 }
 498         }
 499         lua_setfield(L, -2, "auth_suites");
 500
 501         lua_newtable(L);
 502         for( i = 0, j = 1; i < 8; i++ )
 503         {
 504                 if( c->auth_algs & (1 << i) )
 505                 {
 506                         lua_pushstring(L, IWINFO_AUTH_NAMES[i]);
 507                         lua_rawseti(L, -2, j++);
 508                 }
 509         }
 510         lua_setfield(L, -2, "auth_algs");
 511 }
 512
 513
 514 /* Wrapper for assoclist */
 515 static int iwinfo_L_assoclist(lua_State *L, int (*func)(const char *, char *, int *))
 516 {
 517         int i, len;
 518         char rv[IWINFO_BUFSIZE];
 519         char macstr[18];
 520         const char *ifname = luaL_checkstring(L, 1);
 521         struct iwinfo_assoclist_entry *e;
 522
 523         lua_newtable(L);
 524         memset(rv, 0, sizeof(rv));
 525
 526         if( !(*func)(ifname, rv, &len) )
 527         {
 528                 for( i = 0; i < len; i += sizeof(struct iwinfo_assoclist_entry) )
 529                 {
 530                         e = (struct iwinfo_assoclist_entry *) &rv[i];
 531
 532                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
 533                                 e->mac[0], e->mac[1], e->mac[2],
 534                                 e->mac[3], e->mac[4], e->mac[5]);
 535
 536                         lua_newtable(L);
 537
 538                         lua_pushnumber(L, e->signal);
 539                         lua_setfield(L, -2, "signal");
 540
 541                         lua_pushnumber(L, e->noise);
 542                         lua_setfield(L, -2, "noise");
 543
 544                         lua_setfield(L, -2, macstr);
 545                 }
 546         }
 547
 548         return 1;
 549 }
 550
 551 /* Wrapper for tx power list */
 552 static int iwinfo_L_txpwrlist(lua_State *L, int (*func)(const char *, char *, int *))
 553 {
 554         int i, x, len;
 555         char rv[IWINFO_BUFSIZE];
 556         const char *ifname = luaL_checkstring(L, 1);
 557         struct iwinfo_txpwrlist_entry *e;
 558
 559         lua_newtable(L);
 560         memset(rv, 0, sizeof(rv));
 561
 562         if( !(*func)(ifname, rv, &len) )
 563         {
 564                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_txpwrlist_entry), x++ )
 565                 {
 566                         e = (struct iwinfo_txpwrlist_entry *) &rv[i];
 567
 568                         lua_newtable(L);
 569
 570                         lua_pushnumber(L, e->mw);
 571                         lua_setfield(L, -2, "mw");
 572
 573                         lua_pushnumber(L, e->dbm);
 574                         lua_setfield(L, -2, "dbm");
 575
 576                         lua_rawseti(L, -2, x);
 577                 }
 578         }
 579
 580         return 1;
 581 }
 582
 583 /* Wrapper for scan list */
 584 static int iwinfo_L_scanlist(lua_State *L, int (*func)(const char *, char *, int *))
 585 {
 586         int i, x, len;
 587         char rv[IWINFO_BUFSIZE];
 588         char macstr[18];
 589         const char *ifname = luaL_checkstring(L, 1);
 590         struct iwinfo_scanlist_entry *e;
 591
 592         lua_newtable(L);
 593         memset(rv, 0, sizeof(rv));
 594
 595         if( !(*func)(ifname, rv, &len) )
 596         {
 597                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_scanlist_entry), x++ )
 598                 {
 599                         e = (struct iwinfo_scanlist_entry *) &rv[i];
 600
 601                         lua_newtable(L);
 602
 603                         /* BSSID */
 604                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
 605                                 e->mac[0], e->mac[1], e->mac[2],
 606                                 e->mac[3], e->mac[4], e->mac[5]);
 607
 608                         lua_pushstring(L, macstr);
 609                         lua_setfield(L, -2, "bssid");
 610
 611                         /* ESSID */
 612                         if( e->ssid[0] )
 613                         {
 614                                 lua_pushstring(L, (char *) e->ssid);
 615                                 lua_setfield(L, -2, "ssid");
 616                         }
 617
 618                         /* Channel */
 619                         lua_pushinteger(L, e->channel);
 620                         lua_setfield(L, -2, "channel");
 621
 622                         /* Mode */
 623                         lua_pushstring(L, (char *) e->mode);
 624                         lua_setfield(L, -2, "mode");
 625
 626                         /* Quality, Signal */
 627                         lua_pushinteger(L, e->quality);
 628                         lua_setfield(L, -2, "quality");
 629
 630                         lua_pushinteger(L, e->quality_max);
 631                         lua_setfield(L, -2, "quality_max");
 632
 633                         lua_pushnumber(L, (e->signal - 0x100));
 634                         lua_setfield(L, -2, "signal");
 635
 636                         /* Crypto */
 637                         iwinfo_L_cryptotable(L, &e->crypto);
 638                         lua_setfield(L, -2, "encryption");
 639
 640                         lua_rawseti(L, -2, x);
 641                 }
 642         }
 643
 644         return 1;
 645 }
 646
 647 /* Wrapper for frequency list */
 648 static int iwinfo_L_freqlist(lua_State *L, int (*func)(const char *, char *, int *))
 649 {
 650         int i, x, len;
 651         char rv[IWINFO_BUFSIZE];
 652         const char *ifname = luaL_checkstring(L, 1);
 653         struct iwinfo_freqlist_entry *e;
 654
 655         lua_newtable(L);
 656         memset(rv, 0, sizeof(rv));
 657
 658         if( !(*func)(ifname, rv, &len) )
 659         {
 660                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_freqlist_entry), x++ )
 661                 {
 662                         e = (struct iwinfo_freqlist_entry *) &rv[i];
 663
 664                         lua_newtable(L);
 665
 666                         /* MHz */
 667                         lua_pushinteger(L, e->mhz);
 668                         lua_setfield(L, -2, "mhz");
 669
 670                         /* Channel */
 671                         lua_pushinteger(L, e->channel);
 672                         lua_setfield(L, -2, "channel");
 673
 674                         /* Restricted (DFS/TPC/Radar) */
 675                         lua_pushboolean(L, e->restricted);
 676                         lua_setfield(L, -2, "restricted");
 677
 678                         lua_rawseti(L, -2, x);
 679                 }
 680         }
 681
 682         return 1;
 683 }
 684
 685 /* Wrapper for crypto settings */
 686 static int iwinfo_L_encryption(lua_State *L, int (*func)(const char *, char *))
 687 {
 688         const char *ifname = luaL_checkstring(L, 1);
 689         struct iwinfo_crypto_entry c = { 0 };
 690
 691         if( !(*func)(ifname, (char *)&c) )
 692         {
 693                 iwinfo_L_cryptotable(L, &c);
 694                 return 1;
 695         }
 696
 697         return 0;
 698 }
 699
 700 /* Wrapper for hwmode list */
 701 static int iwinfo_L_hwmodelist(lua_State *L, int (*func)(const char*, int *))
 702 {
 703         const char *ifname = luaL_checkstring(L, 1);
 704         int hwmodes = 0;
 705
 706         if( !(*func)(ifname, &hwmodes) )
 707         {
 708                 lua_newtable(L);
 709
 710                 lua_pushboolean(L, hwmodes & IWINFO_80211_A);
 711                 lua_setfield(L, -2, "a");
 712
 713                 lua_pushboolean(L, hwmodes & IWINFO_80211_B);
 714                 lua_setfield(L, -2, "b");
 715
 716                 lua_pushboolean(L, hwmodes & IWINFO_80211_G);
 717                 lua_setfield(L, -2, "g");
 718
 719                 lua_pushboolean(L, hwmodes & IWINFO_80211_N);
 720                 lua_setfield(L, -2, "n");
 721
 722                 return 1;
 723         }
 724
 725         return 0;
 726 }
 727
 728 /* Wrapper for country list */
 729 static char * iwinfo_L_country_lookup(char *buf, int len, int iso3166)
 730 {
 731         int i;
 732         struct iwinfo_country_entry *c;
 733
 734         for( i = 0; i < len; i += sizeof(struct iwinfo_country_entry) )
 735         {
 736                 c = (struct iwinfo_country_entry *) &buf[i];
 737
 738                 if( c->iso3166 == iso3166 )
 739                         return c->ccode;
 740         }
 741
 742         return NULL;
 743 }
 744
 745 static int iwinfo_L_countrylist(lua_State *L, int (*func)(const char *, char *, int *))
 746 {
 747         int len, i, j;
 748         char rv[IWINFO_BUFSIZE], alpha2[3];
 749         char *ccode;
 750         const char *ifname = luaL_checkstring(L, 1);
 751         struct iwinfo_iso3166_label *l;
 752
 753         lua_newtable(L);
 754         memset(rv, 0, sizeof(rv));
 755
 756         if( !(*func)(ifname, rv, &len) )
 757         {
 758                 for( l = ISO3166_Names, j = 1; l->iso3166; l++ )
 759                 {
 760                         if( (ccode = iwinfo_L_country_lookup(rv, len, l->iso3166)) != NULL )
 761                         {
 762                                 sprintf(alpha2, "%c%c",
 763                                         (l->iso3166 / 256), (l->iso3166 % 256));
 764
 765                                 lua_newtable(L);
 766
 767                                 lua_pushstring(L, alpha2);
 768                                 lua_setfield(L, -2, "alpha2");
 769
 770                                 lua_pushstring(L, ccode);
 771                                 lua_setfield(L, -2, "ccode");
 772
 773                                 lua_pushstring(L, l->name);
 774                                 lua_setfield(L, -2, "name");
 775
 776                                 lua_rawseti(L, -2, j++);
 777                         }
 778                 }
 779         }
 780
 781         return 1;
 782 }
 783
 784
 785 #ifdef USE_WL
 786 /* Broadcom */
 787 LUA_WRAP_INT(wl,channel)
 788 LUA_WRAP_INT(wl,frequency)
 789 LUA_WRAP_INT(wl,txpower)
 790 LUA_WRAP_INT(wl,bitrate)
 791 LUA_WRAP_INT(wl,signal)
 792 LUA_WRAP_INT(wl,noise)
 793 LUA_WRAP_INT(wl,quality)
 794 LUA_WRAP_INT(wl,quality_max)
 795 LUA_WRAP_INT(wl,mbssid_support)
 796 LUA_WRAP_STRING(wl,mode)
 797 LUA_WRAP_STRING(wl,ssid)
 798 LUA_WRAP_STRING(wl,bssid)
 799 LUA_WRAP_STRING(wl,country)
 800 LUA_WRAP_LIST(wl,assoclist)
 801 LUA_WRAP_LIST(wl,txpwrlist)
 802 LUA_WRAP_LIST(wl,scanlist)
 803 LUA_WRAP_LIST(wl,freqlist)
 804 LUA_WRAP_LIST(wl,countrylist)
 805 LUA_WRAP_LIST(wl,hwmodelist)
 806 LUA_WRAP_LIST(wl,encryption)
 807 #endif
 808
 809 #ifdef USE_MADWIFI
 810 /* Madwifi */
 811 LUA_WRAP_INT(madwifi,channel)
 812 LUA_WRAP_INT(madwifi,frequency)
 813 LUA_WRAP_INT(madwifi,txpower)
 814 LUA_WRAP_INT(madwifi,bitrate)
 815 LUA_WRAP_INT(madwifi,signal)
 816 LUA_WRAP_INT(madwifi,noise)
 817 LUA_WRAP_INT(madwifi,quality)
 818 LUA_WRAP_INT(madwifi,quality_max)
 819 LUA_WRAP_INT(madwifi,mbssid_support)
 820 LUA_WRAP_STRING(madwifi,mode)
 821 LUA_WRAP_STRING(madwifi,ssid)
 822 LUA_WRAP_STRING(madwifi,bssid)
 823 LUA_WRAP_STRING(madwifi,country)
 824 LUA_WRAP_LIST(madwifi,assoclist)
 825 LUA_WRAP_LIST(madwifi,txpwrlist)
 826 LUA_WRAP_LIST(madwifi,scanlist)
 827 LUA_WRAP_LIST(madwifi,freqlist)
 828 LUA_WRAP_LIST(madwifi,countrylist)
 829 LUA_WRAP_LIST(madwifi,hwmodelist)
 830 LUA_WRAP_LIST(madwifi,encryption)
 831 #endif
 832
 833 #ifdef USE_NL80211
 834 /* NL80211 */
 835 LUA_WRAP_INT(nl80211,channel)
 836 LUA_WRAP_INT(nl80211,frequency)
 837 LUA_WRAP_INT(nl80211,txpower)
 838 LUA_WRAP_INT(nl80211,bitrate)
 839 LUA_WRAP_INT(nl80211,signal)
 840 LUA_WRAP_INT(nl80211,noise)
 841 LUA_WRAP_INT(nl80211,quality)
 842 LUA_WRAP_INT(nl80211,quality_max)
 843 LUA_WRAP_INT(nl80211,mbssid_support)
 844 LUA_WRAP_STRING(nl80211,mode)
 845 LUA_WRAP_STRING(nl80211,ssid)
 846 LUA_WRAP_STRING(nl80211,bssid)
 847 LUA_WRAP_STRING(nl80211,country)
 848 LUA_WRAP_LIST(nl80211,assoclist)
 849 LUA_WRAP_LIST(nl80211,txpwrlist)
 850 LUA_WRAP_LIST(nl80211,scanlist)
 851 LUA_WRAP_LIST(nl80211,freqlist)
 852 LUA_WRAP_LIST(nl80211,countrylist)
 853 LUA_WRAP_LIST(nl80211,hwmodelist)
 854 LUA_WRAP_LIST(nl80211,encryption)
 855 #endif
 856
 857 /* Wext */
 858 LUA_WRAP_INT(wext,channel)
 859 LUA_WRAP_INT(wext,frequency)
 860 LUA_WRAP_INT(wext,txpower)
 861 LUA_WRAP_INT(wext,bitrate)
 862 LUA_WRAP_INT(wext,signal)
 863 LUA_WRAP_INT(wext,noise)
 864 LUA_WRAP_INT(wext,quality)
 865 LUA_WRAP_INT(wext,quality_max)
 866 LUA_WRAP_INT(wext,mbssid_support)
 867 LUA_WRAP_STRING(wext,mode)
 868 LUA_WRAP_STRING(wext,ssid)
 869 LUA_WRAP_STRING(wext,bssid)
 870 LUA_WRAP_STRING(wext,country)
 871 LUA_WRAP_LIST(wext,assoclist)
 872 LUA_WRAP_LIST(wext,txpwrlist)
 873 LUA_WRAP_LIST(wext,scanlist)
 874 LUA_WRAP_LIST(wext,freqlist)
 875 LUA_WRAP_LIST(wext,countrylist)
 876 LUA_WRAP_LIST(wext,hwmodelist)
 877 LUA_WRAP_LIST(wext,encryption)
 878
 879 #ifdef USE_WL
 880 /* Broadcom table */
 881 static const luaL_reg R_wl[] = {
 882         LUA_REG(wl,channel),
 883         LUA_REG(wl,frequency),
 884         LUA_REG(wl,txpower),
 885         LUA_REG(wl,bitrate),
 886         LUA_REG(wl,signal),
 887         LUA_REG(wl,noise),
 888         LUA_REG(wl,quality),
 889         LUA_REG(wl,quality_max),
 890         LUA_REG(wl,mode),
 891         LUA_REG(wl,ssid),
 892         LUA_REG(wl,bssid),
 893         LUA_REG(wl,country),
 894         LUA_REG(wl,assoclist),
 895         LUA_REG(wl,txpwrlist),
 896         LUA_REG(wl,scanlist),
 897         LUA_REG(wl,freqlist),
 898         LUA_REG(wl,countrylist),
 899         LUA_REG(wl,hwmodelist),
 900         LUA_REG(wl,encryption),
 901         LUA_REG(wl,mbssid_support),
 902         { NULL, NULL }
 903 };
 904 #endif
 905
 906 #ifdef USE_MADWIFI
 907 /* Madwifi table */
 908 static const luaL_reg R_madwifi[] = {
 909         LUA_REG(madwifi,channel),
 910         LUA_REG(madwifi,frequency),
 911         LUA_REG(madwifi,txpower),
 912         LUA_REG(madwifi,bitrate),
 913         LUA_REG(madwifi,signal),
 914         LUA_REG(madwifi,noise),
 915         LUA_REG(madwifi,quality),
 916         LUA_REG(madwifi,quality_max),
 917         LUA_REG(madwifi,mode),
 918         LUA_REG(madwifi,ssid),
 919         LUA_REG(madwifi,bssid),
 920         LUA_REG(madwifi,country),
 921         LUA_REG(madwifi,assoclist),
 922         LUA_REG(madwifi,txpwrlist),
 923         LUA_REG(madwifi,scanlist),
 924         LUA_REG(madwifi,freqlist),
 925         LUA_REG(madwifi,countrylist),
 926         LUA_REG(madwifi,hwmodelist),
 927         LUA_REG(madwifi,encryption),
 928         LUA_REG(madwifi,mbssid_support),
 929         { NULL, NULL }
 930 };
 931 #endif
 932
 933 #ifdef USE_NL80211
 934 /* NL80211 table */
 935 static const luaL_reg R_nl80211[] = {
 936         LUA_REG(nl80211,channel),
 937         LUA_REG(nl80211,frequency),
 938         LUA_REG(nl80211,txpower),
 939         LUA_REG(nl80211,bitrate),
 940         LUA_REG(nl80211,signal),
 941         LUA_REG(nl80211,noise),
 942         LUA_REG(nl80211,quality),
 943         LUA_REG(nl80211,quality_max),
 944         LUA_REG(nl80211,mode),
 945         LUA_REG(nl80211,ssid),
 946         LUA_REG(nl80211,bssid),
 947         LUA_REG(nl80211,country),
 948         LUA_REG(nl80211,assoclist),
 949         LUA_REG(nl80211,txpwrlist),
 950         LUA_REG(nl80211,scanlist),
 951         LUA_REG(nl80211,freqlist),
 952         LUA_REG(nl80211,countrylist),
 953         LUA_REG(nl80211,hwmodelist),
 954         LUA_REG(nl80211,encryption),
 955         LUA_REG(nl80211,mbssid_support),
 956         { NULL, NULL }
 957 };
 958 #endif
 959
 960 /* Wext table */
 961 static const luaL_reg R_wext[] = {
 962         LUA_REG(wext,channel),
 963         LUA_REG(wext,frequency),
 964         LUA_REG(wext,txpower),
 965         LUA_REG(wext,bitrate),
 966         LUA_REG(wext,signal),
 967         LUA_REG(wext,noise),
 968         LUA_REG(wext,quality),
 969         LUA_REG(wext,quality_max),
 970         LUA_REG(wext,mode),
 971         LUA_REG(wext,ssid),
 972         LUA_REG(wext,bssid),
 973         LUA_REG(wext,country),
 974         LUA_REG(wext,assoclist),
 975         LUA_REG(wext,txpwrlist),
 976         LUA_REG(wext,scanlist),
 977         LUA_REG(wext,freqlist),
 978         LUA_REG(wext,countrylist),
 979         LUA_REG(wext,hwmodelist),
 980         LUA_REG(wext,encryption),
 981         LUA_REG(wext,mbssid_support),
 982         { NULL, NULL }
 983 };
 984
 985 /* Common */
 986 static const luaL_reg R_common[] = {
 987         { "type", iwinfo_L_type },
 988         { "__gc", iwinfo_L__gc  },
 989         { NULL, NULL }
 990 };
 991
 992
 993 LUALIB_API int luaopen_iwinfo(lua_State *L) {
 994         luaL_register(L, IWINFO_META, R_common);
 995
 996 #ifdef USE_WL
 997         luaL_newmetatable(L, IWINFO_WL_META);
 998         luaL_register(L, NULL, R_wl);
 999         lua_pushvalue(L, -1);
1000         lua_setfield(L, -2, "__index");
1001         lua_setfield(L, -2, "wl");
1002 #endif
1003
1004 #ifdef USE_MADWIFI
1005         luaL_newmetatable(L, IWINFO_MADWIFI_META);
1006         luaL_register(L, NULL, R_madwifi);
1007         lua_pushvalue(L, -1);
1008         lua_setfield(L, -2, "__index");
1009         lua_setfield(L, -2, "madwifi");
1010 #endif
1011
1012 #ifdef USE_NL80211
1013         luaL_newmetatable(L, IWINFO_NL80211_META);
1014         luaL_register(L, NULL, R_nl80211);
1015         lua_pushvalue(L, -1);
1016         lua_setfield(L, -2, "__index");
1017         lua_setfield(L, -2, "nl80211");
1018 #endif
1019
1020         luaL_newmetatable(L, IWINFO_WEXT_META);
1021         luaL_register(L, NULL, R_wext);
1022         lua_pushvalue(L, -1);
1023         lua_setfield(L, -2, "__index");
1024         lua_setfield(L, -2, "wext");
1025
1026         return 1;
1027 }