Project

General

Profile

Statistics
| Revision:

root / branches / autonomous_recharging / code / projects / libwireless / lib / xbee.c @ 793

History | View | Annotate | Download (22.1 KB)

1
/**
2
 * Copyright (c) 2007 Colony Project
3
 *
4
 * Permission is hereby granted, free of charge, to any person
5
 * obtaining a copy of this software and associated documentation
6
 * files (the "Software"), to deal in the Software without
7
 * restriction, including without limitation the rights to use,
8
 * copy, modify, merge, publish, distribute, sublicense, and/or sell
9
 * copies of the Software, and to permit persons to whom the
10
 * Software is furnished to do so, subject to the following
11
 * conditions:
12
 *
13
 * The above copyright notice and this permission notice shall be
14
 * included in all copies or substantial portions of the Software.
15
 *
16
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
18
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
20
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
21
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
23
 * OTHER DEALINGS IN THE SOFTWARE.
24
 **/
25

    
26
/**
27
 * @file xbee.c
28
 * @brief XBee Interface
29
 *
30
 * Implementation of low level communication with the XBee in API mode.
31
 *
32
 * @author Brian Coltin, Colony Project, CMU Robotics Club
33
 **/
34

    
35
#include "xbee.h"
36
#include "wl_defs.h"
37

    
38
#ifndef ROBOT
39

    
40
#include <fcntl.h>
41
#include <unistd.h>
42
#include <pthread.h>
43
#include <errno.h>
44
#include <termios.h>
45

    
46
#else
47

    
48
#include <serial.h>
49
#include <avr/interrupt.h>
50

    
51
#endif
52

    
53
#include <stdio.h>
54
#include <stdlib.h>
55
#include <string.h>
56

    
57
#define XBEE_FRAME_START 0x7E
58
#define XBEE_GET_PACKET_TIMEOUT 1000
59

    
60
/*Frame Types*/
61
#define XBEE_FRAME_STATUS 0x8A
62
#define XBEE_FRAME_AT_COMMAND 0x08
63
#define XBEE_FRAME_AT_COMMAND_RESPONSE 0x88
64
#define XBEE_FRAME_TX_REQUEST_64 0x00
65
#define XBEE_FRAME_TX_REQUEST_16 0x01
66
#define XBEE_FRAME_TX_STATUS XBEE_TX_STATUS
67
#define XBEE_FRAME_RX_64 0x80
68
#define XBEE_FRAME_RX_16 XBEE_RX
69

    
70
/*Internal Function Prototypes*/
71

    
72
/*I/O Functions*/
73
static int xbee_send(char* buf, int size);
74
static int xbee_send_string(char* c);
75

    
76
#ifndef ROBOT
77
static int xbee_read(char* buf, int size);
78
#endif
79

    
80
/*Command Mode Functions
81
 * Called during initialization.
82
 */
83
static int xbee_enter_command_mode(void);
84
static int xbee_exit_command_mode(void);
85
static int xbee_enter_api_mode(void);
86
static int xbee_exit_api_mode(void);
87
static int xbee_wait_for_string(char* s, int len);
88
static int xbee_wait_for_ok(void);
89

    
90
/*API Mode Functions*/
91

    
92
static int xbee_handle_packet(char* packet, int len);
93
static void xbee_handle_at_command_response(char* command, char result, char* extra, int extraLen);
94
static void xbee_handle_status(char status);
95
static int xbee_verify_checksum(char* packet, int len);
96
static char xbee_compute_checksum(char* packet, int len);
97
static int xbee_send_frame(char* buf, int len);
98
static int xbee_send_read_at_command(char* command);
99
static int xbee_send_modify_at_command(char* command, char* value);
100

    
101
/*Global Variables*/
102

    
103
#ifndef ROBOT
104
static char* xbee_com_port = XBEE_PORT_DEFAULT;
105
static int xbee_stream;
106
static pthread_t* xbee_listen_thread;
107
#endif
108

    
109
// TODO: is this a good size?
110
#define XBEE_BUFFER_SIZE        128
111
#define PACKET_BUFFER_SIZE        108
112
// a buffer for data received from the XBee
113
char arrival_buf[XBEE_BUFFER_SIZE];
114
// location of last unread byte in buffer
115
volatile int buffer_last = 0;
116
// first unread byte in buffer
117
volatile int buffer_first = 0;
118

    
119

    
120
//used to store packets as they are read
121
static char xbee_buf[PACKET_BUFFER_SIZE];
122
static int currentBufPos = 0;
123

    
124
//XBee status
125
static unsigned int xbee_panID = XBEE_PAN_DEFAULT;
126
static unsigned int xbee_pending_panID = XBEE_PAN_DEFAULT;
127
static int xbee_channel = XBEE_CHANNEL_DEFAULT;
128
static int xbee_pending_channel = XBEE_CHANNEL_DEFAULT;
129
static volatile unsigned int xbee_address = 0;
130

    
131
/*Function Implementations*/
132

    
133
#ifdef ROBOT
134

    
135
/**
136
 * Interrupt for the robot. Adds bytes received from the xbee
137
 * to the buffer.
138
 **/
