Fix USB driver and example

This commit is contained in:
NIIBE Yutaka
2016-04-18 17:23:10 +09:00
parent 82749ab97a
commit 0bbb43dd3a
4 changed files with 136 additions and 17 deletions

View File

@@ -199,7 +199,7 @@ main (int argc, const char *argv[])
/* Send ZLP at the beginning. */ /* Send ZLP at the beginning. */
stream_send (st, s, 0); stream_send (st, s, 0);
memcpy (s, "xx: Hello, World with Chopstx!\r\n\000", 32); memcpy (s, "xx: Hello, World with Chopstx!\r\n", 32);
s[0] = hexchar (count >> 4); s[0] = hexchar (count >> 4);
s[1] = hexchar (count & 0x0f); s[1] = hexchar (count & 0x0f);
count++; count++;
@@ -209,14 +209,19 @@ main (int argc, const char *argv[])
while (1) while (1)
{ {
int size = stream_recv (st, s); int size = stream_recv (st, s + 4);
if (size < 0) if (size < 0)
break; break;
if (size >= 0) if (size >= 0)
{ {
if (stream_send (st, s, size) < 0) s[0] = hexchar (size >> 4);
s[1] = hexchar (size & 0x0f);
s[size + 4] = '\r';
s[size + 5] = '\n';
if (stream_send (st, s, size + 6) < 0)
break; break;
} }

View File

@@ -9,8 +9,10 @@
struct stream { struct stream {
chopstx_mutex_t mtx; chopstx_mutex_t mtx;
chopstx_cond_t cnd; chopstx_cond_t cnd;
int sending;
unsigned int recv_len;
uint8_t recv_buf[BUFSIZE];
uint8_t buf_send[BUFSIZE]; /* Not yet used. */ uint8_t buf_send[BUFSIZE]; /* Not yet used. */
uint8_t buf_recv[BUFSIZE]; /* Not yet used. */
uint8_t cnt_send_head; /* Not yet used. */ uint8_t cnt_send_head; /* Not yet used. */
uint8_t cnt_send_tail; /* Not yet used. */ uint8_t cnt_send_tail; /* Not yet used. */
uint8_t cnt_recv_head; /* Not yet used. */ uint8_t cnt_recv_head; /* Not yet used. */

View File

@@ -1,9 +1,20 @@
#include <stdint.h> #include <stdint.h>
#include <stdlib.h> #include <stdlib.h>
#include <chopstx.h> #include <chopstx.h>
#include <string.h>
#include "usb_lld.h" #include "usb_lld.h"
#include "stream.h" #include "stream.h"
static uint8_t send_buf[64];
static unsigned int send_len;
static uint8_t send_buf1[64];
static uint8_t recv_buf[64];
static unsigned int recv_len;
static uint8_t inputline[64];
static unsigned int inputline_len;
static struct stream stream; static struct stream stream;
#define USB_CDC_REQ_SET_LINE_CODING 0x20 #define USB_CDC_REQ_SET_LINE_CODING 0x20
@@ -184,6 +195,8 @@ usb_cb_ctrl_write_finish (uint8_t req, uint8_t req_no, struct req_args *arg)
stream.flags &= ~FLAG_CONNECTED; stream.flags &= ~FLAG_CONNECTED;
stream.flags |= ((arg->value & CDC_CTRL_DTR) != 0)? FLAG_CONNECTED : 0; stream.flags |= ((arg->value & CDC_CTRL_DTR) != 0)? FLAG_CONNECTED : 0;
chopstx_cond_signal (&stream.cnd); chopstx_cond_signal (&stream.cnd);
recv_len = 0;
usb_lld_rx_enable (ENDP3, recv_buf, 64);
chopstx_mutex_unlock (&stream.mtx); chopstx_mutex_unlock (&stream.mtx);
} }
} }
@@ -389,21 +402,118 @@ usb_cb_interface (uint8_t cmd, struct req_args *arg)
} }
static void
stream_echo_char (int c)
{
chopstx_mutex_lock (&stream.mtx);
if (send_len < sizeof (send_buf))
send_buf[send_len++] = c;
else
{
/* All that we can is ignoring the output. */
;
}
if (stream.sending == 0)
{
memcpy (send_buf1, send_buf, send_len);
usb_lld_tx_enable (ENDP1, send_buf1, send_len);
send_len = 0;
stream.sending = 1;
}
chopstx_mutex_unlock (&stream.mtx);
}
void void
usb_cb_tx_done (uint8_t ep_num) usb_cb_tx_done (uint8_t ep_num)
{ {
if (ep_num == ENDP1) if (ep_num == ENDP1)
{ {
chopstx_mutex_lock (&stream.mtx); chopstx_mutex_lock (&stream.mtx);
if ((stream.flags & FLAG_SEND_AVAIL)) stream.sending = 0;
if (send_len)
{ {
stream.flags &= ~FLAG_SEND_AVAIL; stream.sending = 1;
chopstx_cond_signal (&stream.cnd); memcpy (send_buf1, send_buf, send_len);
usb_lld_tx_enable (ENDP1, send_buf1, send_len);
send_len = 0;
}
else
{
if ((stream.flags & FLAG_SEND_AVAIL))
{
stream.flags &= ~FLAG_SEND_AVAIL;
chopstx_cond_signal (&stream.cnd);
}
} }
chopstx_mutex_unlock (&stream.mtx); chopstx_mutex_unlock (&stream.mtx);
} }
else if (ep_num == ENDP2) else if (ep_num == ENDP2)
{ {
/* Nothing */
}
}
static void
stream_input_char (int c)
{
unsigned int i;
/* Process DEL, C-U, C-R, and RET as editing command. */
switch (c)
{
case 0x0d: /* Control-M */
stream_echo_char (0x0d);
stream_echo_char (0x0a);
chopstx_mutex_lock (&stream.mtx);
if ((stream.flags & FLAG_RECV_AVAIL) == 0)
{
memcpy (stream.recv_buf, inputline, inputline_len);
stream.recv_len = inputline_len;
stream.flags |= FLAG_RECV_AVAIL;
chopstx_cond_signal (&stream.cnd);
}
chopstx_mutex_unlock (&stream.mtx);
inputline_len = 0;
break;
case 0x12: /* Control-R */
stream_echo_char ('^');
stream_echo_char ('R');
stream_echo_char (0x0d);
stream_echo_char (0x0a);
for (i = 0; i < inputline_len; i++)
stream_echo_char (inputline[i]);
break;
case 0x15: /* Control-U */
for (i = 0; i < inputline_len; i++)
{
stream_echo_char (0x08);
stream_echo_char (0x20);
stream_echo_char (0x08);
}
inputline_len = 0;
break;
case 0x7f: /* DEL */
if (inputline_len > 0)
{
stream_echo_char (0x08);
stream_echo_char (0x20);
stream_echo_char (0x08);
inputline_len--;
}
break;
default:
if (inputline_len < sizeof (inputline))
{
stream_echo_char (c);
inputline[inputline_len++] = c;
}
else
/* Beep */
stream_echo_char (0x0a);
break;
} }
} }
@@ -412,13 +522,13 @@ usb_cb_rx_ready (uint8_t ep_num)
{ {
if (ep_num == ENDP3) if (ep_num == ENDP3)
{ {
chopstx_mutex_lock (&stream.mtx); int i, r;
if ((stream.flags & FLAG_RECV_AVAIL) == 0)
{ r = usb_lld_rx_data_len (ENDP3);
stream.flags |= FLAG_RECV_AVAIL; for (i = 0; i < r; i++)
chopstx_cond_signal (&stream.cnd); stream_input_char (recv_buf[i]);
}
chopstx_mutex_unlock (&stream.mtx); usb_lld_rx_enable (ENDP3, recv_buf, 64);
} }
} }
@@ -452,6 +562,7 @@ stream_send (struct stream *st, uint8_t *buf, uint8_t count)
r = -1; r = -1;
else else
{ {
stream.sending = 1;
usb_lld_tx_enable (ENDP1, buf, count); usb_lld_tx_enable (ENDP1, buf, count);
stream.flags |= FLAG_SEND_AVAIL; stream.flags |= FLAG_SEND_AVAIL;
do do
@@ -467,6 +578,7 @@ stream_send (struct stream *st, uint8_t *buf, uint8_t count)
} }
while (1); while (1);
} }
stream.sending = 0;
chopstx_mutex_unlock (&st->mtx); chopstx_mutex_unlock (&st->mtx);
return r; return r;
} }
@@ -482,14 +594,14 @@ stream_recv (struct stream *st, uint8_t *buf)
r = -1; r = -1;
else else
{ {
usb_lld_rx_enable (ENDP3, buf, 64);
stream.flags &= ~FLAG_RECV_AVAIL; stream.flags &= ~FLAG_RECV_AVAIL;
do do
{ {
chopstx_cond_wait (&st->cnd, &st->mtx); chopstx_cond_wait (&st->cnd, &st->mtx);
if ((stream.flags & FLAG_RECV_AVAIL)) if ((stream.flags & FLAG_RECV_AVAIL))
{ {
r = usb_lld_rx_data_len (ENDP3); r = stream.recv_len;
memcpy (buf, stream.recv_buf, r);
break; break;
} }
else if ((stream.flags & FLAG_CONNECTED) == 0) else if ((stream.flags & FLAG_CONNECTED) == 0)

View File

@@ -863,8 +863,8 @@ handle_transaction (uint8_t stat)
if ((stat & 0x08) == 0) if ((stat & 0x08) == 0)
{ {
dev_p->recv++; dev_p->recv++;
usb_cb_rx_ready (ep_num);
ep[ep_num].rx_odd ^= 1; ep[ep_num].rx_odd ^= 1;
usb_cb_rx_ready (ep_num);
} }
else else
{ {