int sent_arppacket(int raw_sockfd, unsigned char * dst_ipaddr)
{
ARP_HEADER * arp;
char raw_buffer[46];
memset(dst_sockll.sll_addr, -1, sizeof(dst_sockll.sll_addr)); // set dmac addr FF:FF:FF:FF:FF:FF
if (raw_buffer == NULL)
{
perror("ARP: Oops, out of memory\r");
return 1;
}
bzero(raw_buffer, 46);
// Allow 14 bytes for the ethernet header
arp = (ARP_HEADER *)(raw_buffer);// + 14);
arp->hardware_type =htons(DIX_ETHERNET);
arp->protocol_type = htons(IP_PACKET);
arp->hwaddr_len = 6;
arp->ipaddr_len = 4;
arp->message_type = htons(ARP_REQUEST);
// My hardware address and IP addresses
memcpy(arp->source_hwaddr, my_hwaddr, 6);
memcpy(arp->source_ipaddr, my_ipaddr, 4);
// Destination hwaddr and dest IP addr
memcpy(arp->dest_hwaddr, broadcast_hwaddr, 6);
memcpy(arp->dest_ipaddr, dst_ipaddr, 4);
if( (sendto(raw_sockfd, raw_buffer, 46, 0, (struct sockaddr *)&dst_sockll, sizeof(dst_sockll))) < 0 )
{
perror("sendto");
return 1;
}
NMP_DEBUG_M("Send ARP Request success to: .%d.%d\n", (int *)dst_ipaddr[2],(int *)dst_ipaddr[3]);
return 0;
}
void
check_nmp_db(CLIENT_DETAIL_INFO_TABLE *p_client_tab, int client_no)
{
char new_mac[13];
char *search_list, *nv, *nvp, *b;
char *db_mac, *db_user_def, *db_device_name, *db_type, *db_http, *db_printer, *db_itune;
if(!nmp_client_list)
return;
NMP_DEBUG("check_nmp_db:\n");
search_list = strdup(nmp_client_list);
convert_mac_to_string(p_client_tab->mac_addr[client_no], new_mac);
NMP_DEBUG("search_list= %s\n", search_list);
if(strstr(search_list, new_mac)==NULL)
return;
nvp = nv = search_list;
while (nv && (b = strsep(&nvp, "<")) != NULL) {
if (vstrsep(b, ">", &db_mac, &db_user_def, &db_device_name, &db_type, &db_http, &db_printer, &db_itune) != 7)
continue;
NMP_DEBUG_M("DB: %s,%s,%s,%s,%s,%s,%s\n", db_mac, db_user_def, db_device_name, db_type, db_http, db_printer, db_itune);
if (!strcmp(db_mac, new_mac)) {
NMP_DEBUG("*** %s at DB!!! Update to memory\n",new_mac);
strcpy(p_client_tab->user_define[client_no], db_user_def);
strcpy(p_client_tab->device_name[client_no], db_device_name);
p_client_tab->type[client_no] = atoi(db_type);
p_client_tab->http[client_no] = atoi(db_http);
p_client_tab->printer[client_no] = atoi(db_printer);
p_client_tab->itune[client_no] = atoi(db_itune);
break;
}
}
SAFE_FREE(search_list);
}
int main(int argc, char *argv[])
{
int arp_sockfd, arp_getlen, i;
int send_count=0, file_num=0;
struct sockaddr_in router_addr, device_addr;
char router_ipaddr[17], router_mac[17], buffer[ARP_BUFFER_SIZE];
unsigned char scan_ipaddr[4]; // scan ip
FILE *fp_ip;
fd_set rfds;
ARP_HEADER * arp_ptr;
struct timeval tv1, tv2, arp_timeout;
int shm_client_detail_info_id;
int ip_dup, mac_dup, real_num;
int lock;
FILE *fp = fopen("/var/run/networkmap.pid", "w");
if(fp != NULL){
fprintf(fp, "%d", getpid());
fclose(fp);
}
#ifdef DEBUG
eval("rm", "/var/client*");
#endif
//Initial client tables
lock = file_lock("networkmap");
shm_client_detail_info_id = shmget((key_t)1001, sizeof(CLIENT_DETAIL_INFO_TABLE), 0666|IPC_CREAT);
if (shm_client_detail_info_id == -1){
fprintf(stderr,"shmget failed\n");
file_unlock(lock);
exit(1);
}
CLIENT_DETAIL_INFO_TABLE *p_client_detail_info_tab = (P_CLIENT_DETAIL_INFO_TABLE)shmat(shm_client_detail_info_id,(void *) 0,0);
//Reset shared memory
memset(p_client_detail_info_tab, 0x00, sizeof(CLIENT_DETAIL_INFO_TABLE));
p_client_detail_info_tab->ip_mac_num = 0;
p_client_detail_info_tab->detail_info_num = 0;
file_unlock(lock);
#ifdef NMP_DB
nmp_client_list = strdup(nvram_safe_get("nmp_client_list"));
NMP_DEBUG_M("NMP Client:\n%s\n", nmp_client_list);
signal(SIGUSR2, reset_db);
#endif
//Get Router's IP/Mac
strcpy(router_ipaddr, nvram_safe_get("lan_ipaddr"));
#ifdef RTCONFIG_RGMII_BRCM5301X
strcpy(router_mac, nvram_safe_get("et1macaddr"));
#else
strcpy(router_mac, nvram_safe_get("et0macaddr"));
#endif
inet_aton(router_ipaddr, &router_addr.sin_addr);
memcpy(my_ipaddr, &router_addr.sin_addr, 4);
//Prepare scan
networkmap_fullscan = 1;
nvram_set("networkmap_fullscan", "1");
if (argc > 1) {
if (strcmp(argv[1], "--bootwait") == 0) {
sleep(30);
}
}
if (strlen(router_mac)!=0) ether_atoe(router_mac, my_hwaddr);
signal(SIGUSR1, refresh_sig); //catch UI refresh signal
// create UDP socket and bind to "br0" to get ARP packet//
arp_sockfd = create_socket(INTERFACE);
if(arp_sockfd < 0)
perror("create socket ERR:");
else {
arp_timeout.tv_sec = 0;
arp_timeout.tv_usec = 50000;
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
dst_sockll = src_sockll; //Copy sockaddr info to dst
memset(dst_sockll.sll_addr, -1, sizeof(dst_sockll.sll_addr)); // set dmac= FF:FF:FF:FF:FF:FF
}
while(1)//main while loop
{
while(1) { //full scan and reflush recv buffer
fullscan:
if(networkmap_fullscan == 1) { //Scan all IP address in the subnetwork
if(scan_count == 0) {
asusdiscovery(); //find asus device
// (re)-start from the begining
memset(scan_ipaddr, 0x00, 4);
memcpy(scan_ipaddr, &router_addr.sin_addr, 3);
arp_timeout.tv_sec = 0;
arp_timeout.tv_usec = 50000;
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
NMP_DEBUG("Starting full scan!\n");
if(nvram_match("refresh_networkmap", "1")) {//reset client tables
lock = file_lock("networkmap");
memset(p_client_detail_info_tab, 0x00, sizeof(CLIENT_DETAIL_INFO_TABLE));
//p_client_detail_info_tab->detail_info_num = 0;
//p_client_detail_info_tab->ip_mac_num = 0;
file_unlock(lock);
nvram_unset("refresh_networkmap");
}
else {
int x = 0;
for(; x<255; x++)
p_client_detail_info_tab->exist[x]=0;
}
}
scan_count++;
scan_ipaddr[3]++;
if( scan_count<255 && memcmp(scan_ipaddr, my_ipaddr, 4) ) {
sent_arppacket(arp_sockfd, scan_ipaddr);
}
else if(scan_count==255) { //Scan completed
arp_timeout.tv_sec = 2;
arp_timeout.tv_usec = 0; //Reset timeout at monitor state for decase cpu loading
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
networkmap_fullscan = 0;
//scan_count = 0;
nvram_set("networkmap_fullscan", "0");
NMP_DEBUG("Finish full scan!\n");
}
}// End of full scan
memset(buffer, 0, ARP_BUFFER_SIZE);
arp_getlen=recvfrom(arp_sockfd, buffer, ARP_BUFFER_SIZE, 0, NULL, NULL);
if(arp_getlen == -1) {
if( scan_count<255)
goto fullscan;
else
break;
}
else {
arp_ptr = (ARP_HEADER*)(buffer);
NMP_DEBUG("*Receive an ARP Packet from: %d.%d.%d.%d to %d.%d.%d.%d:%02X:%02X:%02X:%02X - len:%d\n",
(int *)arp_ptr->source_ipaddr[0],(int *)arp_ptr->source_ipaddr[1],
(int *)arp_ptr->source_ipaddr[2],(int *)arp_ptr->source_ipaddr[3],
(int *)arp_ptr->dest_ipaddr[0],(int *)arp_ptr->dest_ipaddr[1],
(int *)arp_ptr->dest_ipaddr[2],(int *)arp_ptr->dest_ipaddr[3],
arp_ptr->dest_hwaddr[0],arp_ptr->dest_hwaddr[1],
arp_ptr->dest_hwaddr[2],arp_ptr->dest_hwaddr[3],
arp_getlen);
//Check ARP packet if source ip and router ip at the same network
if( !memcmp(my_ipaddr, arp_ptr->source_ipaddr, 3) ) {
swapbytes16(arp_ptr->message_type);
if( //ARP packet to router
(arp_ptr->message_type == 0x02 && // ARP response
memcmp(arp_ptr->dest_ipaddr, my_ipaddr, 4) == 0 && // dest IP
memcmp(arp_ptr->dest_hwaddr, my_hwaddr, 6) == 0) // dest MAC
||
(arp_ptr->message_type == 0x01 && // ARP request
memcmp(arp_ptr->dest_ipaddr, my_ipaddr, 4) == 0) // dest IP
){
//NMP_DEBUG(" It's an ARP Response to Router!\n");
NMP_DEBUG("*RCV %d.%d.%d.%d-%02X:%02X:%02X:%02X:%02X:%02X\n",
(int *)arp_ptr->source_ipaddr[0],(int *)arp_ptr->source_ipaddr[1],
(int *)arp_ptr->source_ipaddr[2],(int *)arp_ptr->source_ipaddr[3],
arp_ptr->source_hwaddr[0],arp_ptr->source_hwaddr[1],
arp_ptr->source_hwaddr[2],arp_ptr->source_hwaddr[3],
arp_ptr->source_hwaddr[4],arp_ptr->source_hwaddr[5]);
for(i=0; i<p_client_detail_info_tab->ip_mac_num; i++) {
ip_dup = memcmp(p_client_detail_info_tab->ip_addr[i], arp_ptr->source_ipaddr, 4);
mac_dup = memcmp(p_client_detail_info_tab->mac_addr[i], arp_ptr->source_hwaddr, 6);
if((ip_dup == 0) && (mac_dup == 0)) {
lock = file_lock("networkmap");
p_client_detail_info_tab->exist[i] = 1;
file_unlock(lock);
break;
}
else if((ip_dup != 0) && (mac_dup != 0)) {
continue;
}
else if( (scan_count>=255) && ((ip_dup != 0) && (mac_dup == 0)) ) {
NMP_DEBUG("IP changed, update immediately\n");
NMP_DEBUG("*CMP %d.%d.%d.%d-%02X:%02X:%02X:%02X:%02X:%02X\n",
p_client_detail_info_tab->ip_addr[i][0],p_client_detail_info_tab->ip_addr[i][1],
p_client_detail_info_tab->ip_addr[i][2],p_client_detail_info_tab->ip_addr[i][3],
p_client_detail_info_tab->mac_addr[i][0],p_client_detail_info_tab->mac_addr[i][1],
p_client_detail_info_tab->mac_addr[i][2],p_client_detail_info_tab->mac_addr[i][3],
p_client_detail_info_tab->mac_addr[i][4],p_client_detail_info_tab->mac_addr[i][5]);
lock = file_lock("networkmap");
memcpy(p_client_detail_info_tab->ip_addr[i],
arp_ptr->source_ipaddr, 4);
memcpy(p_client_detail_info_tab->mac_addr[i],
arp_ptr->source_hwaddr, 6);
p_client_detail_info_tab->exist[i] = 1;
file_unlock(lock);
/*
real_num = p_client_detail_info_tab->detail_info_num;
p_client_detail_info_tab->detail_info_num = i;
#ifdef NMP_DB
check_nmp_db(p_client_detail_info_tab, i);
#endif
FindAllApp(my_ipaddr, p_client_detail_info_tab);
FindHostname(p_client_detail_info_tab);
p_client_detail_info_tab->detail_info_num = real_num;
*/
break;
}
}
//NMP_DEBUG("Out check!\n");
//i=0, table is empty.
//i=num, no the same ip at table.
if(i==p_client_detail_info_tab->ip_mac_num){
lock = file_lock("networkmap");
memcpy(p_client_detail_info_tab->ip_addr[p_client_detail_info_tab->ip_mac_num],
arp_ptr->source_ipaddr, 4);
memcpy(p_client_detail_info_tab->mac_addr[p_client_detail_info_tab->ip_mac_num],
arp_ptr->source_hwaddr, 6);
p_client_detail_info_tab->exist[p_client_detail_info_tab->ip_mac_num] = 1;
#ifdef NMP_DB
check_nmp_db(p_client_detail_info_tab, i);
#endif
p_client_detail_info_tab->ip_mac_num++;
file_unlock(lock);
#ifdef DEBUG //Write client info to file
fp_ip=fopen("/var/client_ip_mac.txt", "a");
if (fp_ip==NULL) {
NMP_DEBUG("File Open Error!\n");
}
else {
NMP_DEBUG_M("Fill: %d-> %d.%d\n", i,p_client_detail_info_tab->ip_addr[i][2],p_client_detail_info_tab->ip_addr[i][3]);
fprintf(fp_ip, "%d.%d.%d.%d,%02X:%02X:%02X:%02X:%02X:%02X\n",
p_client_detail_info_tab->ip_addr[i][0],p_client_detail_info_tab->ip_addr[i][1],
p_client_detail_info_tab->ip_addr[i][2],p_client_detail_info_tab->ip_addr[i][3],
p_client_detail_info_tab->mac_addr[i][0],p_client_detail_info_tab->mac_addr[i][1],
p_client_detail_info_tab->mac_addr[i][2],p_client_detail_info_tab->mac_addr[i][3],
p_client_detail_info_tab->mac_addr[i][4],p_client_detail_info_tab->mac_addr[i][5]);
}
fclose(fp_ip);
#endif
}
}
else { //Nomo ARP Packet or ARP response to other IP
//Compare IP and IP buffer if not exist
for(i=0; i<p_client_detail_info_tab->ip_mac_num; i++) {
if( !memcmp(p_client_detail_info_tab->ip_addr[i], arp_ptr->source_ipaddr, 4) &&
!memcmp(p_client_detail_info_tab->mac_addr[i], arp_ptr->source_hwaddr, 6)) {
NMP_DEBUG_M("Find the same IP/MAC at the table!\n");
break;
}
}
if( i==p_client_detail_info_tab->ip_mac_num ) //Find a new IP or table is empty! Send an ARP request.
{
NMP_DEBUG("New device or IP/MAC changed!!\n");
if(memcmp(my_ipaddr, arp_ptr->source_ipaddr, 4))
sent_arppacket(arp_sockfd, arp_ptr->source_ipaddr);
else
NMP_DEBUG("New IP is the same as Router IP! Ignore it!\n");
}
}//End of Nomo ARP Packet
}//Source IP in the same subnetwork
}//End of arp_getlen != -1
} // End of while for flush buffer
/*
int y = 0;
while(p_client_detail_info_tab->type[y]!=0){
NMP_DEBUG("%d: %d.%d.%d.%d,%02X:%02X:%02X:%02X:%02X:%02X,%s,%d,%d,%d,%d,%d\n", y ,
p_client_detail_info_tab->ip_addr[y][0],p_client_detail_info_tab->ip_addr[y][1],
p_client_detail_info_tab->ip_addr[y][2],p_client_detail_info_tab->ip_addr[y][3],
p_client_detail_info_tab->mac_addr[y][0],p_client_detail_info_tab->mac_addr[y][1],
p_client_detail_info_tab->mac_addr[y][2],p_client_detail_info_tab->mac_addr[y][3],
p_client_detail_info_tab->mac_addr[y][4],p_client_detail_info_tab->mac_addr[y][5],
p_client_detail_info_tab->device_name[y],
p_client_detail_info_tab->type[y],
p_client_detail_info_tab->http[y],
p_client_detail_info_tab->printer[y],
p_client_detail_info_tab->itune[y],
p_client_detail_info_tab->exist[y]);
y++;
}
*/
//Find All Application of clients
//NMP_DEBUG("\ndetail ? ip : %d ? %d\n\n", p_client_detail_info_tab->detail_info_num, p_client_detail_info_tab->ip_mac_num);
if(p_client_detail_info_tab->detail_info_num < p_client_detail_info_tab->ip_mac_num) {
nvram_set("networkmap_status", "1");
FindAllApp(my_ipaddr, p_client_detail_info_tab);
FindHostname(p_client_detail_info_tab);
#ifdef DEBUG //Fill client detail info table
fp_ip=fopen("/var/client_detail_info.txt", "a");
if (fp_ip==NULL) {
NMP_DEBUG("File Open Error!\n");
}
else {
fprintf(fp_ip, "%s,%d,%d,%d,%d\n",
p_client_detail_info_tab->device_name[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->type[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->http[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->printer[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->itune[p_client_detail_info_tab->detail_info_num]);
fclose(fp_ip);
}
#endif
#ifdef NMP_DB
write_to_nvram(p_client_detail_info_tab);
#endif
p_client_detail_info_tab->detail_info_num++;
}
#ifdef NMP_DB
else {
NMP_DEBUG_M("commit_no, cli_no, updated: %d, %d, %d\n",
commit_no, p_client_detail_info_tab->detail_info_num, client_updated);
if( (commit_no != p_client_detail_info_tab->detail_info_num) || client_updated ) {
NMP_DEBUG("Commit nmp client list\n");
nvram_commit();
commit_no = p_client_detail_info_tab->detail_info_num;
client_updated = 0;
}
}
#endif
if(p_client_detail_info_tab->detail_info_num == p_client_detail_info_tab->ip_mac_num)
nvram_set("networkmap_status", "0"); // Done scanning and resolving
} //End of main while loop
close(arp_sockfd);
return 0;
}
void
write_to_nvram(CLIENT_DETAIL_INFO_TABLE *p_client_tab)
{
char new_mac[13], *dst_list;
char *nv, *nvp, *b, *search_list;
char *db_mac, *db_user_def, *db_device_name, *db_type, *db_http, *db_printer, *db_itune;
convert_mac_to_string(p_client_tab->mac_addr[p_client_tab->detail_info_num], new_mac);
NMP_DEBUG("*** write_to_nvram: %s ***\n",new_mac);
search_list = strdup(nmp_client_list);
b = strstr(search_list, new_mac);
if(b!=NULL) { //find the client in the DB
dst_list = malloc(sizeof(char)*10000);
NMP_DEBUG_M("client data in DB: %s\n", new_mac);
nvp = nv = b;
*(b-1) = '\0';
strcpy(dst_list, search_list);
NMP_DEBUG_M("dst_list= %s\n", dst_list);
//b++;
while (nv && (b = strsep(&nvp, "<")) != NULL) {
if (b == NULL) continue;
if (vstrsep(b, ">", &db_mac, &db_user_def, &db_device_name, &db_type, &db_http, &db_printer, &db_itune) != 7) continue;
NMP_DEBUG_M("%s,%s,%d,%d,%d,%d\n", db_mac, db_user_def, db_device_name, atoi(db_type), atoi(db_http), atoi(db_printer), atoi(db_itune));
if (!strcmp(p_client_tab->device_name[p_client_tab->detail_info_num], db_device_name) &&
p_client_tab->type[p_client_tab->detail_info_num] == atoi(db_type) &&
p_client_tab->http[p_client_tab->detail_info_num] == atoi(db_http) &&
p_client_tab->printer[p_client_tab->detail_info_num] == atoi(db_printer) &&
p_client_tab->itune[p_client_tab->detail_info_num] == atoi(db_itune) )
{
NMP_DEBUG("DATA the same!\n");
return;
}
sprintf(dst_list, "%s<%s<%s", dst_list, db_mac, db_user_def);
if (strcmp(p_client_tab->device_name[p_client_tab->detail_info_num], "")) {
NMP_DEBUG("Update device name: %s.\n", p_client_tab->device_name[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%s", dst_list, p_client_tab->device_name[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_device_name);
if (p_client_tab->type[p_client_tab->detail_info_num] != 6) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->type[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->type[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_type);
if (!strcmp(db_http, "0") ) {
client_updated = 1;
NMP_DEBUG("Update http: %d\n", p_client_tab->http[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->http[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_http);
if (!strcmp(db_printer, "0") ) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->printer[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->printer[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_printer);
if (!strcmp(db_itune, "0")) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->itune[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->itune[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_itune);
if(nvp != NULL) {
strcat(dst_list, "<");
strcat(dst_list, nvp);
}
nmp_client_list = strdup(dst_list);
free(dst_list);
NMP_DEBUG_M("*** Update nmp_client_list:\n%s\n", nmp_client_list);
nvram_set("nmp_client_list", nmp_client_list);
break;
}
}
else { //new client
NMP_DEBUG_M("new client: %d-%s,%s,%d\n",p_client_tab->detail_info_num,
new_mac,
p_client_tab->device_name[p_client_tab->detail_info_num],
p_client_tab->type[p_client_tab->detail_info_num]);
sprintf(nmp_client_list,"%s<%s>>%s>%d>%d>%d>%d", nmp_client_list,
new_mac,
p_client_tab->device_name[p_client_tab->detail_info_num],
p_client_tab->type[p_client_tab->detail_info_num],
p_client_tab->http[p_client_tab->detail_info_num],
p_client_tab->printer[p_client_tab->detail_info_num],
p_client_tab->itune[p_client_tab->detail_info_num]
);
nvram_set("nmp_client_list", nmp_client_list);
}
}
int main()
{
int arp_sockfd, arp_getlen, i;
int send_count=0, file_num=0;
struct sockaddr_in router_addr, device_addr;
char router_ipaddr[17], router_mac[17], buffer[512];
unsigned char scan_ipaddr[4]; // scan ip
FILE *fp_ip;
fd_set rfds;
ARP_HEADER * arp_ptr;
struct timeval tv1, tv2, arp_timeout;
int shm_client_detail_info_id;
FILE *fp = fopen("/var/run/networkmap.pid", "w");
if(fp != NULL){
fprintf(fp, "%d", getpid());
fclose(fp);
}
#ifdef DEBUG
eval("rm", "/var/client*");
#endif
//Initial client tables
spinlock_lock(SPINLOCK_Networkmap);
shm_client_detail_info_id = shmget((key_t)1001, sizeof(CLIENT_DETAIL_INFO_TABLE), 0666|IPC_CREAT);
if (shm_client_detail_info_id == -1){
fprintf(stderr,"shmget failed\n");
exit(1);
}
CLIENT_DETAIL_INFO_TABLE *p_client_detail_info_tab = (P_CLIENT_DETAIL_INFO_TABLE)shmat(shm_client_detail_info_id,(void *) 0,0);
//Reset shared memory
memset(p_client_detail_info_tab, 0x00, sizeof(CLIENT_DETAIL_INFO_TABLE));
p_client_detail_info_tab->ip_mac_num = 0;
p_client_detail_info_tab->detail_info_num = 0;
spinlock_unlock(SPINLOCK_Networkmap);
//Get Router's IP/Mac
strcpy(router_ipaddr, nvram_safe_get("lan_ipaddr"));
strcpy(router_mac, nvram_safe_get("et0macaddr"));
inet_aton(router_ipaddr, &router_addr.sin_addr);
memcpy(my_ipaddr, &router_addr.sin_addr, 4);
//Prepare scan
memset(scan_ipaddr, 0x00, 4);
memcpy(scan_ipaddr, &router_addr.sin_addr, 3);
networkmap_fullscan = 1;
nvram_set("networkmap_fullscan", "1");
if (strlen(router_mac)!=0) ether_atoe(router_mac, my_hwaddr);
signal(SIGUSR1, refresh_sig); //catch UI refresh signal
// create UDP socket and bind to "br0" to get ARP packet//
arp_sockfd = create_socket(INTERFACE);
if(arp_sockfd < 0)
perror("create socket ERR:");
else {
arp_timeout.tv_sec = 0;
arp_timeout.tv_usec = 10000;
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
dst_sockll = src_sockll; //Copy sockaddr info to dst
memset(dst_sockll.sll_addr, -1, sizeof(dst_sockll.sll_addr)); // set dmac= FF:FF:FF:FF:FF:FF
}
while(1)//main while loop
{
while(1) { //full scan and reflush recv buffer
fullscan:
if(networkmap_fullscan == 1) { //Scan all IP address in the subnetwork
if(scan_count == 0) {
arp_timeout.tv_sec = 0;
arp_timeout.tv_usec = 10000;
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
NMP_DEBUG("Starting full scan!\n");
//reset client tables
spinlock_lock(SPINLOCK_Networkmap);
memset(p_client_detail_info_tab, 0x00, sizeof(CLIENT_DETAIL_INFO_TABLE));
p_client_detail_info_tab->detail_info_num = 0;
spinlock_unlock(SPINLOCK_Networkmap);
}
scan_count++;
scan_ipaddr[3]++;
if( scan_count<255 && memcmp(scan_ipaddr, my_ipaddr, 4) ) {
sent_arppacket(arp_sockfd, scan_ipaddr);
}
else if(scan_count>255) { //Scan completed
arp_timeout.tv_sec = 1;
arp_timeout.tv_usec = 500000; //Reset timeout at monitor state for decase cpu loading
setsockopt(arp_sockfd, SOL_SOCKET, SO_RCVTIMEO, &arp_timeout, sizeof(arp_timeout));//set receive timeout
networkmap_fullscan = 0;
//scan_count = 0;
nvram_set("networkmap_fullscan", "0");
NMP_DEBUG("Finish full scan!\n");
}
}// End of full scan
arp_getlen=recvfrom(arp_sockfd, buffer, 512, 0, NULL, NULL);
if(arp_getlen == -1) {
if( scan_count<255)
goto fullscan;
else
break;
}
else {
arp_ptr = (ARP_HEADER*)(buffer);
NMP_DEBUG("*Receive an ARP Packet from: %d.%d.%d.%d, len:%d\n",
(int *)arp_ptr->source_ipaddr[0],(int *)arp_ptr->source_ipaddr[1],
(int *)arp_ptr->source_ipaddr[2],(int *)arp_ptr->source_ipaddr[3],
arp_getlen);
//Check ARP packet if source ip and router ip at the same network
if( !memcmp(my_ipaddr, arp_ptr->source_ipaddr, 3) ) {
swapbytes16(arp_ptr->message_type);
//ARP Response packet to router
if( arp_ptr->message_type == 0x02 && // ARP response
memcmp(arp_ptr->dest_ipaddr, my_ipaddr, 4) == 0 && // dest IP
memcmp(arp_ptr->dest_hwaddr, my_hwaddr, 6) == 0) // dest MAC
{
//NMP_DEBUG(" It's an ARP Response to Router!\n");
for(i=0; i<p_client_detail_info_tab->ip_mac_num; i++) {
if( !memcmp(p_client_detail_info_tab->ip_addr[i], arp_ptr->source_ipaddr, 4) )
break;
}
//i=0, table is empty.
//i=num, no the same ip at table.
if(i==p_client_detail_info_tab->ip_mac_num){
spinlock_lock(SPINLOCK_Networkmap);
memcpy(p_client_detail_info_tab->ip_addr[p_client_detail_info_tab->ip_mac_num],
arp_ptr->source_ipaddr, 4);
memcpy(p_client_detail_info_tab->mac_addr[p_client_detail_info_tab->ip_mac_num],
arp_ptr->source_hwaddr, 6);
p_client_detail_info_tab->ip_mac_num++;
spinlock_unlock(SPINLOCK_Networkmap);
#ifdef DEBUG //Write client info to file
fp_ip=fopen("/var/client_ip_mac.txt", "a");
if (fp_ip==NULL) {
printf("File Open Error!\n");
}
else {
printf("Fill: %d-> %d.%d", i,p_client_detail_info_tab->ip_addr[i][2],p_client_detail_info_tab->ip_addr[i][3]);
fprintf(fp_ip, "%d.%d.%d.%d,%02X:%02X:%02X:%02X:%02X:%02X\n",
p_client_detail_info_tab->ip_addr[i][0],p_client_detail_info_tab->ip_addr[i][1],
p_client_detail_info_tab->ip_addr[i][2],p_client_detail_info_tab->ip_addr[i][3],
p_client_detail_info_tab->mac_addr[i][0],p_client_detail_info_tab->mac_addr[i][1],
p_client_detail_info_tab->mac_addr[i][2],p_client_detail_info_tab->mac_addr[i][3],
p_client_detail_info_tab->mac_addr[i][4],p_client_detail_info_tab->mac_addr[i][5]);
}
fclose(fp_ip);
#endif
}
}
else { //Nomo ARP Packet or ARP response to other IP
//Compare IP and IP buffer if not exist
for(i=0; i<p_client_detail_info_tab->ip_mac_num; i++) {
if( !memcmp(p_client_detail_info_tab->ip_addr[i], arp_ptr->source_ipaddr, 4) ) {
NMP_DEBUG_M("Find the same IP at the table!\n");
break;
}
}
if( i==p_client_detail_info_tab->ip_mac_num ) //Find a new IP or table is empty! Send an ARP request.
{
NMP_DEBUG("New IP\n");
if(memcmp(my_ipaddr, arp_ptr->source_ipaddr, 4))
sent_arppacket(arp_sockfd, arp_ptr->source_ipaddr);
else
NMP_DEBUG("New IP is the same as Router IP! Ignore it!\n");
}
}//End of Nomo ARP Packet
}//Source IP in the same subnetwork
}//End of arp_getlen != -1
} // End of while for flush buffer
//Find All Application of clients
if(p_client_detail_info_tab->detail_info_num < p_client_detail_info_tab->ip_mac_num) {
FindAllApp(my_ipaddr, p_client_detail_info_tab);
#ifdef DEBUG //Fill client detail info table
fp_ip=fopen("/var/client_detail_info.txt", "a");
if (fp_ip==NULL) {
printf("File Open Error!\n");
}
else {
fprintf(fp_ip, "%s,%d,%d,%d,%d\n",
p_client_detail_info_tab->device_name[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->type[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->http[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->printer[p_client_detail_info_tab->detail_info_num],
p_client_detail_info_tab->itune[p_client_detail_info_tab->detail_info_num]);
fclose(fp_ip);
}
#endif
p_client_detail_info_tab->detail_info_num++;
}
} //End of main while loop
close(arp_sockfd);
return 0;
}
void
write_to_cfg_node(CLIENT_DETAIL_INFO_TABLE *p_client_tab)
{
char new_mac[13], *dst_list;
char *nv, *nvp, *b = NULL, *search_list = NULL;
char *db_mac, *db_user_def, *db_device_name, *db_type, *db_http, *db_printer, *db_itune;
FILE *fp = NULL;
convert_mac_to_string(p_client_tab->mac_addr[p_client_tab->detail_info_num], new_mac);
//Andy Chiu, 2014/10/27. get station list
int count = 0;
int ret = 0;
char *tmpbuf = NULL;
NMP_DEBUG("*** write_to_cfg_node: %s ***\n",new_mac);
if(nmp_client_list)
{
search_list = strdup(nmp_client_list);
b = strstr(search_list, new_mac);
}
if(b!=NULL) { //find the client in the DB
dst_list = malloc(sizeof(char)*10000);
NMP_DEBUG_M("client data in DB: %s\n", new_mac);
nvp = nv = b;
*(b-1) = '\0';
strcpy(dst_list, search_list);
NMP_DEBUG_M("dst_list= %s\n", dst_list);
//b++;
while (nv && (b = strsep(&nvp, "<")) != NULL) {
if (b == NULL) continue;
if (vstrsep(b, ">", &db_mac, &db_user_def, &db_device_name, &db_type, &db_http, &db_printer, &db_itune) != 7) continue;
NMP_DEBUG_M("%s,%s,%d,%d,%d,%d\n", db_mac, db_user_def, db_device_name, atoi(db_type), atoi(db_http), atoi(db_printer), atoi(db_itune));
if (!strcmp(p_client_tab->device_name[p_client_tab->detail_info_num], db_device_name) &&
p_client_tab->type[p_client_tab->detail_info_num] == atoi(db_type) &&
p_client_tab->http[p_client_tab->detail_info_num] == atoi(db_http) &&
p_client_tab->printer[p_client_tab->detail_info_num] == atoi(db_printer) &&
p_client_tab->itune[p_client_tab->detail_info_num] == atoi(db_itune) )
{
NMP_DEBUG("DATA the same!\n");
return;
}
sprintf(dst_list, "%s<%s<%s", dst_list, db_mac, db_user_def);
if (strcmp(p_client_tab->device_name[p_client_tab->detail_info_num], "")) {
NMP_DEBUG("Update device name: %s.\n", p_client_tab->device_name[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%s", dst_list, p_client_tab->device_name[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_device_name);
if (p_client_tab->type[p_client_tab->detail_info_num] != 6) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->type[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->type[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_type);
if (!strcmp(db_http, "0") ) {
client_updated = 1;
NMP_DEBUG("Update http: %d\n", p_client_tab->http[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->http[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_http);
if (!strcmp(db_printer, "0") ) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->printer[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->printer[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_printer);
if (!strcmp(db_itune, "0")) {
client_updated = 1;
NMP_DEBUG("Update type: %d\n", p_client_tab->itune[p_client_tab->detail_info_num]);
sprintf(dst_list, "%s>%d", dst_list, p_client_tab->itune[p_client_tab->detail_info_num]);
}
else
sprintf(dst_list, "%s>%s", dst_list, db_itune);
if(nvp != NULL) {
strcat(dst_list, "<");
strcat(dst_list, nvp);
}
//Andy Chiu, 2014/10/22. Fix memory leak
SAFE_FREE(nmp_client_list);
nmp_client_list = strdup(dst_list);
SAFE_FREE(dst_list);
NMP_DEBUG_M("*** Update nmp_client_list:\n%s\n", nmp_client_list);
write_to_file(nmp_client_path, nmp_client_list, strlen(nmp_client_list), "w");
//nvram_set("nmp_client_list", nmp_client_list);
break;
}
}
else { //new client
NMP_DEBUG_M("new client: %d-%s,%s,%d\n",p_client_tab->detail_info_num,
new_mac,
p_client_tab->device_name[p_client_tab->detail_info_num],
p_client_tab->type[p_client_tab->detail_info_num]);
tmpbuf = malloc(10000);
if(!tmpbuf)
{
NMP_DEBUG("[%s, %d]Can't alloc memory\n", __FUNCTION__, __LINE__);
return;
}
//sprintf(nmp_client_list,"%s<%s>>%s>%d>%d>%d>%d", nmp_client_list,
sprintf(tmpbuf,"%s<%s>>%s>%d>%d>%d>%d", nmp_client_list? nmp_client_list: "",
new_mac,
p_client_tab->device_name[p_client_tab->detail_info_num],
p_client_tab->type[p_client_tab->detail_info_num],
p_client_tab->http[p_client_tab->detail_info_num],
p_client_tab->printer[p_client_tab->detail_info_num],
p_client_tab->itune[p_client_tab->detail_info_num]
);
SAFE_FREE(nmp_client_list);
nmp_client_list = strdup(tmpbuf);
SAFE_FREE(tmpbuf);
//nvram_set("nmp_client_list", nmp_client_list);
NMP_DEBUG("[%s, %d]\n", __FUNCTION__, __LINE__);
write_to_file(nmp_client_path, nmp_client_list, strlen(nmp_client_list), "w");
NMP_DEBUG("[%s, %d]\n", __FUNCTION__, __LINE__);
}
SAFE_FREE(search_list);
//Andy Chiu, 2014/10/27. update the cfg_node
int i;
char buf[128];
fp = fopen(cl_path, "w"); //Andy Chiu, 2014/12/03
count = 0;
for(i = 0; i < 255; ++i)
{
if(p_client_tab->exist[i]) //item exist
{
char attr[64];
char buf[256];
sprintf(buf, "<%d.%d.%d.%d>%02x:%02x:%02x:%02x:%02x:%02x>%s>%d>%d>%d>%d", p_client_tab->ip_addr[i][0],
p_client_tab->ip_addr[i][1], p_client_tab->ip_addr[i][2], p_client_tab->ip_addr[i][3],
p_client_tab->mac_addr[i][0], p_client_tab->mac_addr[i][1] , p_client_tab->mac_addr[i][2],
p_client_tab->mac_addr[i][3], p_client_tab->mac_addr[i][4], p_client_tab->mac_addr[i][5],
p_client_tab->device_name[i], p_client_tab->type[i], p_client_tab->http[i], p_client_tab->printer[i], p_client_tab->itune[i]);
sprintf(attr, "dev%d", count);
// ret = tcapi_set("ClientList_Entry", attr, buf);
// if(ret)
// NMP_DEBUG_M("set device(%d) failed!(%d)(%s)\n", count, ret, buf);
if(fp) //Andy Chiu, 2014/12/03
fwrite(buf, sizeof(char), strlen(buf), fp);
++count;
}
if( count >= p_client_tab->detail_info_num + 1)
break;
}
if(fp) //Andy Chiu, 2014/12/03
fclose(fp);
sprintf(buf, "%d", p_client_tab->detail_info_num + 1);
ret = tcapi_set("ClientList_Common", "size", buf);
if(ret)
NMP_DEBUG_M("set ClientList_Common:size failed!\n");
NMP_DEBUG_M("new client list size is %d\n", p_client_tab->detail_info_num + 1);
}
