libs/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 /* Determine type */
  22 static int iwinfo_L_type(lua_State *L)
  23 {
  24         const char *ifname = luaL_checkstring(L, 1);
  25
  26         if( madwifi_probe(ifname) )
  27                 lua_pushstring(L, "madwifi");
  28
  29         else if( wl_probe(ifname) )
  30                 lua_pushstring(L, "wl");
  31
  32         else if( wext_probe(ifname) )
  33                 lua_pushstring(L, "wext");
  34
  35         else
  36                 lua_pushnil(L);
  37
  38         return 1;
  39 }
  40
  41 /* Wrapper for assoclist */
  42 static int iwinfo_L_assoclist(lua_State *L, int (*func)(const char *, char *, int *))
  43 {
  44         int i, len;
  45         char rv[IWINFO_BUFSIZE];
  46         char macstr[18];
  47         const char *ifname = luaL_checkstring(L, 1);
  48         struct iwinfo_assoclist_entry *e;
  49
  50         lua_newtable(L);
  51         memset(rv, 0, sizeof(rv));
  52
  53         if( !(*func)(ifname, rv, &len) )
  54         {
  55                 for( i = 0; i < len; i += sizeof(struct iwinfo_assoclist_entry) )
  56                 {
  57                         e = (struct iwinfo_assoclist_entry *) &rv[i];
  58
  59                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
  60                                 e->mac[0], e->mac[1], e->mac[2],
  61                                 e->mac[3], e->mac[4], e->mac[5]);
  62
  63                         lua_newtable(L);
  64
  65                         lua_pushnumber(L, e->signal);
  66                         lua_setfield(L, -2, "signal");
  67
  68                         lua_pushnumber(L, e->noise);
  69                         lua_setfield(L, -2, "noise");
  70
  71                         lua_setfield(L, -2, macstr);
  72                 }
  73         }
  74
  75         return 1;
  76 }
  77
  78 /* Wrapper for tx power list */
  79 static int iwinfo_L_txpwrlist(lua_State *L, int (*func)(const char *, char *, int *))
  80 {
  81         int i, x, len;
  82         char rv[IWINFO_BUFSIZE];
  83         const char *ifname = luaL_checkstring(L, 1);
  84         struct iwinfo_txpwrlist_entry *e;
  85
  86         lua_newtable(L);
  87         memset(rv, 0, sizeof(rv));
  88
  89         if( !(*func)(ifname, rv, &len) )
  90         {
  91                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_txpwrlist_entry), x++ )
  92                 {
  93                         e = (struct iwinfo_txpwrlist_entry *) &rv[i];
  94
  95                         lua_newtable(L);
  96
  97                         lua_pushnumber(L, e->mw);
  98                         lua_setfield(L, -2, "mw");
  99
 100                         lua_pushnumber(L, e->dbm);
 101                         lua_setfield(L, -2, "dbm");
 102
 103                         lua_rawseti(L, -2, x);
 104                 }
 105         }
 106
 107         return 1;
 108 }
 109
 110 /* Wrapper for scan list */
 111 static int iwinfo_L_scanlist(lua_State *L, int (*func)(const char *, char *, int *))
 112 {
 113         int i, j, x, y, len;
 114         char rv[IWINFO_BUFSIZE];
 115         char macstr[18];
 116         const char *ifname = luaL_checkstring(L, 1);
 117         struct iwinfo_scanlist_entry *e;
 118
 119         lua_newtable(L);
 120         memset(rv, 0, sizeof(rv));
 121
 122         if( !(*func)(ifname, rv, &len) )
 123         {
 124                 for( i = 0, x = 1; i < len; i += sizeof(struct iwinfo_scanlist_entry), x++ )
 125                 {
 126                         e = (struct iwinfo_scanlist_entry *) &rv[i];
 127
 128                         lua_newtable(L);
 129
 130                         /* BSSID */
 131                         sprintf(macstr, "%02X:%02X:%02X:%02X:%02X:%02X",
 132                                 e->mac[0], e->mac[1], e->mac[2],
 133                                 e->mac[3], e->mac[4], e->mac[5]);
 134
 135                         lua_pushstring(L, macstr);
 136                         lua_setfield(L, -2, "bssid");
 137
 138                         /* ESSID */
 139                         if( e->ssid[0] )
 140                         {
 141                                 lua_pushstring(L, (char *) e->ssid);
 142                                 lua_setfield(L, -2, "ssid");
 143                         }
 144
 145                         /* Channel */
 146                         lua_pushinteger(L, e->channel);
 147                         lua_setfield(L, -2, "channel");
 148
 149                         /* Mode */
 150                         lua_pushstring(L, (char *) e->mode);
 151                         lua_setfield(L, -2, "mode");
 152
 153                         /* Quality, Signal */
 154                         lua_pushinteger(L, e->quality);
 155                         lua_setfield(L, -2, "quality");
 156
 157                         lua_pushinteger(L, e->quality_max);
 158                         lua_setfield(L, -2, "quality_max");
 159
 160                         lua_pushnumber(L, (e->signal - 0x100));
 161                         lua_setfield(L, -2, "signal");
 162
 163                         /* Crypto */
 164                         lua_pushboolean(L, (!e->crypto.wpa_version && e->crypto.enabled));
 165                         lua_setfield(L, -2, "wep");
 166
 167                         if( e->crypto.wpa_version )
 168                         {
 169                                 lua_pushinteger(L, e->crypto.wpa_version);
 170                                 lua_setfield(L, -2, "wpa");
 171
 172                                 lua_newtable(L);
 173                                 for( j = 0, y = 1; j < IW_IE_CYPHER_NUM; j++ )
 174                                 {
 175                                         if( e->crypto.group_ciphers & (1<<j) )
 176                                         {
 177                                                 lua_pushstring(L, iw_ie_cypher_name[j]);
 178                                                 lua_rawseti(L, -2, y++);
 179                                         }
 180                                 }
 181                                 lua_setfield(L, -2, "group_ciphers");
 182
 183                                 lua_newtable(L);
 184                                 for( j = 0, y = 1; j < IW_IE_CYPHER_NUM; j++ )
 185                                 {
 186                                         if( e->crypto.pair_ciphers & (1<<j) )
 187                                         {
 188                                                 lua_pushstring(L, iw_ie_cypher_name[j]);
 189                                                 lua_rawseti(L, -2, y++);
 190                                         }
 191                                 }
 192                                 lua_setfield(L, -2, "pair_ciphers");
 193
 194                                 lua_newtable(L);
 195                                 for( j = 0, y = 1; j < IW_IE_KEY_MGMT_NUM; j++ )
 196                                 {
 197                                         if( e->crypto.auth_suites & (1<<j) )
 198                                         {
 199                                                 lua_pushstring(L, iw_ie_key_mgmt_name[j]);
 200                                                 lua_rawseti(L, -2, y++);
 201                                         }
 202                                 }
 203                                 lua_setfield(L, -2, "auth_suites");
 204                         }
 205
 206                         lua_rawseti(L, -2, x);
 207                 }
 208         }
 209
 210         return 1;
 211 }
 212
 213
 214 /* Broadcom */
 215 LUA_WRAP_INT(wl,channel)
 216 LUA_WRAP_INT(wl,frequency)
 217 LUA_WRAP_INT(wl,bitrate)
 218 LUA_WRAP_INT(wl,signal)
 219 LUA_WRAP_INT(wl,noise)
 220 LUA_WRAP_INT(wl,quality)
 221 LUA_WRAP_INT(wl,quality_max)
 222 LUA_WRAP_INT(wl,mbssid_support)
 223 LUA_WRAP_STRING(wl,mode)
 224 LUA_WRAP_STRING(wl,ssid)
 225 LUA_WRAP_STRING(wl,bssid)
 226 LUA_WRAP_STRING(wl,enctype)
 227 LUA_WRAP_LIST(wl,assoclist)
 228 LUA_WRAP_LIST(wl,txpwrlist)
 229 LUA_WRAP_LIST(wl,scanlist)
 230
 231 /* Madwifi */
 232 LUA_WRAP_INT(madwifi,channel)
 233 LUA_WRAP_INT(madwifi,frequency)
 234 LUA_WRAP_INT(madwifi,bitrate)
 235 LUA_WRAP_INT(madwifi,signal)
 236 LUA_WRAP_INT(madwifi,noise)
 237 LUA_WRAP_INT(madwifi,quality)
 238 LUA_WRAP_INT(madwifi,quality_max)
 239 LUA_WRAP_INT(madwifi,mbssid_support)
 240 LUA_WRAP_STRING(madwifi,mode)
 241 LUA_WRAP_STRING(madwifi,ssid)
 242 LUA_WRAP_STRING(madwifi,bssid)
 243 LUA_WRAP_STRING(madwifi,enctype)
 244 LUA_WRAP_LIST(madwifi,assoclist)
 245 LUA_WRAP_LIST(madwifi,txpwrlist)
 246 LUA_WRAP_LIST(madwifi,scanlist)
 247
 248 /* Wext */
 249 LUA_WRAP_INT(wext,channel)
 250 LUA_WRAP_INT(wext,frequency)
 251 LUA_WRAP_INT(wext,bitrate)
 252 LUA_WRAP_INT(wext,signal)
 253 LUA_WRAP_INT(wext,noise)
 254 LUA_WRAP_INT(wext,quality)
 255 LUA_WRAP_INT(wext,quality_max)
 256 LUA_WRAP_INT(wext,mbssid_support)
 257 LUA_WRAP_STRING(wext,mode)
 258 LUA_WRAP_STRING(wext,ssid)
 259 LUA_WRAP_STRING(wext,bssid)
 260 LUA_WRAP_STRING(wext,enctype)
 261 LUA_WRAP_LIST(wext,assoclist)
 262 LUA_WRAP_LIST(wext,txpwrlist)
 263 LUA_WRAP_LIST(wext,scanlist)
 264
 265 /* Broadcom table */
 266 static const luaL_reg R_wl[] = {
 267         LUA_REG(wl,channel),
 268         LUA_REG(wl,frequency),
 269         LUA_REG(wl,bitrate),
 270         LUA_REG(wl,signal),
 271         LUA_REG(wl,noise),
 272         LUA_REG(wl,quality),
 273         LUA_REG(wl,quality_max),
 274         LUA_REG(wl,mode),
 275         LUA_REG(wl,ssid),
 276         LUA_REG(wl,bssid),
 277         LUA_REG(wl,enctype),
 278         LUA_REG(wl,assoclist),
 279         LUA_REG(wl,txpwrlist),
 280         LUA_REG(wl,scanlist),
 281         LUA_REG(wl,mbssid_support),
 282         { NULL, NULL }
 283 };
 284
 285 /* Madwifi table */
 286 static const luaL_reg R_madwifi[] = {
 287         LUA_REG(madwifi,channel),
 288         LUA_REG(madwifi,frequency),
 289         LUA_REG(madwifi,bitrate),
 290         LUA_REG(madwifi,signal),
 291         LUA_REG(madwifi,noise),
 292         LUA_REG(madwifi,quality),
 293         LUA_REG(madwifi,quality_max),
 294         LUA_REG(madwifi,mode),
 295         LUA_REG(madwifi,ssid),
 296         LUA_REG(madwifi,bssid),
 297         LUA_REG(madwifi,enctype),
 298         LUA_REG(madwifi,assoclist),
 299         LUA_REG(madwifi,txpwrlist),
 300         LUA_REG(madwifi,scanlist),
 301         LUA_REG(madwifi,mbssid_support),
 302         { NULL, NULL }
 303 };
 304
 305 /* Wext table */
 306 static const luaL_reg R_wext[] = {
 307         LUA_REG(wext,channel),
 308         LUA_REG(wext,frequency),
 309         LUA_REG(wext,bitrate),
 310         LUA_REG(wext,signal),
 311         LUA_REG(wext,noise),
 312         LUA_REG(wext,quality),
 313         LUA_REG(wext,quality_max),
 314         LUA_REG(wext,mode),
 315         LUA_REG(wext,ssid),
 316         LUA_REG(wext,bssid),
 317         LUA_REG(wext,enctype),
 318         LUA_REG(wext,assoclist),
 319         LUA_REG(wext,txpwrlist),
 320         LUA_REG(wext,scanlist),
 321         LUA_REG(wext,mbssid_support),
 322         { NULL, NULL }
 323 };
 324
 325 /* Common */
 326 static const luaL_reg R_common[] = {
 327         { "type", iwinfo_L_type },
 328         { NULL, NULL }
 329 };
 330
 331
 332 LUALIB_API int luaopen_iwinfo(lua_State *L) {
 333         luaL_register(L, IWINFO_META, R_common);
 334
 335         luaL_newmetatable(L, IWINFO_WL_META);
 336         luaL_register(L, NULL, R_wl);
 337         lua_pushvalue(L, -1);
 338         lua_setfield(L, -2, "__index");
 339         lua_setfield(L, -2, "wl");
 340
 341         luaL_newmetatable(L, IWINFO_MADWIFI_META);
 342         luaL_register(L, NULL, R_madwifi);
 343         lua_pushvalue(L, -1);
 344         lua_setfield(L, -2, "__index");
 345         lua_setfield(L, -2, "madwifi");
 346
 347         luaL_newmetatable(L, IWINFO_WEXT_META);
 348         luaL_register(L, NULL, R_wext);
 349         lua_pushvalue(L, -1);
 350         lua_setfield(L, -2, "__index");
 351         lua_setfield(L, -2, "wext");
 352
 353         return 1;
 354 }
 355