139
#ifndef FIREFLY
140
ISR(USART1_RX_vect)
141
{
142
        char c = UDR1;
143
        arrival_buf[buffer_last] = c;
144
        int t = buffer_last + 1;
145
        if (t == XBEE_BUFFER_SIZE)
146
                t = 0;
147
        if (t == buffer_first)
148
        {
149
                WL_DEBUG_PRINT("\nOut of space in buffer.\n");
150
        }
151
        buffer_last = t;
152
}
153
#else
154
SIGNAL(SIG_USART0_RECV)
155
{
156
        char c = UDR0;
157
        arrival_buf[buffer_last] = c;
158
        int t = buffer_last + 1;
159
        if (t == XBEE_BUFFER_SIZE)
160
                t = 0;
161
        if (t == buffer_first)
162
        {
163
                WL_DEBUG_PRINT("Out of space in buffer.\n");
164
        }
165
        buffer_last = t;
166
}
167
#endif
168

    
169
#else
170

    
171
// Computer code
172

    
173
/**
174
 * Thread that listens to the xbee.
175
 **/
176
static void* listen_to_xbee(void* x)
177
{
178
        char c;
179
        while (1)
180
        {
181
                if (xbee_read(&c, 1) != 0) {
182
                        WL_DEBUG_PRINT("xbee_read failed.\n");
183
                        return NULL;
184
                }
185

    
186
                arrival_buf[buffer_last] = c;
187
                int t = buffer_last + 1;
188
                if (t == XBEE_BUFFER_SIZE)
189
                        t = 0;
190
                if (t == buffer_first)
191
                {
192
                        WL_DEBUG_PRINT("Out of space in buffer.\n");
193
                }
194
                buffer_last = t;
195

    
196
                usleep(1000);
197
        }
198

    
199
        return NULL;
200
}
201

    
202
#endif
203

    
204
/**
205
 * Initializes the XBee library so that other functions may be used.
206
 **/
