#include <stdio.h>
#include <conio.h>
#include <ctype.h>

int main ( void )
{
   char regowner[80];
   char pid[80];
   char pns[80];
   char conf;
   FILE *create_reg;
   FILE *append_reg;
   
   printf("Yahaa VB Worm Removal Tool\n\n");
   do {
       printf("Please enter the Registered Owner\n(e.g. Amat bin Dollah)\n>");
       fscanf (stdin, " %79[^\n]", &regowner);
       printf("\nPlease enter the Processor Name\n(e.g. Intel(R) Pentium(R) 4 CPU 2.80GHz)\n>");
       fscanf (stdin, " %79[^\n]", &pns);
       printf("\nPlease enter the Product ID\n(e.g. 12345-123-1234567-12345)\n>");
       fscanf (stdin, " %79[^\n]", &pid);
       printf("\n\n");
       printf("You entered:\nRegistered Owner: %s\nProcessor Name: %s\nProduct ID: %s\n\nAre these values correct? (y=yes, n=no)",regowner,pns,pid);
       scanf(" %c", &conf);
   }
   while ( conf != 'y');

   create_reg= fopen("append.reg", "w");
   fprintf(create_reg, "Windows Registry Editor Version 5.00\n\n");
   fclose(create_reg);
   
   append_reg= fopen("append.reg", "a");
   fprintf(append_reg, "[HKEY_LOCAL_MACHINE\\Software\\Microsoft\\Windows\\CurrentVersion\\Run]\n\"autoupdate\"=\"\"\n\n");
   fprintf(append_reg, "[HKEY_CURRENT_USER\\Software\\Microsoft\\Internet Explorer\\Main]\n\"Window Title\"=\"\"\n\n");
   fprintf(append_reg, "[HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion]\n\"RegisteredOwner\"=\"%s\"\n\n", regowner);
   fprintf(append_reg, "[HKEY_LOCAL_MACHINE\\HARDWARE\\DESCRIPTION\\System\\CentralProcessor\\0]\n\"ProcessorNameString\"=\"%s\"\n\n", pns);
   fprintf(append_reg, "[HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion]\n\"ProductId\"=\"%s\"\n\n", pid);
   fprintf(append_reg, "[HKEY_CLASSES_ROOT\\vbsfile]\n\"DefaultIcon\"=\"C:\\Windows\\System32\\WScript.exe,2 vbsfile\"");
   fclose(append_reg);

   printf("\n\nRegistration Keys written. Press any key to continue\n\n");
   
   getch();
   return 0;
}