Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(105)

Unified Diff: sandbox/win/src/service_resolver_64.cc

Issue 183833004: Make chrome_elf use thunks instead of function pointers. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 6 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: sandbox/win/src/service_resolver_64.cc
diff --git a/sandbox/win/src/service_resolver_64.cc b/sandbox/win/src/service_resolver_64.cc
index 96124187deb4b383708bca09645f81f3bced75f9..5c5875f221ff42cf9b72196fab0c8ae9c3f68d12 100644
--- a/sandbox/win/src/service_resolver_64.cc
+++ b/sandbox/win/src/service_resolver_64.cc
@@ -116,6 +116,62 @@ size_t ServiceResolverThunk::GetThunkSize() const {
return sizeof(ServiceFullThunk);
}
+NTSTATUS ServiceResolverThunk::CopyThunk(const void* target_module,
+ const char* target_name,
+ void* thunk_storage,
+ size_t storage_bytes,
+ size_t* storage_used) {
+ NTSTATUS ret = ResolveTarget(target_module, target_name, &target_);
+ if(!NT_SUCCESS(ret))
robertshield 2014/02/28 21:02:22 space after if
Cait (Slow) 2014/03/03 20:55:11 Done.
+ return ret;
+
+ size_t thunk_bytes = GetThunkSize();
+ scoped_ptr<char[]> thunk_buffer(new char[thunk_bytes]);
robertshield 2014/02/28 21:02:22 we use BYTE below for storage and chars here. Shou
Cait (Slow) 2014/03/03 20:55:11 Done.
+ ServiceFullThunk* thunk = reinterpret_cast<ServiceFullThunk*>(
+ thunk_buffer.get());
+
+ if (!IsFunctionAService(&thunk->original))
+ return STATUS_UNSUCCESSFUL;
+
+ BYTE* thunk_storage_bytes = reinterpret_cast<BYTE*>(thunk_storage);
+
+ // Mark the thunk storage as readable and writeable, since we
+ // ready to write to it.
+ DWORD old_protect = 0;
+ if(!VirtualProtect(thunk_storage,
robertshield 2014/02/28 21:02:22 space after if
Cait (Slow) 2014/03/03 20:55:11 Done.
+ sizeof(thunk_storage),
+ PAGE_EXECUTE_READWRITE,
+ &old_protect)) {
+ return STATUS_UNSUCCESSFUL;
+ }
+
+ // copy the local thunk buffer to the child
+ SIZE_T written;
robertshield 2014/02/28 21:02:22 = 0
Cait (Slow) 2014/03/03 20:55:11 Done.
+ ret = ::WriteProcessMemory(process_,
+ thunk_storage_bytes,
+ reinterpret_cast<void*>(&thunk->original),
+ thunk_bytes,
+ &written);
+
+ if (!NT_SUCCESS(ret))
+ return ret;
+
+ if (thunk_bytes != written)
+ return STATUS_UNSUCCESSFUL;
+
+ if (VirtualProtect(thunk_storage,
+ sizeof(thunk_storage),
+ PAGE_EXECUTE_READ,
+ &old_protect)) {
+ return STATUS_UNSUCCESSFUL;
+ }
+
+ if (NULL != storage_used)
+ *storage_used = thunk_bytes;
+
+ return ret;
+}
+
bool ServiceResolverThunk::IsFunctionAService(void* local_thunk) const {
ServiceFullThunk function_code;
SIZE_T read;
« sandbox/win/src/service_resolver_32.cc ('K') | « sandbox/win/src/service_resolver_32.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698