// ---------------------------------------------------------------------------- // 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; }