tools/examples/chibi/sniffer_wsbridge/main.c

   1 /**************************************************************************/
   2 /*!
   3     @file     main.c
   4     @author   K. Townsend (microBuilder.eu)
   5
   6     @section LICENSE
   7
   8     Software License Agreement (BSD License)
   9
  10     Copyright (c) 2010, microBuilder SARL
  11     All rights reserved.
  12
  13     Redistribution and use in source and binary forms, with or without
  14     modification, are permitted provided that the following conditions are met:
  15     1. Redistributions of source code must retain the above copyright
  16     notice, this list of conditions and the following disclaimer.
  17     2. Redistributions in binary form must reproduce the above copyright
  18     notice, this list of conditions and the following disclaimer in the
  19     documentation and/or other materials provided with the distribution.
  20     3. Neither the name of the copyright holders nor the
  21     names of its contributors may be used to endorse or promote products
  22     derived from this software without specific prior written permission.
  23
  24     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''AS IS'' AND ANY
  25     EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  26     WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  27     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER BE LIABLE FOR ANY
  28     DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  29     (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  30     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  31     ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  32     (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  33     SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  34 */
  35 /**************************************************************************/
  36 #include "projectconfig.h"
  37 #include "sysinit.h"
  38
  39 #include "core/gpio/gpio.h"
  40
  41 #if defined CFG_CHIBI
  42   #include <string.h>
  43   #include <stdlib.h>
  44   #include "drivers/chibi/chb.h"
  45   #include "drivers/chibi/chb_drvr.h"
  46   #include "core/uart/uart.h"
  47   static chb_rx_data_t rx_data;
  48 #endif
  49
  50 #ifdef CFG_PRINTF_USBCDC
  51   volatile unsigned int lastTick;
  52   #include "core/usbcdc/usb.h"
  53   #include "core/usbcdc/usbcore.h"
  54   #include "core/usbcdc/usbhw.h"
  55   #include "core/usbcdc/cdcuser.h"
  56   #include "core/usbcdc/cdc_buf.h"
  57 #endif
  58
  59 /**************************************************************************/
  60 /*!
  61     Use Chibi as a wireless sniffer and write all captured frames
  62     to UART or USB CDC for wsbridge to handle (see "tools/wsbridge")
  63
  64     projectconfig.h settings:
  65     --------------------------------------------------
  66     CFG_CHIBI             -> Enabled
  67     CFG_CHIBI_PROMISCUOUS -> 1
  68     CFG_CHIBI_BUFFERSIZE  -> 1024
  69 */
  70 /**************************************************************************/
  71 int main(void)
  72 {
  73   // Configure cpu and mandatory peripherals
  74   systemInit();
  75
  76   // Check if projectconfig.h is properly configured for this example
  77   #if !defined CFG_CHIBI
  78     #error "CFG_CHIBI must be enabled in projectconfig.h for this example"
  79   #endif
  80   #if CFG_CHIBI_PROMISCUOUS == 0
  81     #error "CFG_CHIBI_PROMISCUOUS must set to 1 in projectconfig.h for this example"
  82   #endif
  83   #if defined CFG_INTERFACE
  84     #error "CFG_INTERFACE must be disabled in projectconfig.h for this example"
  85   #endif
  86
  87   #if defined CFG_CHIBI && CFG_CHIBI_PROMISCUOUS != 0
  88     // Get a reference to the Chibi peripheral control block
  89     chb_pcb_t *pcb = chb_get_pcb();
  90
  91     // Wait for incoming frames and transmit the raw data over uart
  92     while(1)
  93     {
  94       // Check for incoming messages
  95       while (pcb->data_rcv)
  96       {
  97         // get the length of the data
  98         rx_data.len = chb_read(&rx_data);
  99         // make sure the length is nonzero
 100         if (rx_data.len)
 101         {
 102           // Enable LED to indicate message reception
 103           gpioSetValue (CFG_LED_PORT, CFG_LED_PIN, CFG_LED_ON);
 104
 105
 106           // Send raw data the to PC for processing using wsbridge
 107           uint8_t i;
 108           for (i=0; i<rx_data.len; i++)
 109           {
 110             #ifdef CFG_PRINTF_UART
 111               uartSendByte(rx_data.data[i]);
 112             #endif
 113             #ifdef CFG_PRINTF_USBCDC
 114                // ToDo: This really needs to be refactored!
 115               if (USB_Configuration)
 116               {
 117                 cdcBufferWrite(rx_data.data[i]);
 118                 // Check if we can flush the buffer now or if we need to wait
 119                 unsigned int currentTick = systickGetTicks();
 120                 if (currentTick != lastTick)
 121                 {
 122                   uint8_t frame[64];
 123                   uint32_t bytesRead = 0;
 124                   while (cdcBufferDataPending())
 125                   {
 126                     // Read 64 byte chunks until end of data
 127                     bytesRead = cdcBufferReadLen(frame, 64);
 128                     // debug_printf("%d,", bytesRead);
 129                     USB_WriteEP (CDC_DEP_IN, frame, bytesRead);
 130                     systickDelay(1);
 131                   }
 132                   lastTick = currentTick;
 133                 }
 134               }
 135             #endif
 136           }
 137
 138           // Disable LED
 139           gpioSetValue (CFG_LED_PORT, CFG_LED_PIN, CFG_LED_OFF);
 140         }
 141       }
 142     }
 143   #endif
 144
 145   return 0;
 146 }