207
int xbee_lib_init()
208
{
209
        WL_DEBUG_PRINT("in xbee_init\n");
210
#ifdef ROBOT
211

    
212
        //enable the receiving interrupt
213
#ifdef FIREFLY
214
        UCSR0B |= _BV(RXCIE) | _BV(RXEN);
215
#else
216
#ifdef BAYBOARD
217
        UCSR1B |= _BV(RXCIE1);
218
#else
219
        UCSR1B |= _BV(RXCIE);
220
#endif
221
#endif
222
        sei();
223
#else
224
        xbee_stream = open(xbee_com_port, O_RDWR);
225
        if (xbee_stream == -1/* || lockf(xbee_stream, F_TEST, 0) != 0*/)
226
        {
227
                WL_DEBUG_PRINT("Failed to open connection to XBee on port ");
228
                WL_DEBUG_PRINT_INT(xbee_com_port);
229
                WL_DEBUG_PRINT(".\n");
230
                return -1;
231
        } else {
232
          WL_DEBUG_PRINT("Successfully opened connection to XBee on port ");
233
                WL_DEBUG_PRINT_INT(xbee_com_port);
234
                WL_DEBUG_PRINT(".\n");
235
        }
236

    
237
        // set baud rate, etc. correctly
238
        struct termios options;
239

    
240
        tcgetattr(xbee_stream, &options);
241
        cfsetispeed(&options, B9600);
242
        cfsetospeed(&options, B9600);
243
        options.c_iflag &= ~ICRNL;
244
        options.c_oflag &= ~OCRNL;
245
        options.c_cflag |= (CLOCAL | CREAD);
246
        options.c_cflag &= ~PARENB;
247
        options.c_cflag &= ~CSTOPB;
248
        options.c_cflag &= ~CSIZE;
249
        options.c_cflag |= CS8;
250
        options.c_lflag &= ~ICANON;
251
        options.c_cc[VMIN] = 1;
252
        options.c_cc[VTIME] = 50;
253

    
254
        if (tcsetattr(xbee_stream, TCSANOW, &options))
255
        {
256
                WL_DEBUG_PRINT("Error setting attributes.\n");
257
                return -1;
258
        }
259

    
260
        xbee_listen_thread = (pthread_t*)malloc(sizeof(pthread_t));
261
        if (xbee_listen_thread == NULL)
262
        {
263
                WL_DEBUG_PRINT("Malloc failed.\n");
264
                return -1;
265
        }
266

    
267
        int ret = pthread_create(xbee_listen_thread, NULL, listen_to_xbee, NULL);
268
        if (ret)
269
        {
270
                WL_DEBUG_PRINT("Failed to create listener thread.\n");
271
                return -1;
272
        }
273
#endif
274

    
275
        WL_DEBUG_PRINT("Entering command mode.\n");
276

    
277
        if (xbee_enter_command_mode() != 0) {
278
                return -1;
279
        }
280

    
281
        WL_DEBUG_PRINT("Entered command mode.\n");
282

    
283
        if (xbee_enter_api_mode() != 0) {
284
                return -1;
285
        }
286

    
287
        WL_DEBUG_PRINT("Entered api mode.\n");
288

    
289
        if (xbee_exit_command_mode() != 0) {
290
                return -1;
291
        }
292

    
293
        WL_DEBUG_PRINT("Left command mode.\n");
294

    
295
        if (xbee_send_read_at_command("MY")) {
296
                return -1;
297
        }
298
        WL_DEBUG_PRINT("Getting ATMY address.\n");
299

    
300
#ifndef ROBOT
301
        int i;
302
        for (i = 0; xbee_address == 0 && i < XBEE_GET_PACKET_TIMEOUT; i++) {
303
          ret = xbee_get_packet(NULL);
304

    
305
          usleep(1000);
306

    
307
          if (ret == -1) {
308
            WL_DEBUG_PRINT("xbee_get_packet(NULL) failed.\n");
309
            return -1;
310
          }
311
        }
312
#else
313
        //wait to return until the address is set
314
  //TODO: this shouldn't wait indefinitely.  There should be some sort of reasonable timeout
315
  // so if the address is never set right, an error can be returned instead of having the
316
  // robot hang forever
317
        while (xbee_address == 0) {
318
          xbee_get_packet(NULL);
319
        }
320
#endif
321
        WL_DEBUG_PRINT("Got ATMY address.\n");
322

    
323
#ifndef ROBOT
324
        if (i == XBEE_GET_PACKET_TIMEOUT) { // We timed-out.
325

    
326
          WL_DEBUG_PRINT("xbee_get_packet timed out.\n");
327
          return -1;
328
        } else {
329
          return 0;
330
        }
331
#else
332
        return 0;
333
#endif
334
}
335

    
336
/**
337
 * Call when finished using the XBee library. This releases
338
 * all sued resources.
339
 **/
340
void xbee_terminate()
341
{
342
        #ifndef ROBOT
343
        pthread_cancel(*xbee_listen_thread);
344
        free(xbee_listen_thread);
345
        lockf(xbee_stream, F_ULOCK, 0);
346
        close(xbee_stream);
347
        #endif
348
}
349

    
350
/**
351
 * Send a buffer buf of size bytes to the XBee.
352
 *
353
 * @param buf the buffer of data to send
354
 * @param size the number of bytes to send
355
 **/
