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:
parent
3b0ee3c00a
commit
4ad0383020
16
tools/atrf-proxy/Makefile
Normal file
16
tools/atrf-proxy/Makefile
Normal 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
68
tools/atrf-proxy/PROTOCOL
Normal 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.
|
331
tools/atrf-proxy/atrf-proxy.c
Normal file
331
tools/atrf-proxy/atrf-proxy.c
Normal 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;
|
||||
}
|
Loading…
Reference in New Issue
Block a user