emmy hook

Author: tangzx

CMakeLists.txt

@@ -40,7 +40,6 @@ macro(source_group_by_dir proj_dir source_files)
     endif(MSVC OR APPLE)
 endmacro(source_group_by_dir)
 
-option(EMMY_BUILD_AS_HOOK "build as hook mode" OFF)
 set(EMMY_LUA_VERSION "lua53")
 
 if(${EMMY_LUA_VERSION} STREQUAL "lua53")

Shared/shme.cpp

@@ -0,0 +1,43 @@
+#include "shme.h"
+
+bool CreateMemFile(SharedFile* file) {
+	// Get a handle to our file map
+	const auto MapFile = CreateFileMapping(INVALID_HANDLE_VALUE, nullptr, PAGE_READWRITE, 0, SHMEMSIZE, SHMEMNAME);
+	if (MapFile == nullptr) {
+		MessageBoxA(nullptr, "Failed to create file mapping!", "DLL_PROCESS_ATTACH", MB_OK | MB_ICONERROR);
+		return false;
+	}
+
+	// Get our shared memory pointer
+	const auto MemFile = MapViewOfFile(MapFile, FILE_MAP_ALL_ACCESS, 0, 0, 0);
+	if (MemFile == nullptr) {
+		MessageBoxA(nullptr, "Failed to map shared memory!", "DLL_PROCESS_ATTACH", MB_OK | MB_ICONERROR);
+		return false;
+	}
+	file->lpMemFile = MemFile;
+	file->hMapFile = MapFile;
+	return true;
+}
+
+bool CloseMemFile(SharedFile* file) {
+	UnmapViewOfFile(file->lpMemFile);
+	CloseHandle(file->hMapFile);
+	return true;
+}
+
+bool ReadSharedData(TSharedData& data) {
+	SharedFile file = {};
+	if (!CreateMemFile(&file)) {
+		return false;
+	}
+	memcpy(&data, file.lpMemFile, SHMEMSIZE);
+	// Clean up
+	CloseMemFile(&file);
+	return true;
+}
+
+bool WriteSharedData(HANDLE hMapFile, LPVOID lpMemFile, TSharedData& data) {
+	memset(lpMemFile, 0, SHMEMSIZE);
+	memcpy(lpMemFile, &data, sizeof(TSharedData));
+	return true;
+}

Shared/shme.h

@@ -0,0 +1,30 @@
+#ifndef _SHME_H_
+#define _SHME_H_
+
+#include <Windows.h>
+
+// Data struct to be shared between processes
+struct TSharedData {
+	DWORD dwOffset = 0;
+	HMODULE hModule = nullptr;
+	LPDWORD lpInit = nullptr;
+};
+
+struct SharedFile {
+	HANDLE hMapFile;
+	LPVOID lpMemFile;
+};
+
+// Size (in bytes) of data to be shared
+#define SHMEMSIZE sizeof(TSharedData)
+// Name of the shared file map (NOTE: Global namespaces must have the SeCreateGlobalPrivilege privilege)
+#define SHMEMNAME "InjectedDllName_SHMEM"
+
+bool CreateMemFile(SharedFile* file);
+
+bool CloseMemFile(SharedFile* file);
+
+bool ReadSharedData(TSharedData& data);
+
+bool WriteSharedData(HANDLE hMapFile, LPVOID lpMemFile, TSharedData& data);
+#endif

emmy_core/CMakeLists.txt