356
static int xbee_send(char* buf, int size)
357
{
358
#ifdef ROBOT
359
        int i;
360
        for (i = 0; i < size; i++) {
361
                xbee_putc(buf[i]);
362
        }
363

    
364
        return 0;
365

    
366
#else
367

    
368
        int ret = write(xbee_stream, buf, size);
369
        //success
370
        if (ret == size)
371
                return 0;
372
        if (ret == -1)
373
        {
374
                //interrupted by system signal, probably timer interrupt.
375
                //just try again
376
                if (errno == 4)
377
                {
378
                        return xbee_send(buf, size);
379
                }
380
                WL_DEBUG_PRINT("Failed to write to xbee\r\n");
381
                return -1;
382
        }
383

    
384
        //write was interrupted after writing ret bytes
385
        return xbee_send(buf + ret, size - ret);
386
#endif
387
}
388

    
389
/**
390
 * Sends a string to the XBee.
391
 *
392
 * @param c the string to send to the XBEE
393
 **/
394
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
395
// it reduces code size or not should be done to be sure.
396
static int xbee_send_string(char* c)
397
{
398
        return xbee_send(c, strlen(c));
399
}
400

    
401
#ifndef ROBOT
402
static int xbee_read(char* buf, int size)
403
{
404
        if (read(xbee_stream, buf, size) == -1) {
405
                WL_DEBUG_PRINT("Failed to read from xbee.\r\n");
406
                return -1;
407
        }
408

    
409
        return 0;
410
}
411
#endif
412

    
413
/**
414
 * Enter into command mode.
415
 **/
416
static int xbee_enter_command_mode()
417
{
418
        if (xbee_send_string("+++") != 0) {
419
                return -1;
420
        }
421

    
422
        if (xbee_wait_for_ok() != 0) {
423
          return -1;
424
        }
425
          return 0;
426
}
427

    
428
/**
429
 * Exit from command mode.
430
 **/
431
static int xbee_exit_command_mode()
432
{
433
        if (xbee_send_string("ATCN\r") != 0) {
434
                return -1;
435
        }
436

    
437
        xbee_wait_for_ok();
438

    
439
        return 0;
440
}
441

    
442
/**
443
 * Enter API mode.
444
 **/
445
static int xbee_enter_api_mode()
446
{
447
        if (xbee_send_string("ATAP 1\r") != 0) {
448
                return -1;
449
        }
450
        xbee_wait_for_ok();
451

    
452
        return 0;
453
}
454

    
455
/**
456
 * Wait until the string "OK\r" is received from the XBee.
457
 **/
458
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
459
// it reduces code size or not should be done to be sure.
460
static int xbee_wait_for_ok()
461
{
462
        return xbee_wait_for_string("OK\r", 3);
463
}
464

    
465
/**
466
 * Delay until the specified string is received from
467
 * the XBee. Discards all other XBee data.
468
 *
469
 * @param s the string to receive
470
 * @param len the length of the string
471
 **/
472
static int xbee_wait_for_string(char* s, int len)
473
{
474
        char* curr = s;
475
        while (curr - s < len) {
476
                // check if buffer is empty
477
                if (buffer_last != buffer_first) {
478
                        char c = arrival_buf[buffer_first++];
479
                        usb_putc(c);
480
                        usb_putc('\n');
481
                        if (buffer_first == XBEE_BUFFER_SIZE) {
482
                                buffer_first = 0;
483
                        }
484

    
485
                        if (c == *curr) {
486
                                curr++;
487
                        } else {
488
#ifndef ROBOT
489
                          //return -1; // Computer is less forgiving.
490
                          curr = s;
491
#else
492
                          curr = s;
493
#endif
494
                        }
495
                } // else buffer is empty.
496

    
497
#ifndef ROBOT
498
                usleep(100);
499
#endif
500
        }
501

    
502
        return 0;
503
}
504

    
505
/**
506
 * Verifies that the packets checksum is correct.
507
 * (If the checksum is correct, the sum of the bytes
508
 * is 0xFF.)
509
 *
510
 * @param packet the packet received. This includes the first
511
 * three bytes, which are header information from the XBee.
512
 *
513
 * @param len The length of the packet received from the XBee
514
 *
515
 * @return 0 if the checksum is incorrect, nonzero
516
 * otherwise
517
 **/
