common.c

/*
 * Copyright (c) Dan Harkins, 2008, 2009, 2010
 * Copyright (c) cozybit Inc., 2011
 *
 *  Copyright holder grants permission for redistribution and use in source
 *  and binary forms, with or without modification, provided that the
 *  following conditions are met:
 *     1. Redistribution of source code must retain the above copyright
 *        notice, this list of conditions, and the following disclaimer
 *        in all source files.
 *     2. Redistribution in binary form must retain the above copyright
 *        notice, this list of conditions, and the following disclaimer
 *        in the documentation and/or other materials provided with the
 *        distribution.
 *     3. All advertising materials and documentation mentioning features
 *	  or use of this software must display the following acknowledgement:
 *
 *        "This product includes software written by
 *         Dan Harkins (dharkins at lounge dot org)"
 *
 *  "DISCLAIMER OF LIABILITY
 *
 *  THIS SOFTWARE IS PROVIDED BY DAN HARKINS ``AS IS'' AND
 *  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 *  THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 *  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE INDUSTRIAL LOUNGE BE LIABLE
 *  FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 *  DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 *  SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 *  HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 *  OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 *  SUCH DAMAGE."
 *
 * This license and distribution terms cannot be changed. In other words,
 * this code cannot simply be copied and put under a different distribution
 * license (including the GNU public license).
 */

#include "common.h"

#include <stdarg.h>
#include <stddef.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <sys/time.h>

#include "ieee802_11.h"

unsigned int sae_debug_mask;

/*
 * parser_buffer()
 *      given an attribute/value pair in the form of attr=value
 *      return a pointer to the value
 */
int parse_buffer(char *buf, char **val) {
  char *ptr;

  *val = NULL;
  if (buf[0] == '#') {
    return 0;
  }
  ptr = buf;
  while (*ptr != '\n') {
    ptr++;
  }
  if (ptr == buf) {
    return 0;
  }
  *ptr = '\0';
  ptr = strstr(buf, "=");
  if (ptr == NULL) {
    return -1;
  }
  *ptr = '\0';
  do {
    ptr++;
  } while (*ptr == ' ');
  *val = ptr;

  return 1;
}

#define TIME_FMT "[%s.%06ld] "

void sae_debug(int level, const char *fmt, ...) {
  va_list argptr;
  struct timeval tv;
  struct tm cur_tm;
  char buf[9];

  if (sae_debug_mask & level) {
    gettimeofday(&tv, NULL);
    localtime_r(&tv.tv_sec, &cur_tm);
    strftime(buf, sizeof(buf), "%H:%M:%S", &cur_tm);

    va_start(argptr, fmt);
    printf(TIME_FMT, buf, tv.tv_usec);
    vfprintf(stdout, fmt, argptr);
    va_end(argptr);
  }
}

void sae_hexdump(
    int level,
    const char *label,
    const unsigned char *start,
    int len) {
  const unsigned char *pos;
  int i;
  char buf[9];
  struct timeval tv;
  struct tm cur_tm;

  gettimeofday(&tv, NULL);
  localtime_r(&tv.tv_sec, &cur_tm);
  strftime(buf, sizeof(buf), "%H:%M:%S", &cur_tm);

  if (sae_debug_mask & level) {
    fprintf(stdout, TIME_FMT "----------\n", buf, tv.tv_usec);
    fprintf(stdout, TIME_FMT "%s hexdump", buf, tv.tv_usec, label);
    pos = start;
    for (i = 0; i < len; i++) {
      if (!(i % 16))
        fprintf(stdout, "\n" TIME_FMT "%08x  ", buf, tv.tv_usec, i);
      fprintf(stdout, "%02x ", (unsigned char)*pos++);
    }
    fprintf(stdout, "\n" TIME_FMT "----------\n\n", buf, tv.tv_usec);
    fflush(stdout);
  }
  return;
}

void parse_ies(unsigned char *start, int len, struct info_elems *elems) {
  int left = len;
  unsigned char *pos = start;

  memset(elems, 0, sizeof(*elems));

  while (left >= 2) {
    unsigned char id, elen;

    id = *pos++;
    elen = *pos++;
    left -= 2;

    // fprintf(stderr, "parse_ies id=%d elen=%d\n", id, elen);

    if (elen > left)
      break;

    switch (id) {
      case IEEE80211_EID_SUPPORTED_RATES:
        elems->sup_rates = pos;
        elems->sup_rates_len = elen;
        break;
      case IEEE80211_EID_EXTENDED_SUP_RATES:
        elems->ext_rates = pos;
        elems->ext_rates_len = elen;
        break;
      case IEEE80211_EID_RSN:
        elems->rsn = pos;
        elems->rsn_len = elen;
        break;
      case IEEE80211_EID_HT_CAPABILITY:
        elems->ht_cap = pos;
        elems->ht_cap_len = elen;
        break;
      case IEEE80211_EID_HT_OPERATION:
        elems->ht_info = pos;
        elems->ht_info_len = elen;
        break;
      case IEEE80211_EID_VHT_CAPABILITY:
        elems->vht_cap = pos;
        elems->vht_cap_len = elen;
        break;
      case IEEE80211_EID_VHT_OPERATION:
        elems->vht_info = pos;
        elems->vht_info_len = elen;
        break;
      case IEEE80211_EID_MESH_ID:
        elems->mesh_id = pos;
        elems->mesh_id_len = elen;
        break;
      case IEEE80211_EID_MESH_PEERING:
        elems->mesh_peering = pos;
        elems->mesh_peering_len = elen;
        break;
      case IEEE80211_EID_MESH_CONFIG:
        elems->mesh_config = pos;
        elems->mesh_config_len = elen;
        break;
      case IEEE80211_EID_AMPE:
        elems->ampe = (struct ampe_ie *)pos;
        elems->ampe_len = elen;
        break;
      case IEEE80211_EID_MIC:
        elems->mic = pos;
        elems->mic_len = elen;
        /*  After the MIC there IEs there's the AMPE encrypted IE.  Stop here */
        return;
      default:
        break;
    }
    left -= elen;
    pos += elen;
  }
}