1
0
mirror of git://projects.qi-hardware.com/ben-wpan.git synced 2024-11-04 23:49:42 +02:00

tools/atrf-proxy/: network proxy (untested)

This commit is contained in:
Werner Almesberger 2011-04-10 15:27:03 -03:00
parent 3b0ee3c00a
commit 4ad0383020
3 changed files with 415 additions and 0 deletions

16
tools/atrf-proxy/Makefile Normal file
View File

@ -0,0 +1,16 @@
#
# atrf-proxy/Makefile - ATRF network proxy
#
# Written 2011 by Werner Almesberger
# Copyright 2011 Werner Almesberger
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
MAIN = atrf-proxy
include ../Makefile.common

68
tools/atrf-proxy/PROTOCOL Normal file
View File

@ -0,0 +1,68 @@
Protocol specification
======================
Messages
--------
<open TCP session>
+[greeting]
-message
RESET
+[comment]
-message
RESET_RF
+[comment]
-message
TEST
+[comment]
-message
SLP_TR 0|1
+[comment]
-message
CLKM freq_mhz
+[comment]
-message
SET register value
+[comment]
-message
GET register
+value
-message
WRITE length raw-data
+[comment]
-message
READ
+length raw-data
-message
POLL
+0|1
-message
Message format
--------------
Commands are not case-sensitive. Messages and comments consist of printable
ASCII characters and spaces.
Whitespace separating words is always exactly one space character.
Lines end with a * single newline, without trailing whitespace.
Numeric values are in C notation, i.e., 64, 0100, and 0x40 would all
represent the same value.
"raw-data" is a string of binary data of the indicated length. The other
formatting conventions remain valid after this string, i.e., it must be
followed by a single newline.
The device is implicitly opened when establishing a TCP session. The
device is implicitly closed when closing the TCP session.

View File