518
int xbee_verify_checksum(char* packet, int len)
519
{
520
        unsigned char sum = 0;
521
        int i;
522
        for (i = 3; i < len; i++)
523
                sum += (unsigned char)packet[i];
524
        return sum == 0xFF;
525
}
526

    
527
/**
528
 * Returns the checksum of the given packet.
529
 *
530
 * @param buf the data for the packet to send
531
 * @param len the length of the packet in bytes
532
 *
533
 * @return the checksum of the packet, which will
534
 * become the last byte sent in the packet
535
 **/
536
char xbee_compute_checksum(char* buf, int len)
537
{
538
        int i;
539
        unsigned char sum = 0;
540
        for (i = 0; i < len; i++)
541
                sum += (unsigned char)buf[i];
542
        return 0xFF - sum;
543
}
544

    
545
/**
546
 * Adds header information and checksum to the given
547
 * packet and sends it. Header information includes
548
 * XBEE_FRAME_START and the packet length, as two bytes.
549
 *
550
 * @param buf the packet data
551
 * @param len the size in bytes of the packet data
552
 *
553
 **/
554
static int xbee_send_frame(char* buf, int len)
555
{
556
        char prefix[3];
557
        prefix[0] = XBEE_FRAME_START;
558
        prefix[1] = (len & 0xFF00) >> 8;
559
        prefix[2] = len & 0xFF;
560
        char checksum = xbee_compute_checksum(buf, len);
561

    
562
        if (xbee_send(prefix, 3) != 0) {
563
                return -1;
564
        }
565

    
566
        if (xbee_send(buf, len) != 0) {
567
                return -1;
568
        }
569

    
570
        if (xbee_send(&checksum, 1) != 0) {
571
                return -1;
572
        }
573

    
574
        return 0;
575
}
576

    
577
/**
578
 * Sends an AT command to read a parameter.
579
 *
580
 * @param command the AT command to send. For exmaple,
581
 * use ID to read the PAN ID and MY to return the XBee ID.
582
 * See the XBee reference guide for a complete listing.
583
 **/
584
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
585
// it reduces code size or not should be done to be sure.
586
static int xbee_send_read_at_command(char* command)
587
{
588
        return xbee_send_modify_at_command(command, NULL);
589
}
590

    
591
/**
592
 * Sends the given AT command.
593
 *
594
 * @param command the AT command to send (e.g., MY, ID)
595
 * @param value the value to pass as a parameter
596
 * (or NULL if there is no parameter)
597
 **/
598
static int xbee_send_modify_at_command(char* command, char* value)
599
{
600
        char buf[16];
601
        int i;
602

    
603
        buf[0] = XBEE_FRAME_AT_COMMAND;
604
        buf[1] = 1;
605
        buf[2] = command[0];
606
        buf[3] = command[1];
607
        int valueLen = 0;
608
        if (value != NULL)
609
        {
610
                valueLen = strlen(value);
611
                if (valueLen > 8)
612
                {
613
                        WL_DEBUG_PRINT("AT Command too large.\r\n");
614
                        return -1;
615
                }
616

    
617
                for (i = 0; i < valueLen; i++) {
618
                        buf[4 + i] = value[i];
619
                }
620
        }
621

    
622
        return xbee_send_frame(buf, 4 + valueLen);
623
}
624

    
625
/**
626
 * Send the specified packet.
627
 *
628
 * @param packet the packet data to send
629
 * @param len the number of bytes in the packet
630
 *
631
 * @param dest the ID of the XBee to send the packet to,
632
 * or XBEE_BROADCAST to send the message to all robots
633
 * in the PAN.
634
 *
635
 * @param options a combination of the flags
636
 * XBEE_OPTIONS_NONE, XBEE_OPTIONS_DISABLE_RESPONSE and
637
 * XBEE_OPTIONS_BROADCAST_ALL_PANS
638
 *
639
 * @param frame the frame number to associate this packet
640
 * with. This will be used to identify the response when
641
 * the XBee alerts us as to whether or not our message
642
 * was received.
643
 **/
