def(void, OnRequestInfo, HTTP_RequestInfo info) {
Logger_Info(this->logger, $("Client requested '%'."), info.path);
String_Copy(&this->path, info.path);
this->method = info.method;
this->version = info.version;
}
void GetConfigFile(wchar_t* GameName, wchar_t* Buffer)
{
wchar_t *dot;
wchar_t batchFile[260];
String_Copy(batchFile, GAMES_CONFIG_PATH);
String_Concatenate(batchFile, GameName);
dot = String_FindLastChar(batchFile, '.');
if (dot)
String_Copy(dot, L".ini");
else
String_Concatenate(batchFile, L".ini");
String_Copy(Buffer, batchFile);
}
static void copyGetTokenResult(void *userData, const String *token,
FlickrPerm perm, const FlickrUser *user)
{
getTokenArg *p = (getTokenArg*)userData;
String_Copy(&p->token, token);
FlickrUser_Copy(&p->user,user);
p->perm=perm;
}
static UINT
GetOutputOption(s_CLONEPARMS *parm, UINT iArg, BOOL gotDstFn)
{
PSTR pszArg = Env_ParamStr(iArg);
if (gotDstFn) return ErrOptionSetTwice(pszVOPTOUTPUT,iArg-1);
if (!pszArg) return ArgError(RSTR(NOFILEN),iArg-1);
String_Copy(parm->dstfn, pszArg, 1024);
return (iArg+1);
}
static UINT
GetEnlargeOption(s_CLONEPARMS *parm, UINT iArg)
{
PSTR pszArg = Env_ParamStr(iArg);
if (parm->flags & PARM_FLAG_ENLARGE) return ErrOptionSetTwice(pszVOPTENLARGE,iArg-1);
if (!pszArg) return ArgError(RSTR(NODSIZE),iArg-1);
String_Copy(parm->szDestSize, pszArg, 32);
parm->flags |= PARM_FLAG_ENLARGE;
return (iArg+1);
}
VOID
PushAddToFileList( FILE_LIST* FileList, FILE_LIST_ENTRY *FileEntry )
{
FILE_LIST_ENTRY *fileListEntry;
WCHAR *name;
name = (WCHAR*)Memory_Allocate(
/*PushHeapHandle,
0,*/
(String_GetLength(FileEntry->Name) + 1) * sizeof(WCHAR)
);
String_Copy(name, FileEntry->Name);
if (*FileList == NULL)
{
FILE_LIST_ENTRY *fileList;
*FileList = (FILE_LIST)Memory_Allocate(
/*PushHeapHandle,
0,*/
sizeof(FILE_LIST_ENTRY)
);
fileList = *FileList;
fileList->NextEntry = NULL;
fileList->Bytes = FileEntry->Bytes;
fileList->Name = name;
fileList->Cache = FileEntry->Cache;
return;
}
fileListEntry = (FILE_LIST_ENTRY*) *FileList;
while (fileListEntry->NextEntry != 0)
{
fileListEntry = fileListEntry->NextEntry;
}
fileListEntry->NextEntry = (FILE_LIST_ENTRY *)Memory_Allocate(
/*PushHeapHandle,
0,*/
sizeof(FILE_LIST_ENTRY)
);
fileListEntry = fileListEntry->NextEntry;
fileListEntry->Bytes = FileEntry->Bytes;
fileListEntry->Name = name;
fileListEntry->Cache = FileEntry->Cache;
fileListEntry->NextEntry = 0;
}
void GetConfigFileFromProcessId(int ProcessId, wchar_t* Buffer)
{
wchar_t processName[260];
wchar_t fileName[260];
wchar_t *slash, *dot;
Process_GetFileNameByProcessId(ProcessId, processName);
String_Copy(fileName, GAMES_CONFIG_PATH);
dot = String_FindLastChar(processName, '.');
String_Copy(dot, L".ini");
slash = String_FindLastChar(processName, '\\');
String_Concatenate(fileName, slash + 1);
Log(fileName);
String_Copy(Buffer, fileName);
}
VOID Directory_AppendFileName(
WCHAR* FileName,
WCHAR* Path,
WCHAR* Buffer
)
{
String_Copy(Buffer, Path);
String_Concatenate(Buffer, L"\\");
String_Concatenate(Buffer, FileName);
}
VOID Log(const wchar_t* Format, ...)
{
wchar_t buffer[260];
wchar_t output[260];
va_list _Arglist;
String_Copy(output, L"[PUSH] ");
va_start(_Arglist, Format);
vswprintf_s(buffer, 260, Format, _Arglist);
va_end(_Arglist);
wcsncat(output, buffer, 260);
OutputDebugStringW(output);
}
static sdef(String, Sanitize, String name) {
String out = String_Clone(name);
String_ReplaceAll(&out, $("?"), $(" "));
String_ReplaceAll(&out, $(":"), $(" "));
String_ReplaceAll(&out, $("/"), $("-"));
String_ReplaceAll(&out, $(" "), $(" "));
String_Copy(&out, String_Trim(out));
if (String_EndsWith(out, $("."))) {
out.len--;
}
return out;
}
void test_String_Copy ( )
{
STR80 strTest1, strTest2;
cout << "Testing:" << endl;
cout << "void String_Copy"<<endl;
GetTestString("....Enter a string 1:",strTest1);
GetTestString("....Enter a string 2:",strTest2);
cout <<"....String 1 before execution:"<<strTest1<<endl;
cout <<"....String 2 before execution:"<<strTest2<<endl;
cout <<endl;
String_Copy (strTest1, strTest2);
cout <<endl;
cout <<"....String 1 after execution:"<<strTest1<<endl;
cout <<"....String 2 after execution:"<<strTest2<<endl;
pause_display();
}
VOID Cache( PUSH_GAME* Game )
{
CHAR mountPoint = 0;
UINT64 bytes = 0, availableMemory = 0; // in bytes
SYSTEM_BASIC_PERFORMANCE_INFORMATION performanceInfo;
//BfBatchFile batchFile(Game);
// Check if game is already cached so we donot have wait through another
if (String_Compare(g_szPrevGame, Game->InstallPath) == 0 && !g_bRecache)
return;
g_bRecache = FALSE;
if (!FolderExists(Game->InstallPath))
{
Log(L"Folder not exist!");
}
// Check available memory
NtQuerySystemInformation(
SystemBasicPerformanceInformation,
&performanceInfo,
sizeof(SYSTEM_BASIC_PERFORMANCE_INFORMATION),
NULL
);
availableMemory = performanceInfo.AvailablePages * HwInfoSystemBasicInformation.PageSize;
// Read batch file
PushFileList = 0;
BatchFile_Initialize(Game);
bytes = BatchFile_GetBatchSize();
PushFileList = BatchFile_GetBatchList();
// Check if any files at all are marked for cache, if not return.
if (PushFileList == NULL)
// List is empty hence no files to cache, return.
return;
// Add 200MB padding for disk format;
bytes += 209715200;
if (bytes > availableMemory)
{
MessageBoxW(
NULL,
L"There isn't enough memory to hold all the files you marked for caching.\n"
L"The Ramdisk will be set to an acceptable size and filled with what it can hold.\n"
L"Please upgrade RAM.",
L"Push",
MB_TOPMOST
);
}
RemoveRamDisk();
mountPoint = FindFreeDriveLetter();
CreateRamDisk(bytes, mountPoint);
FormatRamDisk();
CacheFiles(mountPoint);
// Release batchfile list
PushFileList = NULL;
String_Copy(g_szPrevGame, Game->InstallPath);
}
INT32 __stdcall start( )
{
HANDLE sectionHandle, *hMutex;
HANDLE eventHandle;
HANDLE threadHandle;
DWORD sectionSize;
MSG messages;
OBJECT_ATTRIBUTES objAttrib = {0};
PTEB threadEnvironmentBlock;
UNICODE_STRING eventSource;
LDR_DATA_TABLE_ENTRY *module;
SECTION_BASIC_INFORMATION sectionInfo;
LARGE_INTEGER newSectionSize;
InitializeCRT();
threadEnvironmentBlock = NtCurrentTeb();
PushProcessId = threadEnvironmentBlock->ClientId.UniqueProcess;
PushHeapHandle = threadEnvironmentBlock->ProcessEnvironmentBlock->ProcessHeap;
PushSessionId = threadEnvironmentBlock->ProcessEnvironmentBlock->SessionId;
// Check if already running
hMutex = CreateMutexW(0, FALSE, L"PushOneInstance");
if (threadEnvironmentBlock->LastErrorValue == ERROR_ALREADY_EXISTS
|| threadEnvironmentBlock->LastErrorValue == ERROR_ACCESS_DENIED)
{
MessageBoxW(0, L"Only one instance!", 0,0);
ExitProcess(0);
}
//create image event
eventHandle = NULL;
UnicodeString_Init(&eventSource, L"Global\\" PUSH_IMAGE_EVENT_NAME);
objAttrib.Length = sizeof(OBJECT_ATTRIBUTES);
objAttrib.RootDirectory = BaseGetNamedObjectDirectory();
objAttrib.ObjectName = &eventSource;
objAttrib.Attributes = OBJ_OPENIF;
objAttrib.SecurityDescriptor = NULL;
objAttrib.SecurityQualityOfService = NULL;
NtCreateEvent(&eventHandle, EVENT_ALL_ACCESS, &objAttrib, NotificationEvent, FALSE);
// populate file name and path
module = (LDR_DATA_TABLE_ENTRY*)threadEnvironmentBlock->ProcessEnvironmentBlock->Ldr->InLoadOrderModuleList.Flink;
Memory_Copy(PushFilePath, module->FullDllName.Buffer, module->FullDllName.Length);
PushFilePath[module->FullDllName.Length] = L'\0';
// Start Driver.
Driver_Extract();
PushDriverLoaded = Driver_Load();
//initialize instance
PushInstance = Module_GetHandle(L"Push.exe");
// Create interface
MwCreateMainWindow();
// Create section.
sectionSize = sizeof(PUSH_SHARED_MEMORY) + OSD_GetSize();
PushSharedMemory = (PUSH_SHARED_MEMORY*)Memory_MapViewOfSection(PUSH_SECTION_NAME, sectionSize, §ionHandle);
if (!PushSharedMemory)
{
Log(L"Could not create shared memory");
return 0;
}
Log(L"Created section of size %i bytes", sectionSize);
//zero struct
Memory_Clear(PushSharedMemory, sizeof(PUSH_SHARED_MEMORY));
//initialize window handle used by overlay
//PushSharedMemory->WindowHandle = PushMainWindow->Handle;
//initialize default font properties for overlay
String_Copy(PushSharedMemory->FontName, L"Verdana");
PushSharedMemory->FontBold = TRUE;
if (File_Exists(PUSH_SETTINGS_FILE))
{
wchar_t *buffer;
wchar_t marker;
// Check if file is UTF-16LE.
buffer = (WCHAR*) File_Load(PUSH_SETTINGS_FILE, NULL);
marker = buffer[0];
Memory_Free(buffer);
if (marker == 0xFEFF)
//is UTF-LE.
{
// Init settings from ini file.
buffer = Memory_Allocate(100 * sizeof(WCHAR));
Ini_GetString(L"Settings", L"FrameLimit", NULL, buffer, 5, L".\\" PUSH_SETTINGS_FILE);
PushSharedMemory->FrameLimit = _wtoi(buffer);
if (Ini_ReadBoolean(L"Settings", L"ThreadOptimization", FALSE, L".\\" PUSH_SETTINGS_FILE))
PushSharedMemory->ThreadOptimization = TRUE;
if (Ini_ReadBoolean(L"Settings", L"KeepFps", FALSE, L".\\" PUSH_SETTINGS_FILE))
PushSharedMemory->KeepFps = TRUE;
Ini_GetString(L"Settings", L"OverlayInterface", NULL, buffer, 5, L".\\" PUSH_SETTINGS_FILE);
if (String_Compare(buffer, L"PURE") == 0)
PushOverlayInterface = OVERLAY_INTERFACE_PURE;
else if (String_Compare(buffer, L"RTSS") == 0)
PushOverlayInterface = OVERLAY_INTERFACE_RTSS;
Ini_GetString(L"Settings", L"KeyboardHookType", L"AUTO", buffer, 10, L".\\" PUSH_SETTINGS_FILE);
if (String_Compare(buffer, L"AUTO") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_AUTO;
}
else if (String_Compare(buffer, L"SUBCLASS") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_SUBCLASS;
}
else if (String_Compare(buffer, L"MESSAGE") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_MESSAGE;
}
else if (String_Compare(buffer, L"KEYBOARD") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_KEYBOARD;
}
else if (String_Compare(buffer, L"DETOURS") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_DETOURS;
}
else if (String_Compare(buffer, L"RAW") == 0)
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_RAW;
}
else
{
PushSharedMemory->KeyboardHookType = KEYBOARD_HOOK_AUTO;
}
Ini_GetString(L"Settings", L"EngineClockMax", NULL, buffer, 5, L".\\" PUSH_SETTINGS_FILE);
PushSharedMemory->HarwareInformation.DisplayDevice.EngineOverclock = _wtoi(buffer);
Ini_GetString(L"Settings", L"MemoryClockMax", NULL, buffer, 5, L".\\" PUSH_SETTINGS_FILE);
PushSharedMemory->HarwareInformation.DisplayDevice.MemoryOverclock = _wtoi(buffer);
Ini_GetString(L"Settings", L"ControllerTimeout", NULL, buffer, 5, L".\\" PUSH_SETTINGS_FILE);
PushSharedMemory->ControllerTimeout = _wtoi(buffer);
Ini_GetString(L"Settings", L"FontName", L"Verdana", buffer, 100, L".\\" PUSH_SETTINGS_FILE);
String_Copy(PushSharedMemory->FontName, buffer);
Memory_Free(buffer);
if (Ini_ReadBoolean(L"Settings", L"FontBold", FALSE, L".\\" PUSH_SETTINGS_FILE))
PushSharedMemory->FontBold = TRUE;
}
else
{
MessageBoxW(
NULL,
L"Settings file not UTF-16LE! "
L"Resave the file as \"Unicode\" or Push won't read it!",
L"Bad Settings file",
NULL
);
}
}
if (!PushDriverLoaded)
{
wchar_t driverPath[260];
Resource_Extract(L"DRIVERALT", L"WinRing0x64.sys");
GetDriverPath(L"WinRing0x64.sys", driverPath);
Wr0DriverLoaded = Wr0Initialize(driverPath);
}
//initialize HWInfo
GetHardwareInfo();
//initialize OSD items
NtQuerySection(
sectionHandle,
SectionBasicInformation,
§ionInfo,
sizeof(SECTION_BASIC_INFORMATION),
NULL
);
newSectionSize.QuadPart = OSD_Initialize() + sizeof(PUSH_SHARED_MEMORY);
if (newSectionSize.QuadPart > sectionInfo.MaximumSize.QuadPart)
{
Log(L"Shared memory too small!");
}
//Check for controllers/gamepads/bluetooth adapters
//EnumerateDevices();
// Check for running games
Process_EnumProcesses(ProcessEnum);
// Activate process monitoring
if (PushDriverLoaded)
{
PushToggleProcessMonitoring(TRUE);
}
else
{
HANDLE overlayLib = NULL;
void* prcAddress = 0;
Resource_Extract(L"OVERLAY32", PUSH_LIB_NAME_32);
overlayLib = Module_Load(L"overlay32.dll");
prcAddress = Module_GetProcedureAddress(overlayLib, "InstallOverlayHook");
if (prcAddress)
{
InstallOverlayHook = (TYPE_InstallOverlayHook)prcAddress;
InstallOverlayHook();
}
}
g_szPrevGame[5] = '\0';
NtCreateThreadEx(
&PushMonitorThreadHandle,
THREAD_ALL_ACCESS,
NULL,
NtCurrentProcess(),
&MonitorThread,
NULL,
NoThreadFlags,
0, 0, 0,
NULL
);
NtCreateThreadEx(
&threadHandle,
THREAD_ALL_ACCESS,
NULL,
NtCurrentProcess(),
&PipeThread,
NULL,
NoThreadFlags,
0, 0, 0,
NULL
);
// Handle messages
while(GetMessageW(&messages, 0,0,0))
{
TranslateMessage(&messages);
DispatchMessageW(&messages);
}
ExitProcess(0);
return 0;
}
static def(void, OnHeader, String name, String value) {
if (String_Equals(name, $("Location"))) {
String_Copy(&this->location, value);
}
}
void FlickrUser_Copy(FlickrUser *dest, const FlickrUser *src) {
String_Copy(&dest->m_nsid,&src->m_nsid);
String_Copy(&dest->m_username,&src->m_username);
String_Copy(&dest->m_fullname,&src->m_fullname);
}
VOID CacheFile( WCHAR *FileName, CHAR cMountPoint )
{
WCHAR destination[260];
WCHAR *pszFileName;
CHAR bMarkedForCaching = FALSE;
WCHAR* slash;
WCHAR deviceName[260], dosName[260];
HANDLE directoryHandle;
HANDLE linkHandle;
OBJECT_ATTRIBUTES objAttrib;
UNICODE_STRING directoryName;
UNICODE_STRING driveLetter;
UNICODE_STRING linkTarget;
destination[0] = cMountPoint;
destination[1] = ':';
destination[2] = '\\';
destination[3] = '\0';
pszFileName = String_FindLastChar(FileName, '\\') + 1;
if (!bMarkedForCaching)
// file was a member of a folder marked for caching
{
String_Concatenate(destination, g_szLastDir);
String_Concatenate(destination, L"\\");
}
String_Concatenate(destination, pszFileName);
File_Copy(FileName, destination, CopyProgress);
String_Copy(dosName, FileName);
slash = String_FindFirstChar(dosName, '\\');
*slash = L'\0';
UnicodeString_Init(&directoryName, L"\\??");
objAttrib.Length = sizeof(OBJECT_ATTRIBUTES);
objAttrib.RootDirectory = NULL;
objAttrib.ObjectName = &directoryName;
objAttrib.Attributes = OBJ_CASE_INSENSITIVE;
objAttrib.SecurityDescriptor = NULL;
objAttrib.SecurityQualityOfService = NULL;
NtOpenDirectoryObject(&directoryHandle, DIRECTORY_QUERY, &objAttrib);
UnicodeString_Init(&driveLetter, dosName);
objAttrib.Length = sizeof(OBJECT_ATTRIBUTES);
objAttrib.RootDirectory = directoryHandle;
objAttrib.ObjectName = &driveLetter;
objAttrib.Attributes = OBJ_CASE_INSENSITIVE;
objAttrib.SecurityDescriptor = NULL;
objAttrib.SecurityQualityOfService = NULL;
NtOpenSymbolicLinkObject(&linkHandle, SYMBOLIC_LINK_QUERY, &objAttrib);
linkTarget.Length = 0;
linkTarget.MaximumLength = 260 * sizeof(WCHAR);
linkTarget.Buffer = deviceName;
NtQuerySymbolicLinkObject(linkHandle, &linkTarget, NULL);
deviceName[linkTarget.Length / sizeof(WCHAR)] = L'\0';
String_Concatenate(deviceName, L"\\");
String_Concatenate(deviceName, slash + 1);
R0QueueFile(
deviceName,
String_GetLength(deviceName) + 1
);
}
PUBLIC BOOL
CmdLine_Parse(s_CLONEPARMS *parm)
{
BOOL gotSrcFn=FALSE,gotDstFn=FALSE;
PSTR pszArg;
UINT argc = Env_ParamCount();
UINT iArg = 1;
ZeroMemory(parm,sizeof(s_CLONEPARMS));
parm->flags = PARM_FLAG_CLIMODE;
while (iArg <= argc) {
pszArg = Env_ParamStr(iArg);
if (!pszArg) break; // can't happen.
iArg++;
if (pszArg[0]=='-') { // if option
if (pszArg[1]=='-') {
// verbose options
char szItem[32];
pszArg+=2;
for (;;) {
pszArg = VerboseSubOption(szItem,32,pszArg);
if (szItem[0]==0) break;
if (String_Compare(szItem,pszVOPTHELP)==0) {
return Usage(TRUE);
} else if (String_Compare(szItem,pszVOPTOUTPUT)==0) {
iArg = GetOutputOption(parm,iArg,gotDstFn);
if (iArg==0) return FALSE;
gotDstFn = TRUE;
} else if (String_Compare(szItem,pszVOPTKEEPUUID)==0) {
if (!GetOption(parm,iArg,PARM_FLAG_KEEPUUID,pszVOPTKEEPUUID)) return FALSE;
} else if (String_Compare(szItem,pszVOPTCOMPACT)==0) {
if (!GetOption(parm,iArg,PARM_FLAG_COMPACT,pszVOPTCOMPACT)) return FALSE;
} else if (String_Compare(szItem,pszVOPTNOMERGE)==0) {
if (!GetOption(parm,iArg,PARM_FLAG_NOMERGE,pszVOPTNOMERGE)) return FALSE;
} else if (String_Compare(szItem,pszVOPTREPART)==0) {
if (!GetOption(parm,iArg,PARM_FLAG_REPART,pszVOPTREPART)) return FALSE;
} else if (String_Compare(szItem,pszVOPTENLARGE)==0) {
iArg = GetEnlargeOption(parm,iArg);
if (iArg==0) return FALSE;
} else {
return ArgError(RSTR(UNKOPT),iArg-1);
}
}
} else {
// single char options.
CHAR c;
pszArg++;
c = *pszArg++;
if (!c) return ArgError(RSTR(INVOPT),iArg-1);
while (c) {
if (c==pszCHAROPT[4]) { // 'h'
return Usage(TRUE);
} else if (c==pszCHAROPT[0]) { // 'o'
iArg = GetOutputOption(parm,iArg,gotDstFn);
if (iArg==0) return FALSE;
gotDstFn = TRUE;
} else if (c==pszCHAROPT[1]) { // 'k'
if (!GetOption(parm,iArg,PARM_FLAG_KEEPUUID,pszVOPTKEEPUUID)) return FALSE;
} else if (c==pszCHAROPT[3]) { // 'c'
if (!GetOption(parm,iArg,PARM_FLAG_COMPACT,pszVOPTCOMPACT)) return FALSE;
} else if (c==pszCHAROPT[5]) { // 'r'
if (!GetOption(parm,iArg,PARM_FLAG_REPART,pszVOPTREPART)) return FALSE;
} else if (c==pszCHAROPT[2]) { // 'e'
iArg = GetEnlargeOption(parm,iArg);
if (iArg==0) return FALSE;
} else {
return ArgError(RSTR(UNKOPT),iArg-1);
}
c = *pszArg++;
}
}
} else if (gotSrcFn) {
return ArgError(RSTR(SRCTWICE),iArg-1);
} else {
GetFullPathName(pszArg,1024,parm->srcfn,0);
gotSrcFn = TRUE;
}
}
if (!gotSrcFn) {
return ArgError(RSTR(NEEDSRC),iArg-1);
}
if (!gotDstFn) {
Env_GenerateCloneName(parm->dstfn,parm->srcfn);
}
GetFullPathName(parm->dstfn,1024,tmpfn,0);
String_Copy(parm->dstfn, tmpfn, 1024);
if (!(parm->flags & PARM_FLAG_ENLARGE)) parm->flags &= ~PARM_FLAG_REPART;
return TRUE;
}
int main(int ArgN, char** Arg)
{
int RegenFlag = 0;
char* CWaveFile = NULL;
char* CLabelFile = NULL;
char* CRecFile = NULL;
char* CRegenFile = "RegeneratedTrack.txt";
int c;
while((c = getopt(ArgN, Arg, "r")) != -1)
{
switch(c)
{
case 'r':
RegenFlag = 1;
break;
case '?':
printf("Usage: rsegment [-r] "
"[-h] wavfile labelfile recfile\n");
return 1;
default:
abort();
}
}
int i;
for(i = optind; i < ArgN; i ++)
{
switch(i - optind)
{
case 0:
CWaveFile = Arg[i];
break;
case 1:
CLabelFile = Arg[i];
break;
case 2:
CRecFile = Arg[i];
break;
default:
fprintf(stderr, "Warning: redundant argument '%s'.\n", Arg[i]);
}
}
if(! CWaveFile)
{
fprintf(stderr, "Missing argument 'wavfile'.\n");
return 1;
}
if(! CLabelFile)
{
fprintf(stderr, "Missing argument 'labelfile'.\n");
return 1;
}
if(! CRecFile)
{
fprintf(stderr, "Missing argument 'recfile'.\n");
return 1;
}
Wave SorcWave;
File LabelFile;
File RecFile;
File RegenFile;
String WavePath, LabelPath, RecPath, RegenPath;
RNew(String, & WavePath, & LabelPath, & RecPath, & RegenPath);
RNew(File, & LabelFile, & RecFile, & RegenFile);
RCall(Wave, Ctor)(& SorcWave);
String_SetChars(& WavePath, CWaveFile);
String_SetChars(& LabelPath, CLabelFile);
String_SetChars(& RecPath, CRecFile);
String_SetChars(& RegenPath, CRegenFile);
if(! RCall(Wave, FromFile)(& SorcWave, & WavePath))
{
fprintf(stderr, "Cannot load wave file.\n");
return 1;
}
if(! File_Open(& LabelFile, & LabelPath, READONLY))
{
fprintf(stderr, "Cannot load label file.\n");
return 1;
}
if(! File_Open(& RecFile, & RecPath, READONLY))
{
fprintf(stderr, "Cannot load word table.\n");
return 1;
}
if(RegenFlag)
{
if(! File_Open(& RegenFile, & RegenPath, CREATE))
{
fprintf(stderr, "Cannot create '%s'.\n", CRegenFile);
return 1;
}
}
Array_Ctor(String, RecList);
Array_Ctor(int, SegList);
ParseRecFile(& RecFile);
ParseLabelFile(& LabelFile, SorcWave.SampleRate);
Wave SegWave;
RCall(Wave, Ctor)(& SegWave);
SegWave.SampleRate = SorcWave.SampleRate;
String SegName, LineBuf, TempBuf;
RNew(String, & SegName, & LineBuf, & TempBuf);
for(i = 0; i <= SegList_Index; i ++)
{
int Start = SegList[i];
int End;
if(i == SegList_Index)
End = SorcWave.Size;
else
End = SegList[i + 1];
if(i > RecList_Index)
{
fprintf(stderr, "Insufficient records in word table.\n");
fprintf(stderr, "Segmentation aborted at %fs.\n",
(float)Start / SorcWave.SampleRate);
return 1;
}
String* Name = & RecList[i];
String_Copy(& SegName, Name);
String_JoinChars(& SegName, ".wav");
if(! RegenFlag)
{
int Size = End - Start;
Real* Data = RCall(RAlloc, Real)(Size);
RCall(Wave, Read)(& SorcWave, Data, Start, Size);
RCall(Wave, Resize)(& SegWave, Size);
RCall(Wave, Write)(& SegWave, Data, 0, Size);
if(! RCall(Wave, ToFile)(& SegWave, & SegName))
fprintf(stderr, "Exporting '%s' failed. Skipped.\n",
String_GetChars(& SegName));
RFree(Data);
}
if(RegenFlag)
{
CStrFloat(& LineBuf, (float)SegList[i] / SorcWave.SampleRate);
String_Copy(& TempBuf, & LineBuf);
String_JoinChars(& LineBuf, "\t");
String_Join(& LineBuf, & TempBuf);
String_JoinChars(& LineBuf, "\t");
String_Join(& LineBuf, Name);
File_WriteLine(& RegenFile, & LineBuf);
}
}
if(i <= RecList_Index)
fprintf(stderr, "Warning: redundant records in word table starting from"
" '%s'.\n", String_GetChars(& RecList[i]));
if(RegenFlag)
File_Flush(& RegenFile);
File_Close(& RecFile);
File_Close(& LabelFile);
File_Close(& RegenFile);
Array_ObjDtor(String, RecList);
Array_Dtor(String, RecList);
Array_Dtor(int, SegList);
RDelete(& WavePath, & LabelPath, & RecPath, & RegenPath, & SegName,
& LineBuf, & TempBuf);
RDelete(& SorcWave, & LabelFile, & RecFile, & SegWave, & RegenFile);
return 0;
}
void InstallDriver( HANDLE DeviceInformation, SP_DEVINFO_DATA* DeviceData )
{
SP_DEVINSTALL_PARAMS_W deviceInstallParams;
SP_DRVINFO_DATA_V2_W driverData;
UINT32 reboot = 0;
HANDLE module;
DWORD address;
BYTE *addr;
if (!SetupDiSetSelectedDevice(DeviceInformation, DeviceData))
{
Log(L"Set Selected Device fail.");
return;
}
Memory_Clear(&deviceInstallParams, sizeof(SP_DEVINSTALL_PARAMS_W));
if (sizeof(int*) == 8)
{
deviceInstallParams.cbSize = 584;
}
else
{
deviceInstallParams.cbSize = sizeof(SP_DEVINSTALL_PARAMS_W);
}
if (!SetupDiGetDeviceInstallParamsW(DeviceInformation, DeviceData, &deviceInstallParams))
{
Log(L"Get Device Install Params fail.");
return;
}
String_Copy(deviceInstallParams.DriverPath, L"C:\\mij\\drivers\\MijXinput.inf");
deviceInstallParams.Flags |= 65536;
if (!SetupDiSetDeviceInstallParamsW(DeviceInformation, DeviceData, &deviceInstallParams))
{
Log(L"Set Device Install Params fail.");
return;
}
if (!SetupDiBuildDriverInfoList(DeviceInformation, DeviceData, SPDIT_COMPATDRIVER))
{
Log(L"Building Driver Info List fail.");
return;
}
if (!SetupDiSelectBestCompatDrv(DeviceInformation, DeviceData))
{
Log(L"Select Best Compatible Driver fail.");
return;
}
Memory_Clear(&driverData, sizeof(SP_DRVINFO_DATA_V2_W));
driverData.cbSize = sizeof(SP_DRVINFO_DATA_V2_W);
if (!SetupDiGetSelectedDriverW(DeviceInformation, DeviceData, &driverData))
{
Log(L"Get MotioninJoy Driver fail.");
return;
}
//DiInstallDevice willfully returns 0xE0000235 (ERROR_IN_WOW64) in a WOW64 environment.
//I don't know if this was a security restraint but there is no reason why this function
//should not work under WOW64. All we have to do is insert one, literally one jmp patch to
//skip the WOW64 check and the function succeeds as normal.
module = Module_GetHandle(L"newdev.dll");
address = Module_GetProcedureAddress(module, "DiInstallDevice");
address += 0x134;
addr = address;
if ((*addr) == 0x74) //je
{
DWORD oldPageProtection = 0;
//We firstly have to remove page protection of course.
VirtualProtect(addr, 1, PAGE_EXECUTE_READWRITE, &oldPageProtection);
//patch to jne
*addr = 0x75;
//Lastly, make it look like we were never even there by restoring protection
VirtualProtect(addr, 1, oldPageProtection, &oldPageProtection);
}
if (!DiInstallDevice(NULL, DeviceInformation, DeviceData, &driverData, 0, &reboot))
{
Log(L"Install MotioninJoy Driver fail.");
}
}