@@ -8,11 +8,19 @@ option(EMMY_USE_LUA_SOURCE "Build with lua source" OFF)
 set(LINK_LIBRARIES uv_a)
 
 file(GLOB SRC_LIST
-    *.cpp *.c proto/*.cpp
+    *.cpp proto/*.cpp
 )
 file(GLOB HEADER_LIST
     *.h proto/*.h
 )
+file(GLOB HOOK_LIST
+    dllmain.cpp hook/*.h hook/*.cpp
+    hook/libpe/*.h hook/libpe/*.cpp
+)
+file(GLOB SHARED_LIST
+    ${CMAKE_SOURCE_DIR}/Shared/*.h
+    ${CMAKE_SOURCE_DIR}/Shared/*.cpp
+)
 
 # rapid json
 add_definitions(-DRAPIDJSON_HAS_STDSTRING)
@@ -21,10 +29,6 @@ if(EMMY_CORE_BUILD_AS_DLL)
     add_definitions(-DEMMY_CORE_BUILD_AS_DLL)    
 endif(EMMY_CORE_BUILD_AS_DLL)
 
-if(EMMY_BUILD_AS_HOOK)
-    add_definitions(-DEMMY_BUILD_AS_HOOK)
-endif(EMMY_BUILD_AS_HOOK)
-
 if(EMMY_USE_LUA_SOURCE)
     add_definitions(-DEMMY_USE_LUA_SOURCE)
     include_directories(
@@ -34,35 +38,40 @@ if(EMMY_USE_LUA_SOURCE)
 else(EMMY_USE_LUA_SOURCE)
     file(GLOB SRC_LIST
         *.cpp *.c api/*.cpp proto/*.cpp
-        hook/*.cpp
     )
     file(GLOB HEADER_LIST
         *.h api/*.h proto/*.h
-        hook/*.h
     )
 endif(EMMY_USE_LUA_SOURCE)
 
 include_directories(
+    ${CMAKE_SOURCE_DIR}/Shared
     ${CMAKE_SOURCE_DIR}/third-party/libuv-1.29.0/include
     ${CMAKE_SOURCE_DIR}/third-party/rapidjson-1.1.0/include
     ${CMAKE_SOURCE_DIR}/third-party/EasyHook/EasyHookDll
 )
 
 source_group_by_dir(${CMAKE_CURRENT_SOURCE_DIR} SRC_LIST)
 source_group_by_dir(${CMAKE_CURRENT_SOURCE_DIR} HEADER_LIST)
+source_group_by_dir(${CMAKE_CURRENT_SOURCE_DIR} HOOK_LIST)
+source_group("SHME" FILES SHARED_LIST)
 
 add_library(emmy_core MODULE ${SRC_LIST} ${HEADER_LIST})
 set_target_properties(emmy_core PROPERTIES PREFIX "")
+add_library(emmy_hook MODULE ${SRC_LIST} ${HEADER_LIST} ${HOOK_LIST} ${SHARED_LIST})
+target_compile_definitions(emmy_hook PRIVATE EMMY_BUILD_AS_HOOK)
 if(WIN32)
-    add_dependencies(emmy_core EasyHookDll uv_a)
-    target_link_libraries(emmy_core EasyHookDll uv_a)
+    add_dependencies(emmy_core uv_a)
+    target_link_libraries(emmy_core uv_a)
+    add_dependencies(emmy_hook EasyHook uv_a)
+    target_link_libraries(emmy_hook EasyHook uv_a)
 else(WIN32)
     add_dependencies(emmy_core ${LINK_LIBRARIES})
     target_link_libraries(emmy_core ${LINK_LIBRARIES})
 endif(WIN32)
 
 install(
-    TARGETS emmy_core
-    LIBRARY DESTINATION lib
+    TARGETS emmy_core emmy_hook
+    LIBRARY DESTINATION bin
     RUNTIME DESTINATION bin
 )

emmy_core/api/lua_api.cpp

@@ -257,6 +257,6 @@ extern "C" bool SetupLuaAPI() {
 		luaVersion = LuaVersion::LUA_51;
 		LUA_REGISTRYINDEX = -10000;
 	}
-	//printf("[EMMY]lua version: %d\n", luaVersion);
+	printf("[EMMY]lua version: %d\n", luaVersion);
 	return true;
 }

emmy_core/dllmain.c emmy_core/dllmain.cppemmy_core/dllmain.c renamed to emmy_core/dllmain.cpp

@@ -14,21 +14,40 @@
 * limitations under the License.
 */
 
+#if EMMY_BUILD_AS_HOOK
+
+#include "hook/emmy_hook.h"
+
 #if WIN32
 #include <Windows.h>
+#include "shme.h"
+static SharedFile file;
 
 HINSTANCE g_hInstance = NULL;
 
-BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD reason, LPVOID reserved) {
-	g_hInstance = hInstance;
+BOOL WINAPI DllMain(HINSTANCE hModule, DWORD reason, LPVOID reserved) {
+	g_hInstance = hModule;
 
 	if (reason == DLL_PROCESS_ATTACH) {
-		// MessageBox(NULL, "Waiting to attach the debugger", NULL, MB_OK);
+		TSharedData data;
+		DisableThreadLibraryCalls(hModule);
+		if (!CreateMemFile(&file)) {
+			return FALSE;
+		}
+		// Set shared memory to hold what our remote process needs
+		memset(file.lpMemFile, 0, SHMEMSIZE);
+		data.hModule = hModule;
+		data.lpInit = LPDWORD(StartupHookMode);
+		data.dwOffset = DWORD(data.lpInit) - DWORD(data.hModule);
+		memcpy(file.lpMemFile, &data, sizeof(TSharedData));
 	}
 	else if (reason == DLL_PROCESS_DETACH) {
 		// Destroy();
+		CloseMemFile(&file);
 	}
 
 	return TRUE;
 }
 #endif
+
+#endif

emmy_core/emmy_core.cpp

@@ -90,7 +90,7 @@ int stop(lua_State* L) {
 }
 
 int gc(lua_State* L) {
-	EmmyFacade::Get()->Destroy();
+	EmmyFacade::Get()->OnLuaStateGC(L);
 	return 0;
 }
 
@@ -132,18 +132,22 @@ LuaVersion luaVersion = LuaVersion::UNKNOWN;
 extern "C" {
 	bool SetupLuaAPI();
 
-EMMY_CORE_EXPORT int luaopen_emmy_core(struct lua_State* L) {
+	bool install_emmy_core(struct lua_State* L) {
 #ifndef EMMY_USE_LUA_SOURCE
-	if (!SetupLuaAPI()) {
-		return 0;
-	}
+		if (!SetupLuaAPI()) {
+			return false;
+		}
 #endif
+		// register helper lib
+		luaopen_emmy_helper(L);
+		handleStateClose(L);
+		return true;
+	}
 
-	// register helper lib
-	luaopen_emmy_helper(L);
-	handleStateClose(L);
-	luaL_newlib(L, lib);
-
-	return 1;
-}
+	EMMY_CORE_EXPORT int luaopen_emmy_core(struct lua_State* L) {
+		if (!install_emmy_core(L))
+			return false;
+		luaL_newlib(L, lib);
+		return 1;
+	}
 }

emmy_core/emmy_core.h

@@ -45,4 +45,4 @@ enum class LuaVersion {
 	LUA_53 = 53
 };
 
-extern LuaVersion luaVersion;
+extern LuaVersion luaVersion;

emmy_core/emmy_facade.cpp

@@ -152,6 +152,11 @@ void EmmyFacade::Destroy() {
 void EmmyFacade::OnReceiveMessage(const rapidjson::Document& document) {
 	const auto cmd = static_cast<MessageCMD>(document["cmd"].GetInt());
 	switch (cmd) {
+#if EMMY_BUILD_AS_HOOK
+	case MessageCMD::StartHookReq:
+		StartHook();
+		break;
+#endif
 	case MessageCMD::InitReq:
 		OnInitReq(document);
 		break;
@@ -165,11 +170,11 @@ void EmmyFacade::OnReceiveMessage(const rapidjson::Document& document) {
 		OnRemoveBreakPointReq(document);
 		break;
 	case MessageCMD::ActionReq:
-		assert(isIDEReady);
+		//assert(isIDEReady);
 		OnActionReq(document);
 		break;
 	case MessageCMD::EvalReq:
-		assert(isIDEReady);
+		//assert(isIDEReady);
 		OnEvalReq(document);
 		break;
 	default:
@@ -359,3 +364,32 @@ void EmmyFacade::OnEvalResult(EvalContext* context) {
 		transporter->Send(int(MessageCMD::EvalRsp), rspDoc);
 	delete context;
 }
+
+void EmmyFacade::SendLog(LogType type, const char *fmt, ...) {
+	va_list args;
+	va_start(args, fmt);
+	char buff[1024] = { 0 };
+	vsnprintf_s(buff, 1024, fmt, args);
+	va_end(args);
+
+	const std::string msg = buff;
+
+	rapidjson::Document rspDoc;
+	rspDoc.SetObject();
+	auto& allocator = rspDoc.GetAllocator();
+	rspDoc.AddMember("type", (int)type, allocator);
+	rspDoc.AddMember("message", msg, allocator);
+	if (transporter)
+		transporter->Send(int(MessageCMD::LogNotify), rspDoc);
+}
+
+void EmmyFacade::OnLuaStateGC(lua_State* L) {
+#if EMMY_BUILD_AS_HOOK
+	Debugger::Get()->Stop();
+	this->attachedStates.clear();
+	this->isIDEReady = false;
+	this->L = nullptr;
+#else
+	Destroy();
+#endif
+}

emmy_core/emmy_facade.h

@@ -18,10 +18,15 @@
 #include <rapidjson/document.h>
 #include <mutex>
 #include <condition_variable>
+#include <set>
 
 class Transporter;
 class EvalContext;
 
+enum class LogType {
+	Info, Warning, Error
+};
+
 class EmmyFacade {
 	Transporter* transporter;
 	lua_State* L;
@@ -45,11 +50,21 @@ class EmmyFacade {
 	void OnBreak();
 	void Destroy();
 	void OnEvalResult(EvalContext* context);
+	void SendLog(LogType type, const char *fmt, ...);
+	void OnLuaStateGC(lua_State* L);
 private:
 	void OnInitReq(const rapidjson::Document& document);
 	void OnReadyReq(const rapidjson::Document& document);
 	void OnAddBreakPointReq(const rapidjson::Document& document);
 	void OnRemoveBreakPointReq(const rapidjson::Document& document);
 	void OnActionReq(const rapidjson::Document& document);
 	void OnEvalReq(const rapidjson::Document& document);
+#ifdef EMMY_BUILD_AS_HOOK
+private:
+	std::set<lua_State*> attachedStates;
+public:
+	void StartupHookMode(int port);
+	void Attach(lua_State* L);
+	void StartHook();
+#endif
 };