#include <stdio.h>
#include <string.h>
#include <stdlib.h>

struct aa {
 char name[16];
 char val[8];
 struct aa * next ;
}
;
int main(int argc,char*argv[])
{
    FILE*fp,*fp1 ;
    char buffer[128];
    char temp[128];
    struct aa*zhi ;
    struct aa*head,*curren,*tmp ;
    char*p1,*p2,*p3,*p4,*p5 ;
    int len ;
    int biaoz=0 ;
    head=NULL ;
   
    if(argc < 2 ){
   printf("请输入要处理的由ida生成的asm文件名 \n\n本程序由想得太美制作  \n2010年11月10日晚\n");
   return 1;
   }  
  
  
    fp=fopen(argv[1],"r");
    if(fp==NULL)
    {
        printf("不能打开输入文件\n");
        return 1 ;
    }
     

    fp1=fopen("out.asm","w");
    if(fp==NULL)
    {
        printf("不能打开输出文件\n");
        return 1 ;
    }
    fputs("__declspec(naked) int test(unsigned char * buff)\n",fp1);
    fputs("{\n",fp1);
    fputs(" __asm\n",fp1);
    fputs(" {\n",fp1);

    while(fgets(buffer,128,fp)!=NULL)
    {
        len=strlen(buffer);
        if(len<=1)continue ;
        if(!memcmp(buffer,";",1)) continue;        
        p1=strstr(buffer,"=");
        if(p1!=NULL)
        {
            if(head==NULL)
            {
                head=(struct aa*)malloc(sizeof(struct aa));
                memcpy(head->name,buffer,(p1-buffer));
                head->name[p1-buffer]='\0' ;
                p2=strrchr(buffer,' ');
                memcpy(head->val,p2+1,(buffer+len-p2));
                head->val[buffer+len-p2+1]='\0' ;
                head->next=NULL ;
                fprintf(fp1,"#define %s %s", head->name,head->val);
            }
            else
            {
                curren=head ;
                while(curren->next!=NULL)
                {
                    curren=curren->next ;
                    if (memcmp(curren->name,buffer,(p1-buffer)) == 0 && (p1-buffer)==strlen(curren->name)) goto xxv32;
                }
                zhi=(struct aa*)malloc(sizeof(struct aa));
                curren->next=zhi ;
                zhi->next=NULL ;
                memcpy(zhi->name,buffer,(p1-buffer));
                zhi->name[p1-buffer]='\0' ;
                p2=strrchr(buffer,' ');
                memcpy(zhi->val,p2+1,(buffer+len-p2));
                zhi->val[buffer+len-p2+1]='\0' ;
                fprintf(fp1,"#define %s %s", zhi->name,zhi->val);
            }
            xxv32:
              continue ;
        }
     }
    
     fputs("\n",fp1);
     fputs("\n",fp1);
    
     fseek(fp,0, SEEK_SET);
     while(fgets(buffer,128,fp)!=NULL)
    {
     len=strlen(buffer);
        if(len<=1)continue ;
        if(! memcmp(buffer,";",1)) continue; 
         
     if(strstr(buffer,"near")!=NULL)
        {
            p5=strstr(buffer,"proc");
            *(p5-1)=':' ;
            *p5='\0' ;
            strcpy(temp,buffer);
            biaoz+=1 ;
            if(biaoz>=2)
            {
                fputs(temp,fp1);
                fputs("\n",fp1);
            }
            continue ;
        }
      if(strstr(buffer,"endp")!=0)     continue ;
      if(strstr(buffer,"=")!=0)        continue ;
      if(strstr(buffer,"public")!=0)   continue ;
      fputs(buffer,fp1); 
    }
   
    fputs(" }\n",fp1);
    fputs("}\n",fp1);
      
    fclose(fp);
    fclose(fp1);
    printf("转换完成!\n");
    return 0 ;
}