associate sta with ssid of ap's

[my-code/hdw-sniff.git] / parse.c

diff --git a/parse.c b/parse.c
index 494ac06..c4a8891 100644 (file)
--- a/parse.c
+++ b/parse.c
@@ -50,6 +50,7 @@ void parse_package(unsigned char *ptr,const struct pcap_pkthdr *pcap_header,cons
   t_info *info;
   int i;
   t_sta new_sta;
+  t_sta *cmp_sta;
   t_sta *sta;
   unsigned char *package=NULL;
   t_frame4_hdr *f4hdr;
@@ -73,15 +74,6 @@ void parse_package(unsigned char *ptr,const struct pcap_pkthdr *pcap_header,cons
   new=0;
   foo=0;
 
-  if(info->dump_fd!=0) {
-    ret=write(info->dump_fd,pcap_header,sizeof(struct pcap_pkthdr));
-    if(ret!=sizeof(struct pcap_pkthdr))
-      display_console(info,"warning, pcap header write failed!");
-    ret=write(info->dump_fd,package,pcap_header->caplen);
-    if(ret!=pcap_header->caplen)
-      display_console(info,"warning, package write failed!");
-  }
-  
   /* prism or ieee802.11 header ? */
   if(info->mode&MODE_IEEE80211) {
     package=(unsigned char *)pkg;
@@ -172,6 +164,7 @@ void parse_package(unsigned char *ptr,const struct pcap_pkthdr *pcap_header,cons
           memcpy(new_sta.bssid,f3hdr->addr3,ADDR_LEN);
         }
       }
+
       ret=list_search_data(&(info->sniffed_sta),&new_sta,ADDR_LEN);
       if((ret==L_EMPTY_LIST)|(ret==L_NO_SUCH_ELEMENT)) {
         list_add_element(&(info->sniffed_sta),&new_sta,sizeof(t_sta));
@@ -179,6 +172,23 @@ void parse_package(unsigned char *ptr,const struct pcap_pkthdr *pcap_header,cons
         new=1;
       }
       else sta=(t_sta *)info->sniffed_sta.current->data;
+      if(!(sta->ap)) {
+        ret=list_count(&(info->sniffed_sta));
+        list_reset(&(info->sniffed_sta));
+        for(i=0;i<ret;i++) {
+          cmp_sta=(t_sta *)info->sniffed_sta.current->data;
+          if(!memcmp(cmp_sta->bssid,sta->bssid,ADDR_LEN)) {
+            if(FCTL_FROMDS(package[0]))
+              snprintf(sta->ssid,MAX_SSID_LEN,"<- %s",cmp_sta->ssid);
+            else if(FCTL_TODS(package[0]))
+              snprintf(sta->ssid,MAX_SSID_LEN,"-> %s",cmp_sta->ssid);
+            else
+              snprintf(sta->ssid,MAX_SSID_LEN,"<> %s",cmp_sta->ssid);
+            break;
+          }
+          list_next(&(info->sniffed_sta));
+        }
+      }
       // fill in stuff ...
       sta->count_data++;
       switch_active_state(&(sta->active));
@@ -212,5 +222,6 @@ void parse_package(unsigned char *ptr,const struct pcap_pkthdr *pcap_header,cons
     display_console(info,"last: got data frame");
   }
 
+  if(info->dump_handle!=NULL) pcap_dump((unsigned char *)(info->dump_handle),pcap_header,pkg);
 
 }