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         iwinfo_close();
 325 }
 326
 327 /*
 328  * Build a short textual description of the crypto info
 329  */
 330
 331 static char * iwinfo_crypto_print_ciphers(int ciphers)
 332 {
 333         static char str[128] = { 0 };
 334         char *pos = str;
 335
 336         if( ciphers & IWINFO_CIPHER_WEP40 )
 337                 pos += sprintf(pos, "WEP-40, ");
 338
 339         if( ciphers & IWINFO_CIPHER_WEP104 )
 340                 pos += sprintf(pos, "WEP-104, ");
 341
 342         if( ciphers & IWINFO_CIPHER_TKIP )
 343                 pos += sprintf(pos, "TKIP, ");
 344
 345         if( ciphers & IWINFO_CIPHER_CCMP )
 346                 pos += sprintf(pos, "CCMP, ");
 347
 348         if( ciphers & IWINFO_CIPHER_WRAP )
 349                 pos += sprintf(pos, "WRAP, ");
 350
 351         if( ciphers & IWINFO_CIPHER_AESOCB )
 352                 pos += sprintf(pos, "AES-OCB, ");
 353
 354         if( ciphers & IWINFO_CIPHER_CKIP )
 355                 pos += sprintf(pos, "CKIP, ");
 356
 357         if( !ciphers || (ciphers & IWINFO_CIPHER_NONE) )
 358                 pos += sprintf(pos, "NONE, ");
 359
 360         *(pos - 2) = 0;
 361
 362         return str;
 363 }
 364
 365 static char * iwinfo_crypto_print_suites(int suites)
 366 {
 367         static char str[64] = { 0 };
 368         char *pos = str;
 369
 370         if( suites & IWINFO_KMGMT_PSK )
 371                 pos += sprintf(pos, "PSK/");
 372
 373         if( suites & IWINFO_KMGMT_8021x )
 374                 pos += sprintf(pos, "802.1X/");
 375
 376         if( !suites || (suites & IWINFO_KMGMT_NONE) )
 377                 pos += sprintf(pos, "NONE/");
 378
 379         *(pos - 1) = 0;
 380
 381         return str;
 382 }
 383
 384 static char * iwinfo_crypto_desc(struct iwinfo_crypto_entry *c)
 385 {
 386         static char desc[512] = { 0 };
 387
 388         if( c )
 389         {
 390                 if( c->enabled )
 391                 {
 392                         /* WEP */
 393                         if( c->auth_algs && !c->wpa_version )
 394                         {
 395                                 if( (c->auth_algs & IWINFO_AUTH_OPEN) &&
 396                                     (c->auth_algs & IWINFO_AUTH_SHARED) )
 397                                 {
 398                                         sprintf(desc, "WEP Open/Shared (%s)",
 399                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 400                                 }
 401                                 else if( c->auth_algs & IWINFO_AUTH_OPEN )
 402                                 {
 403                                         sprintf(desc, "WEP Open System (%s)",
 404                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 405                                 }
 406                                 else if( c->auth_algs & IWINFO_AUTH_SHARED )
 407                                 {
 408                                         sprintf(desc, "WEP Shared Auth (%s)",
 409                                                 iwinfo_crypto_print_ciphers(c->pair_ciphers));
 410                                 }
 411                         }
 412
 413                         /* WPA */
 414                         else if( c->wpa_version )
 415                         {
 416                                 switch(c->wpa_version) {
 417                                         case 3:
 418                                                 sprintf(desc, "mixed WPA/WPA2 %s (%s)",
 419                                                         iwinfo_crypto_print_suites(c->auth_suites),
 420                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 421                                                 break;
 422
 423                                         case 2:
 424                                                 sprintf(desc, "WPA2 %s (%s)",
 425                                                         iwinfo_crypto_print_suites(c->auth_suites),
 426                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 427                                                 break;
 428
 429                                         case 1:
 430                                                 sprintf(desc, "WPA %s (%s)",
 431                                                         iwinfo_crypto_print_suites(c->auth_suites),
 432                                                         iwinfo_crypto_print_ciphers(c->pair_ciphers));
 433                                                 break;
 434                                 }
 435                         }
 436                         else
 437                         {
 438                                 sprintf(desc, "None");
 439                         }
 440                 }
 441                 else
 442                 {
 443                         sprintf(desc, "None");
 444                 }
 445         }
 446         else
 447         {
 448                 sprintf(desc, "Unknown");
 449         }
 450
 451         return desc;
 452 }
 453
 454 /* Build Lua table from crypto data */
 455 static void iwinfo_L_cryptotable(lua_State *L, struct iwinfo_crypto_entry *c)
 456 {
 457         int i, j;
 458
 459         lua_newtable(L);
 460
 461         lua_pushboolean(L, c->enabled);
 462         lua_setfield(L, -2, "enabled");
 463
 464         lua_pushstring(L, iwinfo_crypto_desc(c));
 465         lua_setfield(L, -2, "description");
 466
 467         lua_pushboolean(L, (c->enabled && !c->wpa_version));
 468         lua_setfield(L, -2, "wep");
 469
 470         lua_pushinteger(L, c->wpa_version);
 471         lua_setfield(L, -2, "wpa");
 472
 473         lua_newtable(L);
 474         for( i = 0, j = 1; i < 8; i++ )
 475         {
 476                 if( c->pair_ciphers & (1 << i) )
 477                 {
 478                         lua_pushstring(L, IWINFO_CIPHER_NAMES[i]);
 479                         lua_rawseti(L, -2, j++);
 480                 }
 481         }
 482         lua_setfield(L, -2, "pair_ciphers");
 483
 484         lua_newtable(L);
 485         for( i = 0, j = 1; i < 8; i++ )
 486         {
 487                 if( c->group_ciphers & (1 << i) )
 488                 {
 489                         lua_pushstring(L, IWINFO_CIPHER_NAMES[i]);
 490                         lua_rawseti(L, -2, j++);
 491                 }
 492         }
 493         lua_setfield(L, -2, "group_ciphers");
 494
 495         lua_newtable(L);
 496         for( i = 0, j = 1; i < 8; i++ )
 497         {
 498                 if( c->auth_suites & (1 << i) )
 499                 {
 500                         lua_pushstring(L, IWINFO_KMGMT_NAMES[i]);
 501                         lua_rawseti(L, -2, j++);
 502                 }
 503         }
 504         lua_setfield(L, -2, "auth_suites");
 505
 506         lua_newtable(L);
 507         for( i = 0, j = 1; i < 8; i++ )
 508         {
 509                 if( c->auth_algs & (1 << i) )
 510                 {
 511                         lua_pushstring(L, IWINFO_AUTH_NAMES[i]);
 512                         lua_rawseti(L, -2, j++);
 513                 }
 514         }
 515         lua_setfield(L, -2, "auth_algs");
 516 }
 517
 518
 519 /* Wrapper for assoclist */
 520 static int iwinfo_L_assoclist(lua_State *L, int (*func)(const char *, char *, int *))
 521 {
 522         int i, len;
 523         char rv[IWINFO_BUFSIZE];
 524         char macstr[18];
 525         const char *ifname = luaL_checkstring(L, 1);
 526         struct iwinfo_assoclist_entry *e;
 527
 528         lua_newtable(L);
 529         memset(rv, 0, sizeof(rv));
 530
 531         if( !(*func)(ifname, rv, &len) )
 532         {
 533                 for( i = 0; i < len; i += sizeof(struct iwinfo_assoclist_entry) )
 534                 {
 535                         e = (struct iwinfo_assoclist_entry *) &rv[i];
 536
 537                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
 538                                 e->mac[0], e->mac[1], e->mac[2],
 539                                 e->mac[3], e->mac[4], e->mac[5]);
 540
 541                         lua_newtable(L);
 542
 543                         lua_pushnumber(L, e->signal);
 544                         lua_setfield(L, -2, "signal");
 545
 546                         lua_pushnumber(L, e->noise);
 547                         lua_setfield(L, -2, "noise");
 548
 549                         lua_setfield(L, -2, macstr);
 550                 }
 551         }
 552
 553         return 1;
 554 }
 555
 556 /* Wrapper for tx power list */
 557 static int iwinfo_L_txpwrlist(lua_State *L, int (*func)(const char *, char *, int *))
 558 {
 559         int i, x, len;
 560         char rv[IWINFO_BUFSIZE];
 561         const char *ifname = luaL_checkstring(L, 1);
 562         struct iwinfo_txpwrlist_entry *e;
 563
 564         lua_newtable(L);
 565         memset(rv, 0, sizeof(rv));
 566
 567         if( !(*func)(ifname, rv, &len) )
 568         {
 569                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_txpwrlist_entry), x++ )
 570                 {
 571                         e = (struct iwinfo_txpwrlist_entry *) &rv[i];
 572
 573                         lua_newtable(L);
 574
 575                         lua_pushnumber(L, e->mw);
 576                         lua_setfield(L, -2, "mw");
 577
 578                         lua_pushnumber(L, e->dbm);
 579                         lua_setfield(L, -2, "dbm");
 580
 581                         lua_rawseti(L, -2, x);
 582                 }
 583         }
 584
 585         return 1;
 586 }
 587
 588 /* Wrapper for scan list */
 589 static int iwinfo_L_scanlist(lua_State *L, int (*func)(const char *, char *, int *))
 590 {
 591         int i, x, len;
 592         char rv[IWINFO_BUFSIZE];
 593         char macstr[18];
 594         const char *ifname = luaL_checkstring(L, 1);
 595         struct iwinfo_scanlist_entry *e;
 596
 597         lua_newtable(L);
 598         memset(rv, 0, sizeof(rv));
 599
 600         if( !(*func)(ifname, rv, &len) )
 601         {
 602                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_scanlist_entry), x++ )
 603                 {
 604                         e = (struct iwinfo_scanlist_entry *) &rv[i];
 605
 606                         lua_newtable(L);
 607
 608                         /* BSSID */
 609                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
 610                                 e->mac[0], e->mac[1], e->mac[2],
 611                                 e->mac[3], e->mac[4], e->mac[5]);
 612
 613                         lua_pushstring(L, macstr);
 614                         lua_setfield(L, -2, "bssid");
 615
 616                         /* ESSID */
 617                         if( e->ssid[0] )
 618                         {
 619                                 lua_pushstring(L, (char *) e->ssid);
 620                                 lua_setfield(L, -2, "ssid");
 621                         }
 622
 623                         /* Channel */
 624                         lua_pushinteger(L, e->channel);
 625                         lua_setfield(L, -2, "channel");
 626
 627                         /* Mode */
 628                         lua_pushstring(L, (char *) e->mode);
 629                         lua_setfield(L, -2, "mode");
 630
 631                         /* Quality, Signal */
 632                         lua_pushinteger(L, e->quality);
 633                         lua_setfield(L, -2, "quality");
 634
 635                         lua_pushinteger(L, e->quality_max);
 636                         lua_setfield(L, -2, "quality_max");
 637
 638                         lua_pushnumber(L, (e->signal - 0x100));
 639                         lua_setfield(L, -2, "signal");
 640
 641                         /* Crypto */
 642                         iwinfo_L_cryptotable(L, &e->crypto);
 643                         lua_setfield(L, -2, "encryption");
 644
 645                         lua_rawseti(L, -2, x);
 646                 }
 647         }
 648
 649         return 1;
 650 }
 651
 652 /* Wrapper for frequency list */
 653 static int iwinfo_L_freqlist(lua_State *L, int (*func)(const char *, char *, int *))
 654 {
 655         int i, x, len;
 656         char rv[IWINFO_BUFSIZE];
 657         const char *ifname = luaL_checkstring(L, 1);
 658         struct iwinfo_freqlist_entry *e;
 659
 660         lua_newtable(L);
 661         memset(rv, 0, sizeof(rv));
 662
 663         if( !(*func)(ifname, rv, &len) )
 664         {
 665                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_freqlist_entry), x++ )
 666                 {
 667                         e = (struct iwinfo_freqlist_entry *) &rv[i];
 668
 669                         lua_newtable(L);
 670
 671                         /* MHz */
 672                         lua_pushinteger(L, e->mhz);
 673                         lua_setfield(L, -2, "mhz");
 674
 675                         /* Channel */
 676                         lua_pushinteger(L, e->channel);
 677                         lua_setfield(L, -2, "channel");
 678
 679                         /* Restricted (DFS/TPC/Radar) */
 680                         lua_pushboolean(L, e->restricted);
 681                         lua_setfield(L, -2, "restricted");
 682
 683                         lua_rawseti(L, -2, x);
 684                 }
 685         }
 686
 687         return 1;
 688 }
 689
 690 /* Wrapper for crypto settings */
 691 static int iwinfo_L_encryption(lua_State *L, int (*func)(const char *, char *))
 692 {
 693         const char *ifname = luaL_checkstring(L, 1);
 694         struct iwinfo_crypto_entry c = { 0 };
 695
 696         if( !(*func)(ifname, (char *)&c) )
 697         {
 698                 iwinfo_L_cryptotable(L, &c);
 699                 return 1;
 700         }
 701
 702         lua_pushnil(L);
 703         return 1;
 704 }
 705
 706 /* Wrapper for hwmode list */
 707 static int iwinfo_L_hwmodelist(lua_State *L, int (*func)(const char *, int *))
 708 {
 709         const char *ifname = luaL_checkstring(L, 1);
 710         int hwmodes = 0;
 711
 712         if( !(*func)(ifname, &hwmodes) )
 713         {
 714                 lua_newtable(L);
 715
 716                 lua_pushboolean(L, hwmodes & IWINFO_80211_A);
 717                 lua_setfield(L, -2, "a");
 718
 719                 lua_pushboolean(L, hwmodes & IWINFO_80211_B);
 720                 lua_setfield(L, -2, "b");
 721
 722                 lua_pushboolean(L, hwmodes & IWINFO_80211_G);
 723                 lua_setfield(L, -2, "g");
 724
 725                 lua_pushboolean(L, hwmodes & IWINFO_80211_N);
 726                 lua_setfield(L, -2, "n");
 727
 728                 return 1;
 729         }
 730
 731         lua_pushnil(L);
 732         return 1;
 733 }
 734
 735 /* Wrapper for mbbsid_support */
 736 static int iwinfo_L_mbssid_support(lua_State *L, int (*func)(const char *, int *))
 737 {
 738         const char *ifname = luaL_checkstring(L, 1);
 739         int support = 0;
 740
 741         if( !(*func)(ifname, &support) )
 742         {
 743                 lua_pushboolean(L, support);
 744                 return 1;
 745         }
 746
 747         lua_pushnil(L);
 748         return 1;
 749 }
 750
 751 /* Wrapper for country list */
 752 static char * iwinfo_L_country_lookup(char *buf, int len, int iso3166)
 753 {
 754         int i;
 755         struct iwinfo_country_entry *c;
 756
 757         for( i = 0; i < len; i += sizeof(struct iwinfo_country_entry) )
 758         {
 759                 c = (struct iwinfo_country_entry *) &buf[i];
 760
 761                 if( c->iso3166 == iso3166 )
 762                         return c->ccode;
 763         }
 764
 765         return NULL;
 766 }
 767
 768 static int iwinfo_L_countrylist(lua_State *L, int (*func)(const char *, char *, int *))
 769 {
 770         int len, i, j;
 771         char rv[IWINFO_BUFSIZE], alpha2[3];
 772         char *ccode;
 773         const char *ifname = luaL_checkstring(L, 1);
 774         struct iwinfo_iso3166_label *l;
 775
 776         lua_newtable(L);
 777         memset(rv, 0, sizeof(rv));
 778
 779         if( !(*func)(ifname, rv, &len) )
 780         {
 781                 for( l = ISO3166_Names, j = 1; l->iso3166; l++ )
 782                 {
 783                         if( (ccode = iwinfo_L_country_lookup(rv, len, l->iso3166)) != NULL )
 784                         {
 785                                 sprintf(alpha2, "%c%c",
 786                                         (l->iso3166 / 256), (l->iso3166 % 256));
 787
 788                                 lua_newtable(L);
 789
 790                                 lua_pushstring(L, alpha2);
 791                                 lua_setfield(L, -2, "alpha2");
 792
 793                                 lua_pushstring(L, ccode);
 794                                 lua_setfield(L, -2, "ccode");
 795
 796                                 lua_pushstring(L, l->name);
 797                                 lua_setfield(L, -2, "name");
 798
 799                                 lua_rawseti(L, -2, j++);
 800                         }
 801                 }
 802         }
 803
 804         return 1;
 805 }
 806
 807
 808 #ifdef USE_WL
 809 /* Broadcom */
 810 LUA_WRAP_INT(wl,channel)
 811 LUA_WRAP_INT(wl,frequency)
 812 LUA_WRAP_INT(wl,txpower)
 813 LUA_WRAP_INT(wl,bitrate)
 814 LUA_WRAP_INT(wl,signal)
 815 LUA_WRAP_INT(wl,noise)
 816 LUA_WRAP_INT(wl,quality)
 817 LUA_WRAP_INT(wl,quality_max)
 818 LUA_WRAP_STRING(wl,mode)
 819 LUA_WRAP_STRING(wl,ssid)
 820 LUA_WRAP_STRING(wl,bssid)
 821 LUA_WRAP_STRING(wl,country)
 822 LUA_WRAP_LIST(wl,assoclist)
 823 LUA_WRAP_LIST(wl,txpwrlist)
 824 LUA_WRAP_LIST(wl,scanlist)
 825 LUA_WRAP_LIST(wl,freqlist)
 826 LUA_WRAP_LIST(wl,countrylist)
 827 LUA_WRAP_LIST(wl,hwmodelist)
 828 LUA_WRAP_LIST(wl,encryption)
 829 LUA_WRAP_LIST(wl,mbssid_support)
 830 #endif
 831
 832 #ifdef USE_MADWIFI
 833 /* Madwifi */
 834 LUA_WRAP_INT(madwifi,channel)
 835 LUA_WRAP_INT(madwifi,frequency)
 836 LUA_WRAP_INT(madwifi,txpower)
 837 LUA_WRAP_INT(madwifi,bitrate)
 838 LUA_WRAP_INT(madwifi,signal)
 839 LUA_WRAP_INT(madwifi,noise)
 840 LUA_WRAP_INT(madwifi,quality)
 841 LUA_WRAP_INT(madwifi,quality_max)
 842 LUA_WRAP_STRING(madwifi,mode)
 843 LUA_WRAP_STRING(madwifi,ssid)
 844 LUA_WRAP_STRING(madwifi,bssid)
 845 LUA_WRAP_STRING(madwifi,country)
 846 LUA_WRAP_LIST(madwifi,assoclist)
 847 LUA_WRAP_LIST(madwifi,txpwrlist)
 848 LUA_WRAP_LIST(madwifi,scanlist)
 849 LUA_WRAP_LIST(madwifi,freqlist)
 850 LUA_WRAP_LIST(madwifi,countrylist)
 851 LUA_WRAP_LIST(madwifi,hwmodelist)
 852 LUA_WRAP_LIST(madwifi,encryption)
 853 LUA_WRAP_LIST(madwifi,mbssid_support)
 854 #endif
 855
 856 #ifdef USE_NL80211
 857 /* NL80211 */
 858 LUA_WRAP_INT(nl80211,channel)
 859 LUA_WRAP_INT(nl80211,frequency)
 860 LUA_WRAP_INT(nl80211,txpower)
 861 LUA_WRAP_INT(nl80211,bitrate)
 862 LUA_WRAP_INT(nl80211,signal)
 863 LUA_WRAP_INT(nl80211,noise)
 864 LUA_WRAP_INT(nl80211,quality)
 865 LUA_WRAP_INT(nl80211,quality_max)
 866 LUA_WRAP_STRING(nl80211,mode)
 867 LUA_WRAP_STRING(nl80211,ssid)
 868 LUA_WRAP_STRING(nl80211,bssid)
 869 LUA_WRAP_STRING(nl80211,country)
 870 LUA_WRAP_LIST(nl80211,assoclist)
 871 LUA_WRAP_LIST(nl80211,txpwrlist)
 872 LUA_WRAP_LIST(nl80211,scanlist)
 873 LUA_WRAP_LIST(nl80211,freqlist)
 874 LUA_WRAP_LIST(nl80211,countrylist)
 875 LUA_WRAP_LIST(nl80211,hwmodelist)
 876 LUA_WRAP_LIST(nl80211,encryption)
 877 LUA_WRAP_LIST(nl80211,mbssid_support)
 878 #endif
 879
 880 /* Wext */
 881 LUA_WRAP_INT(wext,channel)
 882 LUA_WRAP_INT(wext,frequency)
 883 LUA_WRAP_INT(wext,txpower)
 884 LUA_WRAP_INT(wext,bitrate)
 885 LUA_WRAP_INT(wext,signal)
 886 LUA_WRAP_INT(wext,noise)
 887 LUA_WRAP_INT(wext,quality)
 888 LUA_WRAP_INT(wext,quality_max)
 889 LUA_WRAP_STRING(wext,mode)
 890 LUA_WRAP_STRING(wext,ssid)
 891 LUA_WRAP_STRING(wext,bssid)
 892 LUA_WRAP_STRING(wext,country)
 893 LUA_WRAP_LIST(wext,assoclist)
 894 LUA_WRAP_LIST(wext,txpwrlist)
 895 LUA_WRAP_LIST(wext,scanlist)
 896 LUA_WRAP_LIST(wext,freqlist)
 897 LUA_WRAP_LIST(wext,countrylist)
 898 LUA_WRAP_LIST(wext,hwmodelist)
 899 LUA_WRAP_LIST(wext,encryption)
 900 LUA_WRAP_LIST(wext,mbssid_support)
 901
 902 #ifdef USE_WL
 903 /* Broadcom table */
 904 static const luaL_reg R_wl[] = {
 905         LUA_REG(wl,channel),
 906         LUA_REG(wl,frequency),
 907         LUA_REG(wl,txpower),
 908         LUA_REG(wl,bitrate),
 909         LUA_REG(wl,signal),
 910         LUA_REG(wl,noise),
 911         LUA_REG(wl,quality),
 912         LUA_REG(wl,quality_max),
 913         LUA_REG(wl,mode),
 914         LUA_REG(wl,ssid),
 915         LUA_REG(wl,bssid),
 916         LUA_REG(wl,country),
 917         LUA_REG(wl,assoclist),
 918         LUA_REG(wl,txpwrlist),
 919         LUA_REG(wl,scanlist),
 920         LUA_REG(wl,freqlist),
 921         LUA_REG(wl,countrylist),
 922         LUA_REG(wl,hwmodelist),
 923         LUA_REG(wl,encryption),
 924         LUA_REG(wl,mbssid_support),
 925         { NULL, NULL }
 926 };
 927 #endif
 928
 929 #ifdef USE_MADWIFI
 930 /* Madwifi table */
 931 static const luaL_reg R_madwifi[] = {
 932         LUA_REG(madwifi,channel),
 933         LUA_REG(madwifi,frequency),
 934         LUA_REG(madwifi,txpower),
 935         LUA_REG(madwifi,bitrate),
 936         LUA_REG(madwifi,signal),
 937         LUA_REG(madwifi,noise),
 938         LUA_REG(madwifi,quality),
 939         LUA_REG(madwifi,quality_max),
 940         LUA_REG(madwifi,mode),
 941         LUA_REG(madwifi,ssid),
 942         LUA_REG(madwifi,bssid),
 943         LUA_REG(madwifi,country),
 944         LUA_REG(madwifi,assoclist),
 945         LUA_REG(madwifi,txpwrlist),
 946         LUA_REG(madwifi,scanlist),
 947         LUA_REG(madwifi,freqlist),
 948         LUA_REG(madwifi,countrylist),
 949         LUA_REG(madwifi,hwmodelist),
 950         LUA_REG(madwifi,encryption),
 951         LUA_REG(madwifi,mbssid_support),
 952         { NULL, NULL }
 953 };
 954 #endif
 955
 956 #ifdef USE_NL80211
 957 /* NL80211 table */
 958 static const luaL_reg R_nl80211[] = {
 959         LUA_REG(nl80211,channel),
 960         LUA_REG(nl80211,frequency),
 961         LUA_REG(nl80211,txpower),
 962         LUA_REG(nl80211,bitrate),
 963         LUA_REG(nl80211,signal),
 964         LUA_REG(nl80211,noise),
 965         LUA_REG(nl80211,quality),
 966         LUA_REG(nl80211,quality_max),
 967         LUA_REG(nl80211,mode),
 968         LUA_REG(nl80211,ssid),
 969         LUA_REG(nl80211,bssid),
 970         LUA_REG(nl80211,country),
 971         LUA_REG(nl80211,assoclist),
 972         LUA_REG(nl80211,txpwrlist),
 973         LUA_REG(nl80211,scanlist),
 974         LUA_REG(nl80211,freqlist),
 975         LUA_REG(nl80211,countrylist),
 976         LUA_REG(nl80211,hwmodelist),
 977         LUA_REG(nl80211,encryption),
 978         LUA_REG(nl80211,mbssid_support),
 979         { NULL, NULL }
 980 };
 981 #endif
 982
 983 /* Wext table */
 984 static const luaL_reg R_wext[] = {
 985         LUA_REG(wext,channel),
 986         LUA_REG(wext,frequency),
 987         LUA_REG(wext,txpower),
 988         LUA_REG(wext,bitrate),
 989         LUA_REG(wext,signal),
 990         LUA_REG(wext,noise),
 991         LUA_REG(wext,quality),
 992         LUA_REG(wext,quality_max),
 993         LUA_REG(wext,mode),
 994         LUA_REG(wext,ssid),
 995         LUA_REG(wext,bssid),
 996         LUA_REG(wext,country),
 997         LUA_REG(wext,assoclist),
 998         LUA_REG(wext,txpwrlist),
 999         LUA_REG(wext,scanlist),
1000         LUA_REG(wext,freqlist),
1001         LUA_REG(wext,countrylist),
1002         LUA_REG(wext,hwmodelist),
1003         LUA_REG(wext,encryption),
1004         LUA_REG(wext,mbssid_support),
1005         { NULL, NULL }
1006 };
1007
1008 /* Common */
1009 static const luaL_reg R_common[] = {
1010         { "type", iwinfo_L_type },
1011         { "__gc", iwinfo_L__gc  },
1012         { NULL, NULL }
1013 };
1014
1015
1016 LUALIB_API int luaopen_iwinfo(lua_State *L) {
1017         luaL_register(L, IWINFO_META, R_common);
1018
1019 #ifdef USE_WL
1020         luaL_newmetatable(L, IWINFO_WL_META);
1021         luaL_register(L, NULL, R_wl);
1022         lua_pushvalue(L, -1);
1023         lua_setfield(L, -2, "__index");
1024         lua_setfield(L, -2, "wl");
1025 #endif
1026
1027 #ifdef USE_MADWIFI
1028         luaL_newmetatable(L, IWINFO_MADWIFI_META);
1029         luaL_register(L, NULL, R_madwifi);
1030         lua_pushvalue(L, -1);
1031         lua_setfield(L, -2, "__index");
1032         lua_setfield(L, -2, "madwifi");
1033 #endif
1034
1035 #ifdef USE_NL80211
1036         luaL_newmetatable(L, IWINFO_NL80211_META);
1037         luaL_register(L, NULL, R_nl80211);
1038         lua_pushvalue(L, -1);
1039         lua_setfield(L, -2, "__index");
1040         lua_setfield(L, -2, "nl80211");
1041 #endif
1042
1043         luaL_newmetatable(L, IWINFO_WEXT_META);
1044         luaL_register(L, NULL, R_wext);
1045         lua_pushvalue(L, -1);
1046         lua_setfield(L, -2, "__index");
1047         lua_setfield(L, -2, "wext");
1048
1049         return 1;
1050 }