From 60d284141d301647e4fdfd0d6d0f792ba984df33 Mon Sep 17 00:00:00 2001
From: Nguyễn Thái Ngọc Duy <pclouds@gmail.com>
Date: Wed, 14 Apr 2010 01:01:36 +0200
Subject: win32: ash: forkshell_init()
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Nguyễn Thái Ngọc Duy <pclouds@gmail.com>
---
 shell/ash.c | 68 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

(limited to 'shell')

diff --git a/shell/ash.c b/shell/ash.c
index f320f26c3..767c095e7 100644
--- a/shell/ash.c
+++ b/shell/ash.c
@@ -13178,6 +13178,7 @@ static const forkpoint_fn forkpoints[] = {
 };
 
 static struct forkshell* forkshell_prepare(struct forkshell *fs);
+static void forkshell_init(const char *idstr);
 #endif
 
 /*
@@ -13247,6 +13248,15 @@ int ash_main(int argc UNUSED_PARAM, char **argv)
 
 	init();
 	setstackmark(&smark);
+
+#if ENABLE_PLATFORM_MINGW32
+	if (argc == 3 && !strcmp(argv[1], "--forkshell")) {
+		forkshell_init(argv[2]);
+
+		/* NOTREACHED */
+		bb_error_msg_and_die("subshell ended unexpectedly");
+	}
+#endif
 	procargs(argv);
 
 #if ENABLE_FEATURE_EDITING_SAVEHISTORY
@@ -13769,6 +13779,64 @@ forkshell_prepare(struct forkshell *fs)
 	new->hMapFile = h;
 	return new;
 }
+
+#undef exception_handler
+#undef trap
+#undef trap_ptr
+static void
+forkshell_init(const char *idstr)
+{
+	struct forkshell *fs;
+	int map_handle;
+	HANDLE h;
+	struct globals_var **gvpp;
+	struct globals_misc **gmpp;
+	int i;
+
+	if (sscanf(idstr, "%x", &map_handle) != 1)
+		bb_error_msg_and_die("invalid forkshell ID");
+
+	h = (HANDLE)map_handle;
+	fs = (struct forkshell *)MapViewOfFile(h, FILE_MAP_WRITE, 0,0, 0);
+	if (!fs)
+		bb_error_msg_and_die("Invalid forkshell memory");
+
+	/* pointer fixup */
+	nodeptr = (int*)((char*)fs + fs->nodeptr_offset);
+	while (*nodeptr) {
+		int *ptr = (int*)((char*)fs + (*nodeptr - (int)fs->old_base));
+		if (*ptr)
+			*ptr -= ((int)fs->old_base - (int)fs);
+		nodeptr++;
+	}
+	/* Now fix up stuff that can't be transferred */
+	fs->fp = forkpoints[fs->fpid];
+	for (i = 0; i < ARRAY_SIZE(varinit_data); i++)
+		fs->gvp->varinit[i].func = varinit_data[i].func;
+	for (i = 0; i < CMDTABLESIZE; i++) {
+		struct tblentry *e = fs->cmdtable[i];
+		while (e) {
+			if (e->cmdtype == CMDBUILTIN)
+				e->param.cmd = builtintab + (int)e->param.cmd;
+			e = e->next;
+		}
+	}
+	fs->gmp->exception_handler = ash_ptr_to_globals_misc->exception_handler;
+	for (i = 0; i < NSIG; i++)
+		fs->gmp->trap[i] = ash_ptr_to_globals_misc->trap[i];
+	fs->gmp->trap_ptr = ash_ptr_to_globals_misc->trap_ptr;
+
+	/* Switch global variables */
+	gvpp = (struct globals_var **)&ash_ptr_to_globals_var;
+	*gvpp = fs->gvp;
+	gmpp = (struct globals_misc **)&ash_ptr_to_globals_misc;
+	*gmpp = fs->gmp;
+	localvars = fs->localvars;
+	cmdtable = fs->cmdtable;
+
+	fs->fp(fs);
+}
+
 /*-
  * Copyright (c) 1989, 1991, 1993, 1994
  *      The Regents of the University of California.  All rights reserved.
-- 
cgit v1.2.3-55-g6feb