644
int xbee_send_packet(char* packet, int len, int dest, char options, char frame)
645
{
646
        char buf[5];
647
        char prefix[3];
648
        int i;
649
        unsigned char checksum = 0;
650

    
651
        if (len > 100)
652
        {
653
                WL_DEBUG_PRINT("Packet is too large.\r\n");
654
                return -1;
655
        }
656

    
657
        //data for sending request
658
        buf[0] = XBEE_FRAME_TX_REQUEST_16;
659
        buf[1] = frame;
660
        buf[2] = (dest >> 8) & 0xFF;
661
        buf[3] = dest & 0xFF;
662
        buf[4] = options;
663

    
664
        //packet prefix, do this here so we don't need an extra buffer
665
        prefix[0] = XBEE_FRAME_START;
666
        prefix[1] = ((5 + len) & 0xFF00) >> 8;
667
        prefix[2] = (5 + len) & 0xFF;
668

    
669
        for (i = 0; i < 5; i++)
670
                checksum += (unsigned char)buf[i];
671
        for (i = 0; i < len; i++)
672
                checksum += (unsigned char)packet[i];
673
        checksum = 0xFF - checksum;
674

    
675
        if (xbee_send(prefix, 3) != 0) {
676
                return -1;
677
        }
678

    
679
        if (xbee_send(buf, 5) != 0) {
680
                return -1;
681
        }
682

    
683
        if (xbee_send(packet, len) != 0) {
684
                return -1;
685
        }
686

    
687
        if (xbee_send((char*)&checksum, 1) != 0) {
688
                return -1;
689
        }
690

    
691
        return 0;
692
}
693

    
694
/**
695
 * Reads a packet received from the XBee. This function
696
 * is non-blocking. The resulting packet is stored in dest.
697
 * Only returns transmission response packets and
698
 * received packets. The returned packet does not include
699
 * header information or the checksum. This method also
700
 * handles special packets dealt with by the XBee library,
701
 * and so should be called frequently while the XBee is in
702
 * use.<br><br>
703
 *
704
 * The first byte of the packet will be either
705
 * XBEE_TX_STATUS or XBEE_RX to indicated
706
 * a response to a sent message or a received message,
707
 * respectively.<br><br>
708
 *
709
 * For a status response packet:<br>
710
 * The first byte will be XBEE_TX_STATUS.<br>
711
 * The second byte will be the frame number.<br>
712
 * The third byte will be the result. 0 indicates success,
713
 * and nonzero indicates that an error ocurred in
714
 * transmitting the packet.<br><br>
715
 *
716
 * For a received packet:<br>
717
 * The first byte will be XBEE_RX.<br>
718
 * The second and third bytes will be the 16-bit
719
 * address of the packet's sender.<br>
720
 * The fourth byte is the signal strength.<br>
721
 * The fifth byte is 1 if the packet were sent to
722
 * a specific address, and 2 if it is a broadcast packet.<br><br>
723
 *
724
 * @param dest set to the packet data
725
 * @return the length of the packet, or -1 if no packet
726
 * is available
727
 **/
