vcs.maemo.org Git - wifihood/blob - wifiscand/wifiscand.c

   1
   2 #include <iwlib.h>
   3
   4 #ifdef HAVE_LIBOSSO
   5 #include <libosso.h>
   6 #else
   7 #include <glib.h>
   8 #include <dbus/dbus.h>
   9
  10 #define WIFISCAN_INTROSPECTION "<!DOCTYPE node PUBLIC \"-//freedesktop//DTD D-BUS Object Introspection 1.0//EN\"\n\
  11          \"http://www.freedesktop.org/standards/dbus/1.0/introspect.dtd\">\n\
  12 <node name=\"/org/javiplx/wifiscan\">\n\
  13   <interface name=\"org.javiplx.wifiscan\">\n\
  14     <method name=\"wakeup\">\n\
  15       <arg name=\"result\" type=\"s\" direction=\"out\" />\n\
  16     </method>\n\
  17     <method name=\"start\">\n\
  18       <arg name=\"result\" type=\"s\" direction=\"out\" />\n\
  19     </method>\n\
  20     <method name=\"scan\">\n\
  21       <arg name=\"result\" type=\"s\" direction=\"out\" />\n\
  22     </method>\n\
  23     <method name=\"stop\">\n\
  24       <arg name=\"result\" type=\"s\" direction=\"out\" />\n\
  25     </method>\n\
  26   </interface>\n\
  27 </node>\n"
  28
  29 #define OSSO_OK 0
  30 #define OSSO_ERROR 1
  31
  32 #define osso_context_t DBusConnection
  33 typedef struct osso_rpc_t_values {
  34   int i;
  35   char *s;
  36   } osso_rpc_t_values;
  37 typedef struct osso_rpc_t {
  38   int type;
  39   osso_rpc_t_values value;
  40   } osso_rpc_t;
  41 #endif
  42
  43 #define WIFISCAND_VERSION_STRING  "1.1"
  44
  45 #define OSSO_NAME    "wifiscan"
  46 #define OSSO_SERVICE "org.javiplx."OSSO_NAME
  47 #define OSSO_OBJECT  "/org/javiplx/"OSSO_NAME
  48 #define OSSO_IFACE   "org.javiplx."OSSO_NAME
  49
  50
  51 typedef struct {
  52         char          mac[18];
  53         int           rssi;
  54         int           noise;
  55 } ScanInfo;
  56
  57
  58 /* --------------------------------------------------------------------------- WirelessInterface */
  59 typedef struct {
  60         wireless_info info;
  61         char*         ifname;
  62         int           sock;
  63 } WirelessInterface;
  64
  65 /* ------------------------------------------------------------------ WirelessInterface_ScanItem */
  66
  67 void WirelessInterface_ScanItem(struct wireless_scan* ap, ScanInfo **scaninfo) {
  68
  69                         *scaninfo = (ScanInfo*) malloc( sizeof(ScanInfo) );
  70                         memset(*scaninfo, 0, sizeof(ScanInfo) );
  71
  72                         iw_sawap_ntop( &ap->ap_addr, (*scaninfo)->mac);
  73
  74                         (*scaninfo)->rssi = ap->stats.qual.level - 0x100;
  75                         (*scaninfo)->noise = ap->stats.qual.level - 0x100;
  76
  77 }
  78
  79 /* ---------------------------------------------------------------------- */
  80
  81 void* makeScan(WirelessInterface* wlan, gchar *message) {
  82         wireless_scan_head context;
  83         iwrange        range;
  84         int delay = 1;
  85
  86         ScanInfo        **scan = NULL;
  87
  88         iw_get_range_info(wlan->sock, wlan->ifname, &range);
  89
  90         /* We don't call iw_scan to allow fine control of delays and timeouts */
  91         context.result = NULL;
  92         context.retry = 0;
  93
  94         while(delay>0) {
  95                 delay = iw_process_scan(wlan->sock, wlan->ifname, range.we_version_compiled, &context);
  96                 if(delay < 0) break;
  97                 usleep(delay * 1000);
  98                 }
  99
 100         if ( delay == 0 ) {
 101
 102                 scan = (ScanInfo**) malloc( 25 * sizeof(ScanInfo *) );
 103                 memset(scan, 0, 25 * sizeof(ScanInfo *) );
 104                 *scan = NULL;
 105                 int i;
 106                 for (i=0; i<25; i++) *(scan+i) = NULL;
 107
 108                 i = 0;
 109                 ScanInfo *sc = NULL;
 110
 111                 /* Extract values */
 112                 struct wireless_scan *ap = context.result;
 113                 while ( ap != NULL && i < 25 ) {
 114                                 WirelessInterface_ScanItem(ap, &sc);
 115                                         *(scan+i) = sc;
 116                                         ap = ap->next;
 117                                         i++;
 118                         }
 119
 120         } else {
 121                 message = "Unknown error";
 122         }
 123
 124         return scan;
 125 }
 126
 127 static void refresh(WirelessInterface* coso) {
 128         struct iwreq wrq;
 129
 130         iw_get_basic_config(coso->sock, coso->ifname, &(coso->info.b));
 131         iw_get_range_info(coso->sock, coso->ifname, &(coso->info.range));
 132
 133         iw_get_ext(coso->sock, coso->ifname, SIOCGIWRATE, &wrq);
 134         memcpy(&(coso->info.bitrate), &wrq.u.bitrate, sizeof(iwparam));
 135
 136         iw_get_ext(coso->sock, coso->ifname, SIOCGIWAP, &wrq);
 137         memcpy(&(coso->info.ap_addr), &wrq.u.ap_addr, sizeof (sockaddr));
 138
 139         iw_get_stats(
 140                 coso->sock, coso->ifname, &(coso->info.stats),
 141                 &(coso->info.range), coso->info.has_range
 142         );
 143 }
 144
 145 /* Application UI data struct */
 146 typedef struct _AppData AppData;
 147 struct _AppData {
 148     WirelessInterface iface;
 149     osso_context_t *osso_context;
 150 };
 151
 152 #ifdef HAVE_LIBOSSO
 153 static GMainLoop *event_loop = NULL;
 154 #endif
 155 static short int start_flags = 0;
 156
 157 /* Callback for normal D-BUS messages */
 158 gint dbus_req_handler(const gchar * interface, const gchar * method,
 159                       GArray * arguments, gpointer data,
 160                       osso_rpc_t * retval)
 161 {
 162     AppData *appdata;
 163     appdata = (AppData *) data;
 164
 165     retval->type = DBUS_TYPE_STRING;
 166     retval->value.s = (gchar*) malloc( sizeof(gchar *) );
 167     retval->value.s[0] = '\0';
 168
 169 #ifndef HAVE_LIBOSSO
 170     if ( strcmp(method,"Introspect")==0 ) {
 171         retval->value.s = (gchar *) realloc(retval->value.s,630*sizeof(gchar *));
 172         snprintf(retval->value.s,strlen(WIFISCAN_INTROSPECTION),WIFISCAN_INTROSPECTION);
 173         return OSSO_OK;
 174     }
 175 #endif
 176
 177     if ( strcmp(method,"wakeup")==0 ) {
 178         retval->value.s = (gchar *) realloc(retval->value.s,16*sizeof(gchar *));
 179         snprintf(retval->value.s,16,"WifiScand ready");
 180         return OSSO_OK;
 181     }
 182
 183     if ( strcmp(method,"start")==0 ) {
 184
 185         if( (appdata->iface.sock=iw_sockets_open()) < 0) {
 186             retval->value.s = (gchar *) realloc(retval->value.s,33*sizeof(gchar *));
 187             snprintf(retval->value.s,33,"Failure in socket initialization");
 188             return OSSO_ERROR;
 189         }
 190
 191         /* Get range stuff */
 192         iwrange        range;
 193         int has_range = (iw_get_range_info(appdata->iface.sock, appdata->iface.ifname, &range) >= 0);
 194
 195         /* Check if the interface could support scanning. */
 196         if((!has_range) || (range.we_version_compiled < 14)) {
 197             retval->value.s = (gchar *) realloc(retval->value.s,35*sizeof(gchar *));
 198             snprintf(retval->value.s,35,"Interface doesn't support scanning");
 199             return OSSO_ERROR;
 200         }
 201
 202         struct ifreq frq;
 203         strncpy(frq.ifr_name, appdata->iface.ifname, IFNAMSIZ);
 204         if(ioctl(appdata->iface.sock, SIOCGIFFLAGS, &frq)) {
 205             retval->value.s = (gchar *) realloc(retval->value.s,28*sizeof(gchar *));
 206             snprintf(retval->value.s,28,"Cannot get interface status");
 207             return OSSO_ERROR;
 208         }
 209
 210         start_flags = frq.ifr_flags;
 211         frq.ifr_flags |= IFF_UP | IFF_RUNNING;
 212
 213         if(ioctl(appdata->iface.sock, SIOCSIFFLAGS, &frq)) {
 214             retval->value.s = (gchar *) realloc(retval->value.s,27*sizeof(gchar *));
 215             snprintf(retval->value.s,27,"Cannot set interface state");
 216             return OSSO_ERROR;
 217         }
 218
 219         refresh(&appdata->iface);
 220         retval->value.s = (gchar *) realloc(retval->value.s,22*sizeof(gchar *));
 221         snprintf(retval->value.s,22,"Interface initialized");
 222         return OSSO_OK;
 223     }
 224
 225     if ( strcmp(method,"stop")==0 ) {
 226         struct ifreq frq;
 227         strncpy(frq.ifr_name, appdata->iface.ifname, IFNAMSIZ);
 228         if(!ioctl(appdata->iface.sock, SIOCGIFFLAGS, &frq)) {
 229             frq.ifr_flags = start_flags;
 230             if(!ioctl(appdata->iface.sock, SIOCSIFFLAGS, &frq))
 231                 refresh(&appdata->iface);
 232         }
 233         iw_sockets_close(appdata->iface.sock);
 234         appdata->iface.sock = 0;
 235 #ifdef HAVE_LIBOSSO
 236         osso_deinitialize(appdata->osso_context);
 237         /* Instead of exiting, signaling finish to main loop could be better
 238         retval->value.s = (gchar *) realloc(retval->value.s,34*sizeof(gchar *));
 239         snprintf(retval->value.s,34,"Interface moved to original state");
 240         */
 241         exit(0);
 242 #else
 243         return OSSO_OK;
 244 #endif
 245     }
 246
 247     if ( strcmp(method,"scan")==0 ) {
 248
 249         ScanInfo **scan = (ScanInfo **) makeScan(&appdata->iface,retval->value.s);
 250         if(scan == NULL) {
 251             retval->value.s = (gchar *) realloc(retval->value.s,64*sizeof(gchar *));
 252             snprintf(retval->value.s,64,"ERROR");
 253             return OSSO_ERROR;
 254         }
 255
 256         int i;
 257         for (i=0; i<25&&*(scan+i)!=NULL; i++) {
 258             ScanInfo* sc = *(scan+i);
 259             retval->value.s = (gchar *) realloc(retval->value.s,23*(i+1)*sizeof(gchar *));
 260             if ( retval->value.s == NULL ) {
 261                 retval->value.s = "Error allocating memory";
 262                 return OSSO_ERROR;
 263             }
 264             sprintf(retval->value.s+strlen(retval->value.s),"%s:%d ",sc->mac,sc->rssi);
 265         }
 266
 267         retval->value.s[strlen(retval->value.s)-1]  = '\0';
 268         return OSSO_OK;
 269     }
 270
 271     retval->value.s = (gchar *) realloc(retval->value.s,64*sizeof(gchar *));
 272     snprintf(retval->value.s,64,"Unknown method");
 273     return OSSO_ERROR;
 274 }
 275
 276 #ifndef HAVE_LIBOSSO
 277
 278 dbus_bool_t stopped_service = FALSE;
 279
 280 static DBusObjectPathVTable *vtable = NULL;
 281
 282 int dbus_set_cb_f( DBusConnection *context,
 283                 const char *service, const char *object,
 284                 const char *interface, void *handler, // FIXME : Set proper type for handler
 285                 void *user_data) {
 286
 287         /* First, we prepare the complex dbus handler structures */
 288         DBusObjectPathVTable *vtable = malloc( sizeof(DBusObjectPathVTable) );
 289         memset(vtable , '\0', sizeof(DBusObjectPathVTable) );
 290         vtable->message_function = (DBusObjectPathMessageFunction) handler; // FIXME : Aqui va el nuevo handler/wrapper
 291
 292         if (!dbus_connection_register_fallback(context, object, vtable, user_data)) {
 293                 return OSSO_ERROR;
 294                 }
 295
 296         DBusError error;
 297         dbus_error_init(&error);
 298
 299         dbus_bus_request_name (context, service, 0, &error);
 300         if ( dbus_error_is_set(&error) ) {
 301                 dbus_error_free (&error);
 302                 return OSSO_ERROR;
 303                 }
 304
 305         dbus_error_free (&error);
 306         return OSSO_OK;
 307 }
 308
 309 void dbus_deinitialize( DBusConnection *context ) {
 310         free(vtable); vtable = NULL;
 311         dbus_connection_unref(context);
 312         dbus_shutdown();
 313 }
 314
 315 static DBusHandlerResult handler_wrapper (DBusConnection  *connection,
 316                    DBusMessage     *message,
 317                    void            *data) {
 318
 319         gint retcode;
 320         osso_rpc_t *retval = malloc(sizeof(osso_rpc_t));
 321
 322         if ( dbus_message_get_type(message) == DBUS_MESSAGE_TYPE_METHOD_CALL ) {
 323                 retcode = dbus_req_handler(dbus_message_get_interface(message),
 324                                 dbus_message_get_member(message),
 325                                 NULL, data, retval);
 326
 327                 if ( strcmp(dbus_message_get_member(message),"stop")==0 )
 328                         stopped_service = TRUE;
 329
 330                 if ( retval->value.s != NULL ) {
 331                         DBusMessage *reply = dbus_message_new_method_return (message);
 332                         if (reply == NULL)
 333                                 exit(0);
 334                         if (!dbus_message_append_args (reply, DBUS_TYPE_STRING, &retval->value.s, DBUS_TYPE_INVALID))
 335                                 exit(0);
 336                         if (!dbus_connection_send (connection, reply, NULL))
 337                                 exit(0);
 338                         dbus_message_unref (reply);
 339                         }
 340
 341                 }
 342
 343         return DBUS_HANDLER_RESULT_HANDLED;
 344 }
 345
 346 #endif
 347
 348 int main( void ) {
 349
 350         osso_context_t *osso_context;
 351 #ifndef HAVE_LIBOSSO
 352         DBusError error;
 353 #endif
 354
 355         /* Initialize maemo application */
 356 #ifdef HAVE_LIBOSSO
 357         osso_context = osso_initialize(OSSO_NAME, WIFISCAND_VERSION_STRING, TRUE, NULL);
 358 #else
 359         dbus_error_init(&error);
 360         osso_context = dbus_bus_get(DBUS_BUS_SESSION, &error);
 361 #endif
 362
 363         /* Check that initialization was ok */
 364         if (osso_context == NULL) {
 365 #ifndef HAVE_LIBOSSO
 366                 fprintf (stderr, "*** Failed to open connection to activating message bus: %s\n", error.message);
 367                 dbus_error_free (&error);
 368 #endif
 369                 exit(OSSO_ERROR);
 370                 }
 371
 372         /* Create AppData */
 373         AppData *appdata;
 374         appdata = g_new0(AppData, 1);
 375         appdata->osso_context = osso_context;
 376
 377         memset(&(appdata->iface.info), 0, sizeof(wireless_info));
 378         appdata->iface.ifname = "wlan0";
 379         appdata->iface.sock   = 0;
 380
 381         /* Add handler for hello D-BUS messages */
 382 #ifdef HAVE_LIBOSSO
 383         osso_return_t result = osso_rpc_set_cb_f(osso_context, OSSO_SERVICE, OSSO_OBJECT, OSSO_IFACE, dbus_req_handler, appdata);
 384 #else
 385         int result = dbus_set_cb_f(osso_context, OSSO_SERVICE, OSSO_OBJECT, OSSO_IFACE, handler_wrapper, appdata);
 386 #endif
 387
 388         if (result != OSSO_OK) {
 389 #ifdef HAVE_LIBOSSO
 390                 osso_system_note_infoprint(appdata->osso_context, "Failure while setting OSSO callback", NULL);
 391 #endif
 392                 return OSSO_ERROR;
 393         }
 394
 395         /* INITIALIZATION FINISH */
 396
 397 #ifdef HAVE_LIBOSSO
 398         event_loop = g_main_loop_new(NULL, FALSE);
 399         g_main_loop_run(event_loop);
 400 #else
 401         while (dbus_connection_read_write_dispatch(osso_context, -1) && stopped_service==FALSE) {}
 402 #endif
 403
 404         /* Deinitialize OSSO */
 405 #ifdef HAVE_LIBOSSO
 406         osso_deinitialize(osso_context);
 407 #else
 408         dbus_deinitialize(osso_context);
 409 #endif
 410
 411         exit(0);
 412 }
 413