@ -0,0 +1,331 @@
/*
* atrf-proxy/atrf-proxy.c - ATRF network proxy
*
* Written 2011 by Werner Almesberger
* Copyright 2011 Werner Almesberger
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*/
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <strings.h> /* for strcasecmp */
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "atrf.h"
#include "netio.h"
#define DEFAULT_PORT 0x1540 /* 5440 */
static int verbose = 0;
static int get_num(struct netio *netio, int last, int *ret)
{
char buf[10];
ssize_t got;
unsigned long n;
char *end;
got = netio_read_until(netio,
last ? "\n" : " ", buf, sizeof(buf), NULL);
if (got < 0) {
*ret = -1;
return -1;
}
buf[got] = 0;
n = strtoul(buf, &end, 0);
if (*end) {
*ret = netio_printf(netio, "-bad number\n");
return -1;
}
/* @@@ check against INT_MAX */
return n;
}
static int cmd_two(struct atrf_dsc *dsc, struct netio *netio, const char *cmd)
{
int n, ret;
n = get_num(netio, 0, &ret);
if (n < 0)
return ret;
if (n > 255)
return netio_printf(netio, "-bad argument\n");
if (!strcasecmp(cmd, "set")) {
int val;
val = get_num(netio, 1, &ret);
if (val < 0)
return ret;
if (val > 255)
return netio_printf(netio, "-bad argument\n");
atrf_reg_write(dsc, n, ret);
if (atrf_error(dsc))
return netio_printf(netio, "-I/O error\n");
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "write")) {
uint8_t *tmp;
ssize_t got;
if (n > 128)
return netio_printf(netio, "-bad argument\n");
tmp = malloc(n+1);
if (!tmp)
return netio_printf(netio, "-out of memory\n");
got = netio_read(netio, tmp, n+1);
if (got != n+1) {
free(tmp);
return -1;
}
if (tmp[n] != '\n') {
free(tmp);
return netio_printf(netio, "-unterminated command\n");
}
atrf_buf_write(dsc, tmp, n);
free(tmp);
if (atrf_error(dsc))
return netio_printf(netio, "-I/O error\n");
return netio_printf(netio, "+\n");
}
abort();
}
static int cmd_more(struct atrf_dsc *dsc, struct netio *netio, const char *cmd)
{
int n, ret;
if (!strcasecmp(cmd, "set"))
return cmd_two(dsc, netio, cmd);
if (!strcasecmp(cmd, "write"))
return cmd_two(dsc, netio, cmd);
n = get_num(netio, 1, &ret);
if (n < 0)
return ret;
if (!strcasecmp(cmd, "slp_tr")) {
if (n > 1)
return netio_printf(netio, "-bad argument\n");
if (atrf_slp_tr(dsc, n) < 0)
return netio_printf(netio, "-I/O error\n");
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "clkm")) {
if (n > 16)
return netio_printf(netio, "-bad argument\n");
if (atrf_set_clkm(dsc, n) < 0)
return netio_printf(netio, "-error\n");
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "get")) {
uint8_t res;
if (n > 255)
return netio_printf(netio, "-bad argument\n");
res = atrf_reg_read(dsc, n);
if (atrf_error(dsc))
return netio_printf(netio, "-I/O error\n");
return netio_printf(netio, "+0x%02x\n", res);
}
return netio_printf(netio, "-unrecognized command\n");
}
static int cmd_zero(struct atrf_dsc *dsc, struct netio *netio, const char *cmd)
{
int res;
if (!strcasecmp(cmd, "reset")) {
atrf_reset(dsc);
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "reset_rf")) {
atrf_reset_rf(dsc);
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "test")) {
atrf_test_mode(dsc);
return netio_printf(netio, "+\n");
}
if (!strcasecmp(cmd, "read")) {
uint8_t buf[128+1]; /* one more for the trailing \n */
int got;
got = atrf_buf_read(dsc, buf, sizeof(buf));
if (got < 0)
return netio_printf(netio, "-I/O error\n");
if (netio_printf(netio, "+%d ", got) < 0)
return -1;
buf[got] = '\n';
return netio_write(netio, buf, got+1);
}
if (!strcasecmp(cmd, "poll")) {
res = atrf_interrupt(dsc);
if (res < 0)
return netio_printf(netio, "-I/O error\n");
if (!res)
usleep(100*1000);
return netio_printf(netio, "+%d\n", res);
}
return netio_printf(netio, "-unrecognized command\n");
}
static void session(const char *driver, struct netio *netio)
{
struct atrf_dsc *dsc;
dsc = atrf_open(driver);
if (!dsc) {
netio_printf(netio, "-unable to open driver\n");
return;
}
if (netio_printf(netio,
"+connected to %s\n", driver ? driver : "default") < 0)
goto done;
while (1) {
char buf[100];
ssize_t got;
char last;
got = netio_read_until(netio, " \n", buf, sizeof(buf), &last);
if (got < 0)
break;
if (!got && last) {
netio_printf(netio, "-empty input\n");
continue;
}
buf[got] = 0;
switch (last) {
case ' ':
if (cmd_more(dsc, netio, buf) < 0)
goto done;
break;
case '\n':
if (cmd_zero(dsc, netio, buf) < 0)
goto done;
break;
case 0:
goto done;
default:
abort();
}
}
done:
atrf_close(dsc);
}
static void loop(const char *driver, int port)
{
struct sockaddr_in addr;
int s;
s = socket(PF_INET, SOCK_STREAM, 0);
if (s < 0) {
perror("socket");
exit(1);
}
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
if (bind(s, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
perror("bind");
exit(1);
}
if (listen(s, 0) < 0) {
perror("listen");
exit(1);
}
while (1) {
socklen_t len = sizeof(addr);
int s2;
struct netio *netio;
s2 = accept(s, (struct sockaddr *) &addr, &len);
if (s2 < 0) {
perror("accept");
continue;
}
if (verbose)
fprintf(stderr, "%s:%u\n", inet_ntoa(addr.sin_addr),
ntohs(addr.sin_port));
netio = netio_open(s2);
if (netio) {
session(driver, netio);
netio_close(netio);
} else {
if (close(s2) < 0)
perror("close");
}
}
}
static void usage(const char *name)
{
fprintf(stderr, "usage: %s [-d driver[:arg]] [-v ...] [port]\n", name);
exit(1);
}
int main(int argc, char **argv)
{
unsigned long port = DEFAULT_PORT;
const char *driver = NULL;
char *end;
int c;
while ((c = getopt(argc, argv, "d:v")) != EOF)
switch (c) {
case 'd':
driver = optarg;
break;
case 'v':
verbose++;
netio_verbose++;
break;
default:
usage(*argv);
}
switch (argc-optind) {
case 0:
break;
case 1:
port = strtoul(argv[optind], &end, 0);
if (*end || !port || port > 0xffff)
usage(*argv);
break;
default:
usage(*argv);
}
loop(driver, port);
return 0;
}