728
int xbee_get_packet(unsigned char* dest)
729
{
730
        //start reading a packet with XBEE_FRAME_START
731
        if (currentBufPos == 0)
732
        {
733
                do
734
                {
735
                        if (buffer_first == XBEE_BUFFER_SIZE)
736
                                buffer_first = 0;
737
                        // check if buffer is empty
738
                        if (buffer_first == buffer_last) {
739
                                return 0;
740
                        }
741
                } while (arrival_buf[buffer_first++] != XBEE_FRAME_START);
742

    
743
                if (buffer_first == XBEE_BUFFER_SIZE) {
744
                        buffer_first = 0;
745
                }
746
                xbee_buf[0] = XBEE_FRAME_START;
747
                currentBufPos++;
748
        }
749

    
750
        int len = -1;
751
        if (currentBufPos >= 3) {
752
                len = (int)xbee_buf[2] + ((int)xbee_buf[1] << 8);
753
        }
754

    
755
        while (len == -1 //packet length has not been read yet
756
                || currentBufPos < len + 4)
757
        {
758
                if (currentBufPos == 3)
759
                {
760
                        len = (int)xbee_buf[2] + ((int)xbee_buf[1] << 8);
761
                        if (len > 120)
762
                        {
763
                                WL_DEBUG_PRINT("Packet too large. Probably error in XBee transmission.\n");
764
                                currentBufPos = 0;
765
                                return -1;
766
                        }
767
                }
768

    
769
                // check if buffer is empty
770
                if (buffer_first == buffer_last) {
771
                        return 0;
772
                }
773
                xbee_buf[currentBufPos++] = arrival_buf[buffer_first++];
774
                if (buffer_first == XBEE_BUFFER_SIZE) {
775
                        buffer_first = 0;
776
                }
777
        }
778

    
779
        currentBufPos = 0;
780

    
781
        if (!xbee_verify_checksum(xbee_buf, len + 4))
782
        {
783
                WL_DEBUG_PRINT("XBee checksum failed.\r\n");
784
                return -1;
785
        }
786

    
787
        //we will take care of the packet
788
        if (xbee_handle_packet(xbee_buf + 3, len) != 0) {
789
                return 0;
790
        }
791

    
792
        if (dest == NULL) {
793
                return 0;
794
        }
795

    
796
        int i;
797
        for (i = 3; i < len + 3; i++) {
798
                dest[i - 3] = xbee_buf[i];
799
        }
800

    
801
        return len;
802
}
803

    
804
/**
805
 * Handles modem status packets.
806
 *
807
 * @param status the type of status packet received.
808
 **/
809
void xbee_handle_status(char status)
810
{
811
        switch (status)
812
        {
813
                case 0:
814
                        WL_DEBUG_PRINT("XBee hardware reset.\r\n");
815
                        break;
816
                case 1:
817
                        WL_DEBUG_PRINT("Watchdog timer reset.\r\n");
818
                        break;
819
                case 2:
820
                        WL_DEBUG_PRINT("Associated.\r\n");
821
                        break;
822
                case 3:
823
                        WL_DEBUG_PRINT("Disassociated.\r\n");
824
                        break;
825
                case 4:
826
                        WL_DEBUG_PRINT("Synchronization lost.\r\n");
827
                        break;
828
                case 5:
829
                        WL_DEBUG_PRINT("Coordinator realignment.\r\n");
830
                        break;
831
                case 6:
832
                        WL_DEBUG_PRINT("Coordinator started.\r\n");
833
                        break;
834
        }
835
}
836

    
837
/**
838
 * Handles AT command response packets.
839
 * @param command the two character AT command, e.g. MY or ID
840
 * @param result 0 for success, 1 for an error
841
 * @param extra the hex value of the requested register
842
 * @param extraLen the length in bytes of extra
843
 **/
844
static void xbee_handle_at_command_response(char* command, char result, char* extra, int extraLen)
845
{
846
        if (result == 1)
847
        {
848
                WL_DEBUG_PRINT("Error with AT");
849
                WL_DEBUG_PRINT(command);
850
                WL_DEBUG_PRINT(" packet.\r\n");
851
        }
852
        WL_DEBUG_PRINT("AT");
853
        WL_DEBUG_PRINT(command);
854
        WL_DEBUG_PRINT(" command was successful.\r\n");
855

    
856
        if (command[0] == 'I' && command[1] == 'D')
857
        {
858
                xbee_panID = xbee_pending_panID;
859
                WL_DEBUG_PRINT("PAN ID set to ");
860
                WL_DEBUG_PRINT_INT(xbee_panID);
861
                WL_DEBUG_PRINT(".\r\n");
862
                return;
863
        }
864

    
865
        if (command[0] == 'C' && command[1] == 'H')
866
        {
867
                xbee_channel = xbee_pending_channel;
868
                WL_DEBUG_PRINT("Channel set to ");
869
                WL_DEBUG_PRINT_INT(xbee_channel);
870
                WL_DEBUG_PRINT(".\r\n");
871
                return;
872
        }
873

    
874
        if (command[0] == 'M' && command[1] == 'Y' && extraLen != 0)
875
        {
876
                xbee_address = 0;
877
                int i;
878
                for (i = 0; i < extraLen; i++) {
879
                        xbee_address = (xbee_address << 8) + extra[i];
880
                }
881

    
882
                WL_DEBUG_PRINT("XBee address is ");
883
                WL_DEBUG_PRINT_INT(xbee_address);
884
                WL_DEBUG_PRINT(".\r\n");
885

    
886
                if (xbee_address == 0)
887
                {
888
                        WL_DEBUG_PRINT("XBee 16-bit address must be set using ATMY.\r\n");
889
                        #ifndef ROBOT
890
                        exit(0);
891
                        #endif
892
                }
893
        }
894
}
895

    
896
/**
897
 * Attempts to handle the packet if it is dealt with
898
 * by the library.
899
 * We will handle the following packet types:
900
 *    Modem Status
901
 *    AT Command Response
902
 *
903
 * @param packet the packet to handle
904
 * @param len the length of the packet
905
 *
906
 * @return 1 if we have handled the packet, 0 otherwise
907
 */
