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( wl_probe(ifname) )
  27                 lua_pushstring(L, "wl");
  28
  29         else if( madwifi_probe(ifname) )
  30                 lua_pushstring(L, "madwifi");
  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, "mac");
 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                         /* Crypto */
 154                         lua_pushinteger(L, e->crypto.wpa_version);
 155                         lua_setfield(L, -2, "wpa_version");
 156
 157                         lua_newtable(L);
 158                         for( j = 0, y = 1; j < sizeof(e->crypto.group_ciphers); j++ )
 159                         {
 160                                 if( e->crypto.group_ciphers[j] )
 161                                 {
 162                                         lua_pushstring(L, (j < IW_IE_CYPHER_NUM)
 163                                                 ? iw_ie_cypher_name[j] : "Proprietary");
 164
 165                                         lua_rawseti(L, -2, y++);
 166                                 }
 167                         }
 168                         lua_setfield(L, -2, "group_ciphers");
 169
 170                         lua_newtable(L);
 171                         for( j = 0, y = 1; j < sizeof(e->crypto.pair_ciphers); j++ )
 172                         {
 173                                 if( e->crypto.pair_ciphers[j] )
 174                                 {
 175                                         lua_pushstring(L, (j < IW_IE_CYPHER_NUM)
 176                                                 ? iw_ie_cypher_name[j] : "Proprietary");
 177
 178                                         lua_rawseti(L, -2, y++);
 179                                 }
 180                         }
 181                         lua_setfield(L, -2, "pair_ciphers");
 182
 183                         lua_newtable(L);
 184                         for( j = 0, y = 1; j < sizeof(e->crypto.auth_suites); j++ )
 185                         {
 186                                 if( e->crypto.auth_suites[j] )
 187                                 {
 188                                         lua_pushstring(L, (j < IW_IE_KEY_MGMT_NUM)
 189                                                 ? iw_ie_key_mgmt_name[j] : "Proprietary");
 190
 191                                         lua_rawseti(L, -2, y++);
 192                                 }
 193                         }
 194                         lua_setfield(L, -2, "auth_suites");
 195
 196                         lua_rawseti(L, -2, x);
 197                 }
 198         }
 199
 200         return 1;
 201 }
 202
 203
 204 /* Broadcom */
 205 LUA_WRAP_INT(wl,channel)
 206 LUA_WRAP_INT(wl,frequency)
 207 LUA_WRAP_INT(wl,bitrate)
 208 LUA_WRAP_INT(wl,signal)
 209 LUA_WRAP_INT(wl,noise)
 210 LUA_WRAP_INT(wl,quality)
 211 LUA_WRAP_INT(wl,quality_max)
 212 LUA_WRAP_INT(wl,mbssid_support)
 213 LUA_WRAP_STRING(wl,mode)
 214 LUA_WRAP_STRING(wl,ssid)
 215 LUA_WRAP_STRING(wl,bssid)
 216 LUA_WRAP_STRING(wl,enctype)
 217 LUA_WRAP_LIST(wl,assoclist)
 218 LUA_WRAP_LIST(wl,txpwrlist)
 219 LUA_WRAP_LIST(wl,scanlist)
 220
 221 /* Madwifi */
 222 LUA_WRAP_INT(madwifi,channel)
 223 LUA_WRAP_INT(madwifi,frequency)
 224 LUA_WRAP_INT(madwifi,bitrate)
 225 LUA_WRAP_INT(madwifi,signal)
 226 LUA_WRAP_INT(madwifi,noise)
 227 LUA_WRAP_INT(madwifi,quality)
 228 LUA_WRAP_INT(madwifi,quality_max)
 229 LUA_WRAP_INT(madwifi,mbssid_support)
 230 LUA_WRAP_STRING(madwifi,mode)
 231 LUA_WRAP_STRING(madwifi,ssid)
 232 LUA_WRAP_STRING(madwifi,bssid)
 233 LUA_WRAP_STRING(madwifi,enctype)
 234 LUA_WRAP_LIST(madwifi,assoclist)
 235 LUA_WRAP_LIST(madwifi,txpwrlist)
 236 LUA_WRAP_LIST(madwifi,scanlist)
 237
 238 /* Wext */
 239 LUA_WRAP_INT(wext,channel)
 240 LUA_WRAP_INT(wext,frequency)
 241 LUA_WRAP_INT(wext,bitrate)
 242 LUA_WRAP_INT(wext,signal)
 243 LUA_WRAP_INT(wext,noise)
 244 LUA_WRAP_INT(wext,quality)
 245 LUA_WRAP_INT(wext,quality_max)
 246 LUA_WRAP_INT(wext,mbssid_support)
 247 LUA_WRAP_STRING(wext,mode)
 248 LUA_WRAP_STRING(wext,ssid)
 249 LUA_WRAP_STRING(wext,bssid)
 250 LUA_WRAP_STRING(wext,enctype)
 251 LUA_WRAP_LIST(wext,assoclist)
 252 LUA_WRAP_LIST(wext,txpwrlist)
 253 LUA_WRAP_LIST(wext,scanlist)
 254
 255 /* Broadcom table */
 256 static const luaL_reg R_wl[] = {
 257         LUA_REG(wl,channel),
 258         LUA_REG(wl,frequency),
 259         LUA_REG(wl,bitrate),
 260         LUA_REG(wl,signal),
 261         LUA_REG(wl,noise),
 262         LUA_REG(wl,quality),
 263         LUA_REG(wl,quality_max),
 264         LUA_REG(wl,mode),
 265         LUA_REG(wl,ssid),
 266         LUA_REG(wl,bssid),
 267         LUA_REG(wl,enctype),
 268         LUA_REG(wl,assoclist),
 269         LUA_REG(wl,txpwrlist),
 270         LUA_REG(wl,scanlist),
 271         LUA_REG(wl,mbssid_support),
 272         { NULL, NULL }
 273 };
 274
 275 /* Madwifi table */
 276 static const luaL_reg R_madwifi[] = {
 277         LUA_REG(madwifi,channel),
 278         LUA_REG(madwifi,frequency),
 279         LUA_REG(madwifi,bitrate),
 280         LUA_REG(madwifi,signal),
 281         LUA_REG(madwifi,noise),
 282         LUA_REG(madwifi,quality),
 283         LUA_REG(madwifi,quality_max),
 284         LUA_REG(madwifi,mode),
 285         LUA_REG(madwifi,ssid),
 286         LUA_REG(madwifi,bssid),
 287         LUA_REG(madwifi,enctype),
 288         LUA_REG(madwifi,assoclist),
 289         LUA_REG(madwifi,txpwrlist),
 290         LUA_REG(madwifi,scanlist),
 291         LUA_REG(madwifi,mbssid_support),
 292         { NULL, NULL }
 293 };
 294
 295 /* Wext table */
 296 static const luaL_reg R_wext[] = {
 297         LUA_REG(wext,channel),
 298         LUA_REG(wext,frequency),
 299         LUA_REG(wext,bitrate),
 300         LUA_REG(wext,signal),
 301         LUA_REG(wext,noise),
 302         LUA_REG(wext,quality),
 303         LUA_REG(wext,quality_max),
 304         LUA_REG(wext,mode),
 305         LUA_REG(wext,ssid),
 306         LUA_REG(wext,bssid),
 307         LUA_REG(wext,enctype),
 308         LUA_REG(wext,assoclist),
 309         LUA_REG(wext,txpwrlist),
 310         LUA_REG(wext,scanlist),
 311         LUA_REG(wext,mbssid_support),
 312         { NULL, NULL }
 313 };
 314
 315 /* Common */
 316 static const luaL_reg R_common[] = {
 317         { "type", iwinfo_L_type },
 318         { NULL, NULL }
 319 };
 320
 321
 322 LUALIB_API int luaopen_iwinfo(lua_State *L) {
 323         luaL_register(L, IWINFO_META, R_common);
 324
 325         luaL_newmetatable(L, IWINFO_WL_META);
 326         luaL_register(L, NULL, R_wl);
 327         lua_pushvalue(L, -1);
 328         lua_setfield(L, -2, "__index");
 329         lua_setfield(L, -2, "wl");
 330
 331         luaL_newmetatable(L, IWINFO_MADWIFI_META);
 332         luaL_register(L, NULL, R_madwifi);
 333         lua_pushvalue(L, -1);
 334         lua_setfield(L, -2, "__index");
 335         lua_setfield(L, -2, "madwifi");
 336
 337         luaL_newmetatable(L, IWINFO_WEXT_META);
 338         luaL_register(L, NULL, R_wext);
 339         lua_pushvalue(L, -1);
 340         lua_setfield(L, -2, "__index");
 341         lua_setfield(L, -2, "wext");
 342
 343         return 1;
 344 }
 345