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