diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..57d5700 --- /dev/null +++ b/.gitignore @@ -0,0 +1,288 @@ +# ---> VisualStudio +## Ignore Visual Studio temporary files, build results, and +## files generated by popular Visual Studio add-ons. +## +## Get latest from https://github.com/github/gitignore/blob/master/VisualStudio.gitignore + +# User-specific files +*.suo +*.user +*.userosscache +*.sln.docstates + +# User-specific files (MonoDevelop/Xamarin Studio) +*.userprefs + +# Build results +[Dd]ebug/ +[Dd]ebugPublic/ +[Rr]elease/ +[Rr]eleases/ +x64/ +x86/ +bld/ +[Bb]in/ +[Oo]bj/ +[Ll]og/ + +# Visual Studio 2015 cache/options directory +.vs/ +# Uncomment if you have tasks that create the project's static files in wwwroot +#wwwroot/ + +# MSTest test Results +[Tt]est[Rr]esult*/ +[Bb]uild[Ll]og.* + +# NUNIT +*.VisualState.xml +TestResult.xml + +# Build Results of an ATL Project +[Dd]ebugPS/ +[Rr]eleasePS/ +dlldata.c + +# .NET Core +project.lock.json +project.fragment.lock.json +artifacts/ +**/Properties/launchSettings.json + +*_i.c +*_p.c +*_i.h +*.ilk +*.meta +*.obj +*.pch +*.pdb +*.pgc +*.pgd +*.rsp +*.sbr +*.tlb +*.tli +*.tlh +*.tmp +*.tmp_proj +*.log +*.vspscc +*.vssscc +.builds +*.pidb +*.svclog +*.scc + +# Chutzpah Test files +_Chutzpah* + +# Visual C++ cache files +ipch/ +*.aps +*.ncb +*.opendb +*.opensdf +*.sdf +*.cachefile +*.VC.db +*.VC.VC.opendb + +# Visual Studio profiler +*.psess +*.vsp +*.vspx +*.sap + +# TFS 2012 Local Workspace +$tf/ + +# Guidance Automation Toolkit +*.gpState + +# ReSharper is a .NET coding add-in +_ReSharper*/ +*.[Rr]e[Ss]harper +*.DotSettings.user + +# JustCode is a .NET coding add-in +.JustCode + +# TeamCity is a build add-in +_TeamCity* + +# DotCover is a Code Coverage Tool +*.dotCover + +# Visual Studio code coverage results +*.coverage +*.coveragexml + +# NCrunch +_NCrunch_* +.*crunch*.local.xml +nCrunchTemp_* + +# MightyMoose +*.mm.* +AutoTest.Net/ + +# Web workbench (sass) +.sass-cache/ + +# Installshield output folder +[Ee]xpress/ + +# DocProject is a documentation generator add-in +DocProject/buildhelp/ +DocProject/Help/*.HxT +DocProject/Help/*.HxC +DocProject/Help/*.hhc +DocProject/Help/*.hhk +DocProject/Help/*.hhp +DocProject/Help/Html2 +DocProject/Help/html + +# Click-Once directory +publish/ + +# Publish Web Output +*.[Pp]ublish.xml +*.azurePubxml +# TODO: Comment the next line if you want to checkin your web deploy settings +# but database connection strings (with potential passwords) will be unencrypted +*.pubxml +*.publishproj + +# Microsoft Azure Web App publish settings. Comment the next line if you want to +# checkin your Azure Web App publish settings, but sensitive information contained +# in these scripts will be unencrypted +PublishScripts/ + +# NuGet Packages +*.nupkg +# The packages folder can be ignored because of Package Restore +**/packages/* +# except build/, which is used as an MSBuild target. +!**/packages/build/ +# Uncomment if necessary however generally it will be regenerated when needed +#!**/packages/repositories.config +# NuGet v3's project.json files produces more ignorable files +*.nuget.props +*.nuget.targets + +# Microsoft Azure Build Output +csx/ +*.build.csdef + +# Microsoft Azure Emulator +ecf/ +rcf/ + +# Windows Store app package directories and files +AppPackages/ +BundleArtifacts/ +Package.StoreAssociation.xml +_pkginfo.txt + +# Visual Studio cache files +# files ending in .cache can be ignored +*.[Cc]ache +# but keep track of directories ending in .cache +!*.[Cc]ache/ + +# Others +ClientBin/ +~$* +*~ +*.dbmdl +*.dbproj.schemaview +*.jfm +*.pfx +*.publishsettings +orleans.codegen.cs + +# Since there are multiple workflows, uncomment next line to ignore bower_components +# (https://github.com/github/gitignore/pull/1529#issuecomment-104372622) +#bower_components/ + +# RIA/Silverlight projects +Generated_Code/ + +# Backup & report files from converting an old project file +# to a newer Visual Studio version. Backup files are not needed, +# because we have git ;-) +_UpgradeReport_Files/ +Backup*/ +UpgradeLog*.XML +UpgradeLog*.htm + +# SQL Server files +*.mdf +*.ldf + +# Business Intelligence projects +*.rdl.data +*.bim.layout +*.bim_*.settings + +# Microsoft Fakes +FakesAssemblies/ + +# GhostDoc plugin setting file +*.GhostDoc.xml + +# Node.js Tools for Visual Studio +.ntvs_analysis.dat +node_modules/ + +# Typescript v1 declaration files +typings/ + +# Visual Studio 6 build log +*.plg + +# Visual Studio 6 workspace options file +*.opt + +# Visual Studio 6 auto-generated workspace file (contains which files were open etc.) +*.vbw + +# Visual Studio LightSwitch build output +**/*.HTMLClient/GeneratedArtifacts +**/*.DesktopClient/GeneratedArtifacts +**/*.DesktopClient/ModelManifest.xml +**/*.Server/GeneratedArtifacts +**/*.Server/ModelManifest.xml +_Pvt_Extensions + +# Paket dependency manager +.paket/paket.exe +paket-files/ + +# FAKE - F# Make +.fake/ + +# JetBrains Rider +.idea/ +*.sln.iml + +# CodeRush +.cr/ + +# Python Tools for Visual Studio (PTVS) +__pycache__/ +*.pyc + +# Cake - Uncomment if you are using it +# tools/** +# !tools/packages.config + +# Telerik's JustMock configuration file +*.jmconfig + +# BizTalk build output +*.btp.cs +*.btm.cs +*.odx.cs +*.xsd.cs diff --git a/RebootApplication.sln b/RebootApplication.sln new file mode 100644 index 0000000..8f9e440 --- /dev/null +++ b/RebootApplication.sln @@ -0,0 +1,31 @@ + +Microsoft Visual Studio Solution File, Format Version 12.00 +# Visual Studio Version 17 +VisualStudioVersion = 17.2.32616.157 +MinimumVisualStudioVersion = 10.0.40219.1 +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "RebootApplication", "RebootApplication\RebootApplication.vcxproj", "{12A91232-55D1-4DC4-A7D2-5145147F2DF5}" +EndProject +Global + GlobalSection(SolutionConfigurationPlatforms) = preSolution + Debug|x64 = Debug|x64 + Debug|x86 = Debug|x86 + Release|x64 = Release|x64 + Release|x86 = Release|x86 + EndGlobalSection + GlobalSection(ProjectConfigurationPlatforms) = postSolution + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Debug|x64.ActiveCfg = Debug|x64 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Debug|x64.Build.0 = Debug|x64 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Debug|x86.ActiveCfg = Debug|Win32 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Debug|x86.Build.0 = Debug|Win32 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Release|x64.ActiveCfg = Release|x64 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Release|x64.Build.0 = Release|x64 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Release|x86.ActiveCfg = Release|Win32 + {12A91232-55D1-4DC4-A7D2-5145147F2DF5}.Release|x86.Build.0 = Release|Win32 + EndGlobalSection + GlobalSection(SolutionProperties) = preSolution + HideSolutionNode = FALSE + EndGlobalSection + GlobalSection(ExtensibilityGlobals) = postSolution + SolutionGuid = {0DA88194-C418-434B-B299-6EEDA22CE658} + EndGlobalSection +EndGlobal diff --git a/RebootApplication/RebootApplication.vcxproj b/RebootApplication/RebootApplication.vcxproj new file mode 100644 index 0000000..f5236fe --- /dev/null +++ b/RebootApplication/RebootApplication.vcxproj @@ -0,0 +1,135 @@ + + + + + Debug + Win32 + + + Release + Win32 + + + Debug + x64 + + + Release + x64 + + + + 16.0 + Win32Proj + {12a91232-55d1-4dc4-a7d2-5145147f2df5} + RebootApplication + 10.0 + + + + Application + true + v143 + Unicode + + + Application + false + v143 + true + Unicode + + + Application + true + v143 + Unicode + + + Application + false + v143 + true + Unicode + + + + + + + + + + + + + + + + + + + + + + Level3 + true + WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + + + + + Level3 + true + true + true + WIN32;NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + true + true + + + + + Level3 + true + _DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + + + + + Level3 + true + true + true + NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + + + Console + true + true + true + + + + + + + + + \ No newline at end of file diff --git a/RebootApplication/RebootApplication.vcxproj.filters b/RebootApplication/RebootApplication.vcxproj.filters new file mode 100644 index 0000000..3e7e62e --- /dev/null +++ b/RebootApplication/RebootApplication.vcxproj.filters @@ -0,0 +1,22 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;c++;cppm;ixx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hh;hpp;hxx;h++;hm;inl;inc;ipp;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + + + Source Files + + + \ No newline at end of file diff --git a/RebootApplication/Source.cpp b/RebootApplication/Source.cpp new file mode 100644 index 0000000..d6397d4 --- /dev/null +++ b/RebootApplication/Source.cpp @@ -0,0 +1,168 @@ +#include +#include +#include +#include + +// windows child process +#include +#include +#include + +// windows terminate process +#include + +// fixed int +#include + +// current directory +#include + +// file read/write +#include +#include + +void EndProc(DWORD procId) +{ + PROCESSENTRY32 pe; + + memset(&pe, 0, sizeof(PROCESSENTRY32)); + pe.dwSize = sizeof(PROCESSENTRY32); + + HANDLE hSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); + + if (::Process32First(hSnap, &pe)) + { + BOOL bContinue = TRUE; + + // kill child processes + while (bContinue) + { + // only kill child processes + if (pe.th32ParentProcessID == procId) + { + HANDLE hChildProc = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, pe.th32ProcessID); + + if (hChildProc) + { + ::TerminateProcess(hChildProc, 1); + ::CloseHandle(hChildProc); + } + } + + bContinue = ::Process32Next(hSnap, &pe); + } + + // kill the main process + HANDLE hProc = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, procId); + + if (hProc) + { + ::TerminateProcess(hProc, 1); + ::CloseHandle(hProc); + } + } +} + +bool CreateProc( + std::string path, + std::string args, + PROCESS_INFORMATION* processInfo +) +{ + std::wstring path_ws = std::wstring(path.begin(), path.end()); + const wchar_t* path_cw = path_ws.c_str(); + + std::wstring cmd_ws = std::wstring(args.begin(), args.end()); + wchar_t* cmd_w = const_cast(cmd_ws.c_str()); + + // creating process + STARTUPINFO startupInfo = { sizeof(startupInfo) }; + if (!CreateProcess(path_cw, cmd_w, NULL, NULL, TRUE, 0, NULL, NULL, &startupInfo, processInfo)) + return false; + return true; +} + +std::string getStringFromFile(std::string fileName) +{ + std::ifstream ifile(fileName); + if (!ifile) + return std::string(); + + std::ostringstream ss; + ss << ifile.rdbuf(); + return ss.str(); +} + +bool outStringToFile(std::string fileName, std::string str) +{ + std::ofstream out(fileName); + if (!out) + return false; + out << str; + return true; +} + +void fileClear(std::string fileName) +{ + std::ifstream(fileName, std::ifstream::out | std::ifstream::trunc); +} + +int main(int argc, char* argv[]) +{ + // disable console input + HANDLE hInput = GetStdHandle(STD_INPUT_HANDLE); + DWORD prev_mode; + GetConsoleMode(hInput, &prev_mode); + SetConsoleMode(hInput, ENABLE_EXTENDED_FLAGS | + (prev_mode & ~ENABLE_QUICK_EDIT_MODE)); + + // args + // 1 - app path + // 2 - file path + // 3 - app args + + // merging app arguments + std::string appPath = argv[1]; + std::fstream file; + std::string appArgs = appPath; + /*for (int i = 3; i < argc; ++i) + appArgs += argv[i] + std::string(" ");*/ + + PROCESS_INFORMATION appProcInfo; + auto startApp = [appPath, appArgs, &appProcInfo]() + { + if (!CreateProc(appPath, appArgs, &appProcInfo)) + { + std::cout << "CreateProcess failed (" << GetLastError() << ")." << std::endl; + } + // bind app process close listen + std::thread([appProcInfo]() + { + WaitForSingleObject(appProcInfo.hProcess, INFINITE); + CloseHandle(appProcInfo.hProcess); + CloseHandle(appProcInfo.hThread); + }).detach(); + }; + startApp(); + try + { + while (true) + { + if (getStringFromFile(argv[2])[0] == '1') + { + std::cout << "reboot" << std::endl; + fileClear(argv[2]); + outStringToFile(argv[2], "0"); + EndProc(GetProcessId(appProcInfo.hProcess)); + startApp(); + } + Sleep(1000); + } + } + catch (const std::exception& e) + { + std::cout << "Error: " << e.what() << std::endl; + return EXIT_FAILURE; + } + return 0; +} \ No newline at end of file