//If p2p device becomes GO, we still polling driver status //to check whether some other p2p devices connected void *polling_client(void *arg) { struct p2p *p=(struct p2p*)arg; while( p->res_go == 0 ){ if( p->no_sta_connected > 0 && ( p->wpsing == _FALSE ) ) { if( read_all_sta(p) == _FALSE ) { p->no_sta_connected--; } } p2p_status(p, 0); if( p->status == P2P_STATE_RX_PROVISION_DIS_REQ || p->status == P2P_STATE_GONEGO_FAIL || p->status == P2P_STATE_GONEGO_ING ) { p->thread_trigger = THREAD_GO ; char peer_devaddr[18]; char peer_req_cm[4]; memset( peer_devaddr, 0x00, 18); memset( peer_req_cm, 0x00, 4); p2p_peer_devaddr(p, peer_devaddr); p2p_peer_req_cm(p, peer_req_cm); p2p_peer_info(p, p->peer_devaddr, peer_req_cm); ui_screen(p); //strncpy(p->peer_devaddr, peer_devaddr, 17); if( (strncmp( peer_req_cm, "dis", 3) == 0) || (strncmp( peer_req_cm, "lab", 3) == 0) ) { printf("Here is your PIN, insert c to continue: %d\n", p->pin); } else if( (strncmp( peer_req_cm, "pbc", 3) == 0) ) { printf("Please push b to accept:\n", p->status); } else if( (strncmp( peer_req_cm, "pad", 3) == 0) ) { printf("Please insert peer PIN code:\n"); } break; } usleep( POLLING_INTERVAL ); } return NULL; }
//Successively query local device status with interval "POLLING_INTERVAL" //when status == P2P_STATE_RX_PROVISION_DIS_REQ, //it require user to insert corresponding WPS config method void *polling_status(void *arg) { struct p2p *p=(struct p2p*)arg; while( p->res == 0 ){ p2p_status(p, 0); if( p->status == P2P_STATE_RX_PROVISION_DIS_REQ ) { p->thread_trigger = THREAD_DEVICE ; char peer_devaddr[18]; char peer_req_cm[4]; memset( peer_devaddr, 0x00, 18); memset( peer_req_cm, 0x00, 4); p2p_peer_devaddr(p, peer_devaddr); p2p_peer_req_cm(p, peer_req_cm); p2p_peer_info(p, p->peer_devaddr, peer_req_cm); ui_screen(p); //strncpy(p->peer_devaddr, peer_devaddr, 17); if( (strncmp( peer_req_cm, "dis", 3) == 0) || (strncmp( peer_req_cm, "lab", 3) == 0) ) { printf("Here is your PIN, insert c to continue: %d\n", p->pin); } else if( (strncmp( peer_req_cm, "pbc", 3) == 0) ) { printf("Please push b to accept:\n"); } else if( (strncmp( peer_req_cm, "pad", 3) == 0) ) { printf("Please insert peer PIN code:\n"); } break; } usleep( POLLING_INTERVAL ); } return NULL; }
//int main() int main(int argc, char **argv) { char peerifa[40] = { 0x00 }; char scan[CMD_SZ]; struct p2p p2pstruct; struct p2p *p=NULL; p = &p2pstruct; if( p != NULL) { memset(p, 0x00, sizeof(struct p2p)); init_p2p(p); } strcpy(p->ifname, argv[1] ); /* Disable P2P functionalities at first*/ p->enable=P2P_ROLE_DISABLE; p2p_enable(p); p2p_get_hostapd_conf(p); usleep(50000); rename_intf(p); do { ui_screen(p); printf("*insert cmd:"); memset( scan, 0x00, CMD_SZ ); scanf("%s", scan); if( p->thread_trigger == THREAD_NONE ) //Active mode for user interface { if( strncmp(scan, "e", 1) == 0 ) //Enable { p->show_scan_result = 1; ui_screen(p); printf("Please insert enable mode;[0]Disable, [1]Device, [2]Client, [3]GO:"); scanf("%d",&p->enable); p2p_enable(p); p->show_scan_result = 1; } else if( strncmp(scan, "a", 1) == 0 ) // Scan P2P device { p2p_scan(p); p->show_scan_result = 1; } else if( strncmp(scan, "d", 1) == 0 ) // Set device name { p->p2p_get = 0; printf("Please insert device name :"); scanf("\n%[^\n]", p->dev_name); p2p_setDN(p); p->show_scan_result = 1; } else if( strncmp(scan, "i", 1) == 0 ) // Intent { p->show_scan_result = 1; ui_screen(p); printf("Please insert intent from [0~15(must be softap)] :"); scanf("%d",&p->intent); p2p_intent(p); p->show_scan_result = 1; } else if( strncmp(scan, "w", 1) == 0 ) // WPS_info { p->show_scan_result = 1; ui_screen(p); printf("Please insert WPS method\n"); printf("[0]None, [1]Peer Display PIN, [2]Self Display Pin, [3]PBC :"); scanf("%d",&p->wps_info); p2p_wpsinfo(p); p->show_scan_result = 1; } else if( strncmp(scan, "c", 1) == 0 ) // PIN_code { char ins_no[CMD_SZ], ins_no_again[CMD_SZ]; memset(ins_no, 0x00, CMD_SZ); strcpy(ins_no, "Please insert 8-digit number, eg:12345670 :" ); memset(ins_no_again, 0x00, CMD_SZ); strcpy(ins_no_again, "Invalid number, insert again, eg:12345670 :" ); p2p_pincode(p, ins_no, ins_no_again); p->show_scan_result = 1; } else if( strncmp(scan, "m", 1) == 0 ) // Set peer device address { p->show_scan_result = 1; ui_screen(p); printf("Please insert number in scan list:"); p2p_devaddr(p); p->show_scan_result = 1; } else if( strncmp(scan, "r", 1) == 0 ) // Get role { p2p_role(p,1); p->show_scan_result = 1; } else if( strncmp(scan, "s", 1) == 0 ) // Get status { p2p_status(p, 1); p->show_scan_result = 1; } else if( strncmp(scan, "p", 1) == 0 ) // Provision discovery { char msg[CMD_SZ]; memset( msg, 0x00, CMD_SZ ); char dis_msg[CMD_SZ]; memset( dis_msg, 0x00, CMD_SZ ); char label_msg[CMD_SZ]; memset( label_msg, 0x00, CMD_SZ ); if(strncmp(p->peer_devaddr, "00:00:00:00:00:00", 17) == 0) { strcpy( msg, "Please insert peer P2P device at first" ); p2p_prov_disc_no_addr(p, msg); p->show_scan_result = 1; } else { strcpy( msg, "Please insert WPS configuration method ;[0]display, [1]keypad, [2]pbc, [3]label:\n" ); strcpy( dis_msg, "Please insert PIN code displays on peer device screen:" ); strcpy( label_msg, "Please insert PIN code displays on peer label:" ); p2p_prov_disc(p, msg, dis_msg, label_msg); } } else if( strncmp(scan, "n", 1) == 0 ) // Set negotiation { p2p_set_nego(p); } else if( strncmp(scan, "f", 1) == 0 ) // Reflash current state { p->show_scan_result = 1; p2p_status(p, 0); p2p_role(p, 0); p2p_ifaddr(p); if( p->status == P2P_STATE_RX_PROVISION_DIS_REQ ) { char peer_devaddr[18]; char peer_req_cm[4]; memset( peer_devaddr, 0x00, 18); memset( peer_req_cm, 0x00, 4); p2p_peer_devaddr(p, peer_devaddr); p2p_peer_req_cm(p, peer_req_cm); p2p_peer_info(p, p->peer_devaddr, peer_req_cm); } #ifndef P2P_AUTO else { if( p->role == P2P_ROLE_CLIENT ) { p2p_client_mode(p); } else if( p->role == P2P_ROLE_GO ) { p2p_go_mode(p); } } #endif //P2P_AUTO } else if( strncmp(scan, "x", 1) == 0 ) // Start wpa_supplicant/hostapd { if( p->role == P2P_ROLE_CLIENT ) { p2p_client_mode(p); } else if( p->role == P2P_ROLE_GO ) { p2p_go_mode(p); } } else if( strncmp(scan, "h", 1) == 0 ) // Set operation channel { char msg[CMD_SZ]; memset( msg, 0x00, CMD_SZ ); strcpy( msg, "Please insert desired operation channel:" ); p2p_set_opch(p, msg, 1); p->show_scan_result = 1; } else if( strncmp(scan, "t", 1) == 0 ) // Set SoftAP ssid { char msg[CMD_SZ]; memset( msg, 0x00, CMD_SZ ); strcpy( msg, "Please insert desired SoftAP ssid:" ); p2p_softap_ssid(p, msg, 1); p->show_scan_result = 1; } else if( strncmp(scan, "l", 1) == 0 ) // Set Listen channel { char msg[CMD_SZ]; memset( msg, 0x00, CMD_SZ ); strcpy( msg, "Please insert desired Listen channel, only ch.1.6.11 are available:" ); p2p_listen_ch(p, msg); p->show_scan_result = 1; } else if( strncmp(scan, "q", 1) == 0 ) // Quit { if( p->res == 0 ) p->res = 1; if( p->res_go == 0 ) p->res_go = 1; break; } else // Insert wrong commamd { p->p2p_get=1; p->show_scan_result = 1; memset( p->print_line, 0x00, CMD_SZ ); sprintf( p->print_line, " BAD argument"); } } else if( p->thread_trigger == THREAD_DEVICE ) //Passive mode for user interface { p->thread_trigger = THREAD_NONE ; if( strncmp(scan, "b", 1) == 0 ) { p->wps_info=3; p2p_wpsinfo(p); p2p_status(p, 0); if(p->status != P2P_STATE_GONEGO_OK) { p2p_set_nego(p); } else { p2p_role(p,0); if( p->role == P2P_ROLE_CLIENT ) { p2p_client_mode(p); } else if( p->role == P2P_ROLE_GO ) { p2p_go_mode(p); } } } else if( strncmp(scan, "c", 1) == 0 ) { p->wps_info=2; p2p_wpsinfo(p); p2p_status(p, 0); if(p->status != P2P_STATE_GONEGO_OK) { p2p_set_nego(p); } else { p2p_role(p,0); p2p_ifaddr(p); if( p->role == P2P_ROLE_CLIENT ) { p2p_client_mode(p); } else if( p->role == P2P_ROLE_GO ) { p2p_go_mode(p); } } } else if( ('0' <= *scan ) && ( *scan <= '9') ) { printf("In passive pin code\n"); p->pin = atoi(scan); p->wps_info=1; p2p_wpsinfo(p); p2p_set_nego(p); } } else if( p->thread_trigger == THREAD_GO ) //Passive mode for user interface { p->thread_trigger = THREAD_NONE ; if( strncmp(scan, "b", 1) == 0 ) { p->wps_info=3; p2p_wpsinfo(p); } else if( strncmp(scan, "c", 1) == 0 ) { p->wps_info=2; p2p_wpsinfo(p); } else if( ('0' <= *scan ) && ( *scan <= '9') ) { printf("In passive pin code\n"); p->pin = atoi(scan); p->wps_info=1; p2p_wpsinfo(p); } p2p_go_mode(p); } } while( 1 ); /* Disable P2P functionalities when exits*/ p->enable= -1 ; p2p_enable(p); system( "rm -f ./supp_status.txt" ); system( "rm -f ./temp.txt" ); system( "rm -f ./scan.txt" ); system( "rm -f ./peer.txt" ); system( "rm -f ./status.txt" ); system( "rm -f ./cm.txt" ); return 0; }