source: src/router/services/tools/site_survey_madwifi.c @ 10490

/*
 * site_survey_madwifi.c
 *
 * Copyright (C) 2006 Sebastian Gottschall <gottschall@dd-wrt.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * $Id:
 */

#include <sys/types.h>
#include <sys/file.h>
#include <sys/ioctl.h>
#include <sys/socket.h>

#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <ctype.h>
#include <getopt.h>
#include <err.h>
#include <shutils.h>

#include "wireless_copy.h"
#include "net80211/ieee80211.h"
#include "net80211/ieee80211_crypto.h"
#include "net80211/ieee80211_ioctl.h"

static int
copy_essid( char buf[], size_t bufsize, const u_int8_t * essid,
            size_t essid_len )
{
    const u_int8_t *p;
    int maxlen;
    int i;

    if( essid_len > bufsize )
        maxlen = bufsize;
    else
        maxlen = essid_len;
    /*
     * determine printable or not 
     */
    for( i = 0, p = essid; i < maxlen; i++, p++ )
    {
        if( *p < ' ' || *p > 0x7e )
            break;
    }
    if( i != maxlen )
    {                           /* not printable, print as hex */
        if( bufsize < 3 )
            return 0;
#if 0
        strlcpy( buf, "0x", bufsize );
#else
        strncpy( buf, "0x", bufsize );
#endif
        bufsize -= 2;
        p = essid;
        for( i = 0; i < maxlen && bufsize >= 2; i++ )
        {
            sprintf( &buf[2 + 2 * i], "%02x", *p++ );
            bufsize -= 2;
        }
        maxlen = 2 + 2 * i;
    }
    else
    {                           /* printable, truncate as needed */
        memcpy( buf, essid, maxlen );
    }
    if( maxlen != essid_len )
        memcpy( buf + maxlen - 3, "...", 3 );
    return maxlen;
}

#define sys_restart() kill(1, SIGHUP)
#define SITE_SURVEY_DB  "/tmp/site_survey"
#define SITE_SURVEY_NUM 256

int write_site_survey( void );
static int open_site_survey( void );
int write_site_survey( void );

struct site_survey_list
{
    unsigned char SSID[33];
    unsigned char BSSID[18];
    unsigned char channel;      /* Channel no. */
    short RSSI;                 /* receive signal strength (in dBm) */
    short phy_noise;            /* noise (in dBm) */
    unsigned short beacon_period;       /* units are Kusec */
    unsigned short capability;  /* Capability information */
    // unsigned char athcaps;
    unsigned char ENCINFO[32];  /* encryption info */
    int rate_count;             /* # rates in this set */
    unsigned char dtim_period;  /* DTIM period */
} site_survey_lists[SITE_SURVEY_NUM];

static const char *ieee80211_ntoa( const uint8_t mac[IEEE80211_ADDR_LEN] )
{
    static char a[18];
    int i;

    i = snprintf( a, sizeof( a ), "%02x:%02x:%02x:%02x:%02x:%02x",
                  mac[0], mac[1], mac[2], mac[3], mac[4], mac[5] );
    return ( i < 17 ? NULL : a );
}

int site_survey_main( int argc, char *argv[] )
{
    char *name = nvram_safe_get( "wl0_ifname" );
    unsigned char mac[20];
    int i = 0;
    char *dev = name;

    unlink( SITE_SURVEY_DB );
    int ap = 0, oldap = 0;

    unsigned char buf[sizeof( struct ieee80211req_scan_result )*256];
    char ssid[31];
    unsigned char *cp;
    int len;
    char *sta = nvram_safe_get( "wifi_display" );

    eval( "iwlist", sta, "scan" );
    len =
        do80211priv( sta, IEEE80211_IOCTL_SCAN_RESULTS, buf, sizeof( buf ) );

    if( len == -1 )
        fprintf( stderr, "unable to get scan results" );
    if( len < sizeof( struct ieee80211req_scan_result ) )
        return;
    cp = buf;
    do
    {
        struct ieee80211req_scan_result *sr;
        unsigned char *vp;
        char ssid[14];

        sr = ( struct ieee80211req_scan_result * )cp;
        vp = ( u_int8_t * ) ( sr + 1 );
        memset( ssid, 0, sizeof( ssid ) );
        strncpy( site_survey_lists[i].SSID, vp, sr->isr_ssid_len );
        strcpy( site_survey_lists[i].BSSID, ieee80211_ntoa( sr->isr_bssid ) );
        site_survey_lists[i].channel = ieee80211_mhz2ieee( sr->isr_freq );
        int noise = 256;

        noise -= ( int )sr->isr_noise;
        site_survey_lists[i].phy_noise = -noise;
        if( sr->isr_noise == 0 )
        {
            site_survey_lists[i].phy_noise = -95;
        }
        site_survey_lists[i].RSSI =
            ( int )site_survey_lists[i].phy_noise + ( int )sr->isr_rssi;
        site_survey_lists[i].capability = sr->isr_capinfo;
        // site_survey_lists[i].athcaps = sr->isr_athflags;
        site_survey_lists[i].rate_count = sr->isr_nrates;
        cp += sr->isr_len, len -= sr->isr_len;
        i++;
    }
    while( len >= sizeof( struct ieee80211req_scan_result ) );

    write_site_survey(  );
    open_site_survey(  );
    for( i = 0;
         i < SITE_SURVEY_NUM && site_survey_lists[i].BSSID[0]
         && site_survey_lists[i].channel != 0; i++ )
    {
        fprintf( stderr,
                 "[%2d] SSID[%20s] BSSID[%s] channel[%2d] rssi[%d] noise[%d] beacon[%d] cap[%x] dtim[%d] rate[%d]\n",
                 i, site_survey_lists[i].SSID, site_survey_lists[i].BSSID,
                 site_survey_lists[i].channel, site_survey_lists[i].RSSI,
                 site_survey_lists[i].phy_noise,
                 site_survey_lists[i].beacon_period,
                 site_survey_lists[i].capability,
                 site_survey_lists[i].dtim_period,
                 site_survey_lists[i].rate_count );
    }

    return 0;
}

int write_site_survey( void )
{
    FILE *fp;

    if( ( fp = fopen( SITE_SURVEY_DB, "w" ) ) )
    {
        fwrite( &site_survey_lists[0], sizeof( site_survey_lists ), 1, fp );
        fclose( fp );
        return 0;
    }
    return 1;
}

static int open_site_survey( void )
{
    FILE *fp;

    bzero( site_survey_lists, sizeof( site_survey_lists ) );

    if( ( fp = fopen( SITE_SURVEY_DB, "r" ) ) )
    {
        fread( &site_survey_lists[0], sizeof( site_survey_lists ), 1, fp );
        fclose( fp );
        return 1;
    }
    return 0;
}