908
static int xbee_handle_packet(char* packet, int len)
909
{
910

    
911
        char command[3] = {1, 2, 3};
912
        if (len <= 0) //this should not happend
913
        {
914
                WL_DEBUG_PRINT("Non-positive packet length.\r\n");
915
                return 0;
916
        }
917

    
918
        switch ((unsigned char)packet[0]) //packet type
919
        {
920
                case XBEE_FRAME_STATUS:
921
                        xbee_handle_status(packet[1]);
922
                        return 1;
923
                case XBEE_FRAME_AT_COMMAND_RESPONSE:
924
                        command[0] = packet[2];
925
                        command[1] = packet[3];
926
                        command[2] = 0;
927
                        xbee_handle_at_command_response(command, packet[4], packet + 5, len - 5);
928
                        return 1;
929
        }
930
        return 0;
931
}
932

    
933
/**
934
 * Sets the personal area network id.
935
 *
936
 * @param id the new personal area network (PAN) id
937
 **/
938
int xbee_set_pan_id(int id)
939
{
940
        char s[3];
941
        s[0] = (id >> 8) & 0xFF;
942
        s[1] = id & 0xFF;
943
        s[2] = 0;
944
        xbee_pending_panID = id;
945
        return xbee_send_modify_at_command("ID", s);
946
}
947

    
948
/**
949
 * Get the PAN ID for the XBee.
950
 *
951
 * @return the personal area network id, or
952
 * XBEE_PAN_DEFAULT if it has not yet been set.
953
 **/
954
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
955
// it reduces code size or not should be done to be sure.
956
unsigned int xbee_get_pan_id()
957
{
958
        return xbee_panID;
959
}
960

    
961
/**
962
 * Set the channel the XBee is using.
963
 *
964
 * @param channel the channel the XBee will not use,
965
 * between 0x0B and 0x1A
966
 *
967
 * @see xbee_get_channel
968
 **/
969
int xbee_set_channel(int channel)
970
{
971
        if (channel < 0x0B || channel > 0x1A)
972
        {
973
                WL_DEBUG_PRINT("Channel out of range.\r\n");
974
                return -1;
975
        }
976

    
977
        char s[3];
978
        s[0] = channel & 0xFF;
979
        s[1] = 0;
980
        xbee_pending_channel = channel;
981

    
982
        return xbee_send_modify_at_command("CH", s);
983
}
984

    
985
/**
986
 * Returns the channel which the XBee is currently using.
987
 *
988
 * @return the channel the XBee is using
989
 *
990
 * @see xbee_set_channel
991
 **/
992
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
993
// it reduces code size or not should be done to be sure.
994
int xbee_get_channel(void)
995
{
996
        return xbee_channel;
997
}
998

    
999
/**
1000
 * Get the 16-bit address of the XBee.
1001
 * This is used to specify who to send messages to
1002
 * and who messages are from.
1003
 *
1004
 * @return the 16-bit address of the XBee.
1005
 **/
1006
//TODO: this function is so simple, it *may* be beneficial to inline this function.  testing of if
1007
// it reduces code size or not should be done to be sure.
1008
unsigned int xbee_get_address()
1009
{
1010
        return xbee_address;
1011
}
1012

    
1013
#ifndef ROBOT
1014
void xbee_set_com_port(char* port)
1015
{
1016
        xbee_com_port = port;
1017
}
1018
#endif