// ----------------------------------------------------------------------------
// Commander functions:
// ----------------------------------------------------------------------------
static void command(char cmd)
{
switch(cmd) {
case 'l': ioLed(2); break; // Turn test LED off
case 'm': scanLED(0); break; // Turn Motor off
case 'M': scanLED(1); break; // Turn Motor on
case 'h': menu(); break; // show menu
case 'd': disconnect(); break; // disconnect station
case 'c': connect_st_mode(); break; // connect
case 'i': wifi_check_ip(NULL); break; // Test for IP and send
case 'u': create_udp(); break; // send test message
case 's': send_msg(); break; // send test message
case 'w': wifiStartScan(); break; // Scan for AP's
case 'a': logwifiaps(); break; // Scan for AP's and log via UDP
default: break;
}
}
int main(int argc, char **argv)
{
struct udp_state *ud;
struct katcp_dispatch *d;
unsigned int result;
struct timeval now;
fd_set fsr;
char *ip_addr = NULL;
uint32_t address, length;
int i, j, c, pos;
int wait, nooftries;
int port = 0;
int rw_flag = 0;
i = j = 1;
pos = 0;
wait = 0;
nooftries = 10;
while (i < argc) {
if (argv[i][0] == '-') {
c = argv[i][j];
switch (c) {
case '\0':
j = 1;
i++;
break;
case '-' :
j++;
break;
case 'h' :
fprintf(stderr, "usage: %s -R [-i ipaddress] [-p port] address length\n", argv[0]);
return 0;
break;
case 'i' :
j++;
if(argv[i][j] == '\0'){
j = 0;
i++;
}
if(i >= argc){
fprintf(stderr, "%s: option -%c requires a parameter\n", argv[0], c);
}
ip_addr = argv[i] + j;
i++;
j = 1;
break;
case 'p' :
j++;
if(argv[i][j] == '\0'){
j = 0;
i++;
}
if(i >= argc){
fprintf(stderr, "%s: option -%c requires a parameter\n", argv[0], c);
}
port = atoi(argv[i] + j);
#if DEBUG
fprintf(stderr, "port number is %d\n", port);
#endif
i++;
j = 1;
break;
case 'R' :
rw_flag = 1;
i++;
break;
default:
fprintf(stderr, "%s: unknown option -%c\n", argv[0], argv[i][j]);
return 2;
}
} else {
pos = i;
i = argc;
}
}
d = setup_katcp(STDOUT_FILENO);
if(d == NULL){
fprintf(stderr, "setup katcp failed\n");
return EX_OSERR;
}
ud = create_udp(d);
if(ud == NULL){
fprintf(stderr, "create udp failed\n");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to allocate local udp state");
write_katcp(d);
return EX_OSERR;
}
#if 0
if(connect_udp(d, ud, port) < 0){
fprintf(stderr, "connect udp failed\n");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to bind udp");
return EX_OSERR;
}
#endif
ud->u_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if(ud->u_fd < 0){
fprintf(stderr, "unable to create udp socket:\n ");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to create udp socket: %s", strerror(errno));
return -1;
}
ud->u_rw = rw_flag;
for(;;){
FD_ZERO(&fsr);
FD_SET(ud->u_fd, &fsr);
address = strtol(argv[pos], NULL, 16);
length = strtol(argv[pos + 1], NULL, 16);
#if DEBUG
printf("pos:%d, address[%x] and length[%x]\n", pos, address, length);
#endif
send_udp(d, ud, ip_addr, port, address, length);
now.tv_sec = 10;
now.tv_usec = 0;
result = select(ud->u_fd + 1, &fsr, NULL, NULL, &now);
if(result == 0){
/* Resend again after timeout */
printf("Resending udp again\n");
send_udp(d, ud, ip_addr, port, address, length);
}
for(wait = 0; wait < nooftries; wait++){
if(FD_ISSET(ud->u_fd, &fsr)){
result = rcv_udp(d, ud);
if(!result){
return EX_OK;
}
}
}
return EX_OSERR;
}
destroy_udp(d, ud);
shutdown_katcp(d);
return EX_OK;
}
