Don't set ao_flash_setup_done until we're actually done.
[fw/altos] / src / ao_packet.c
1 /*
2  * Copyright © 2009 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include "ao.h"
19
20 __xdata struct ao_packet_recv ao_rx_packet;
21 __xdata struct ao_packet ao_tx_packet;
22 __pdata uint8_t ao_packet_rx_len, ao_packet_rx_used, ao_packet_tx_used;
23
24 static __xdata char tx_data[AO_PACKET_MAX];
25 static __xdata char rx_data[AO_PACKET_MAX];
26 static __pdata uint8_t rx_seq;
27
28 __xdata struct ao_task  ao_packet_task;
29 __xdata uint8_t ao_packet_enable;
30 __xdata uint8_t ao_packet_master_sleeping;
31
32 void
33 ao_packet_send(void)
34 {
35         ao_led_on(AO_LED_RED);
36         ao_radio_get();
37
38         /* If any tx data is pending then copy it into the tx packet */
39         if (ao_packet_tx_used && ao_tx_packet.len == 0) {
40                 memcpy(&ao_tx_packet.d, tx_data, ao_packet_tx_used);
41                 ao_tx_packet.len = ao_packet_tx_used;
42                 ao_tx_packet.seq++;
43                 ao_packet_tx_used = 0;
44                 ao_wakeup(&tx_data);
45         }
46         ao_radio_done = 0;
47         ao_dma_set_transfer(ao_radio_dma,
48                             &ao_tx_packet,
49                             &RFDXADDR,
50                             sizeof (struct ao_packet),
51                             DMA_CFG0_WORDSIZE_8 |
52                             DMA_CFG0_TMODE_SINGLE |
53                             DMA_CFG0_TRIGGER_RADIO,
54                             DMA_CFG1_SRCINC_1 |
55                             DMA_CFG1_DESTINC_0 |
56                             DMA_CFG1_PRIORITY_HIGH);
57         ao_dma_start(ao_radio_dma);
58         RFST = RFST_STX;
59         __critical while (!ao_radio_done)
60                 ao_sleep(&ao_radio_done);
61         ao_radio_put();
62         ao_led_off(AO_LED_RED);
63 }
64
65 uint8_t
66 ao_packet_recv(void)
67 {
68         uint8_t dma_done;
69
70         ao_led_on(AO_LED_GREEN);
71         ao_radio_get();
72         ao_dma_set_transfer(ao_radio_dma,
73                             &RFDXADDR,
74                             &ao_rx_packet,
75                             sizeof (struct ao_packet_recv),
76                             DMA_CFG0_WORDSIZE_8 |
77                             DMA_CFG0_TMODE_SINGLE |
78                             DMA_CFG0_TRIGGER_RADIO,
79                             DMA_CFG1_SRCINC_0 |
80                             DMA_CFG1_DESTINC_1 |
81                             DMA_CFG1_PRIORITY_HIGH);
82         ao_dma_start(ao_radio_dma);
83         RFST = RFST_SRX;
84         __critical while (!ao_radio_dma_done)
85                            if (ao_sleep(&ao_radio_dma_done) != 0)
86                                    ao_radio_abort();
87         dma_done = ao_radio_dma_done;
88         ao_radio_put();
89         ao_led_off(AO_LED_GREEN);
90
91         if (dma_done & AO_DMA_DONE) {
92                 if (!(ao_rx_packet.status & PKT_APPEND_STATUS_1_CRC_OK))
93                         return AO_DMA_ABORTED;
94                 if (ao_rx_packet.packet.len == AO_PACKET_SYN) {
95                         rx_seq = ao_rx_packet.packet.seq;
96                         ao_tx_packet.seq = ao_rx_packet.packet.ack;
97                         ao_tx_packet.ack = rx_seq;
98                 } else if (ao_rx_packet.packet.len) {
99                         if (ao_rx_packet.packet.seq == (uint8_t) (rx_seq + (uint8_t) 1) && ao_packet_rx_used == ao_packet_rx_len) {
100                                 memcpy(rx_data, ao_rx_packet.packet.d, ao_rx_packet.packet.len);
101                                 ao_packet_rx_used = 0;
102                                 ao_packet_rx_len = ao_rx_packet.packet.len;
103                                 rx_seq = ao_rx_packet.packet.seq;
104                                 ao_tx_packet.ack = rx_seq;
105                                 ao_wakeup(&ao_stdin_ready);
106                         }
107                 }
108                 if (ao_rx_packet.packet.ack == ao_tx_packet.seq) {
109                         ao_tx_packet.len = 0;
110                         ao_wakeup(&ao_tx_packet);
111                 }
112         }
113         return dma_done;
114 }
115
116 void
117 ao_packet_flush(void)
118 {
119         /* If there is data to send, and this is the master,
120          * then poke the master to send all queued data
121          */
122         if (ao_packet_tx_used && ao_packet_master_sleeping)
123                 ao_wake_task(&ao_packet_task);
124 }
125
126 void
127 ao_packet_putchar(char c) __reentrant
128 {
129         while (ao_packet_tx_used == AO_PACKET_MAX && ao_packet_enable) {
130                 ao_packet_flush();
131                 ao_sleep(&tx_data);
132         }
133
134         if (ao_packet_enable)
135                 tx_data[ao_packet_tx_used++] = c;
136 }
137
138 char
139 ao_packet_pollchar(void) __critical
140 {
141         if (!ao_packet_enable)
142                 return AO_READ_AGAIN;
143
144         if (ao_packet_rx_used == ao_packet_rx_len)
145                 return AO_READ_AGAIN;
146
147         return rx_data[ao_packet_rx_used++];
148 }