2 Copyright © 1995-2012, The AROS Development Team. All rights reserved.
5 Desc: special main function for code which has to use special *nix features.
6 This function gets called from a function with a similar name statically
7 linked with the program. This is so to make the program not depend on a
8 particular libc version.
13 #include LC_LIBDEFS_FILE
15 #include <proto/exec.h>
16 #include <proto/dos.h>
17 #include <exec/lists.h>
19 #include <aros/startup.h>
22 #include <aros/debug.h>
27 #include <sys/param.h>
29 #include "__arosc_privdata.h"
32 static BOOL
clone_vars(struct MinList
*old_vars
);
33 static void restore_vars(struct MinList
*old_vars
);
34 static void free_vars(struct MinList
*vars
);
35 static void update_PATH(void);
37 int __arosc_nixmain(int (*main
)(int argc
, char *argv
[]), int argc
, char *argv
[])
39 struct aroscbase
*aroscbase
= __aros_getbase_aroscbase(), *paroscbase
;
40 int *errorptr
= __arosc_get_errorptr();
41 char *old_argv0
= NULL
;
42 char *new_argv0
= NULL
;
43 struct MinList old_vars
;
45 D(bug("__arosc_nixmain: @begin, Task=%x\n", FindTask(NULL
)));
47 /* Trigger *nix path handling on. */
48 aroscbase
->acb_doupath
= 1;
50 /* argv[0] usually contains the name of the program, possibly with the full
51 path to it. Here we translate that path, which is an AmigaDOS-style path,
52 into an unix-style one. */
53 if (argv
&& argv
[0] && !aroscbase
->acb_parent_does_upath
)
55 new_argv0
= strdup(__path_a2u(argv
[0]));
56 if (new_argv0
== NULL
)
63 /* Here we clone environment variables. We do this because
64 if we've been invoked as a coroutine via dos.library/RunCommand()
65 rather than as a newly created process, then we share our env variables
66 with the caller, but we do not want that. It's kind of wasteful to do
67 it even if we've been started as a fresh process, though, so if we can
69 The cloning does not need to be performed if flags of parent have VFORK_PARENT
70 or EXEC_PARENT flags */
71 paroscbase
= __GM_GetBaseParent(aroscbase
);
72 D(bug("__arosc_nixmain: paroscbase = %x, Task=%x\n", paroscbase
, FindTask(NULL
)));
73 if (!paroscbase
|| !(paroscbase
->acb_flags
& (VFORK_PARENT
| EXEC_PARENT
)))
75 D(bug("__arosc_nixmain: Cloning LocalVars"));
76 if (!clone_vars(&old_vars
))
79 *errorptr
= RETURN_FAIL
;
84 /* If the PATH variable is not defined, then define it to be what CLI's path list
89 /* Call the real main. */
90 jmp_buf exitjmp
, dummyjmp
;
91 if (setjmp(exitjmp
) == 0)
95 __arosc_set_exitjmp(exitjmp
, dummyjmp
);
97 ret
= (*main
)(argc
, argv
);
102 D(bug("__arosc_nixmain: setjmp() != 0\n"));
104 D(bug("__arosc_nixmain: paroscbase = %x, Task=%x\n", paroscbase
, FindTask(NULL
)));
105 if (!paroscbase
|| !(paroscbase
->acb_flags
& (VFORK_PARENT
| EXEC_PARENT
)))
106 restore_vars(&old_vars
);
110 /* Restore the old argv[0]. */
111 if (old_argv0
!= NULL
)
114 argv
[0] = (char *)old_argv0
;
117 D(bug("__arosc_nixmain: @end, Task=%x\n", FindTask(NULL
)));
119 return (errorptr
!= NULL
) ? *errorptr
: 0;
122 /* Clone the process' environment variables list. Once this function returns,
123 the _cloned_ vars are used in place of the old ones.
125 Returns the old list's content in the old_vars argument. Use this list only as
126 argument to restore_vars() and for _nothing_ else.
128 If this function fails, then FALSE is returned, otherwise TRUE is returned.
130 One might argue that the whole affair is solved the wrong way around, that is
131 clone_vars() should return the _cloned_ list and restore_vars() should be really
132 named replace_vars() and take the cloned list as argument and replace the current
133 one with that one, but doing it this way means breaking any programs which saved
134 vars list's internals before invoking this program, although that is not even guaranteed
135 to work normally, as the called program could change the env list anytime... Well,
136 in either case it doesn't do much of a difference, since the same amount of operations
137 would be performed, only the order would change. */
139 BOOL
clone_vars(struct MinList
*old_vars
)
142 struct LocalVar
*lv
, *newVar
;
147 me
= (struct Process
*)FindTask(NULL
);
148 /* Copied and adapted from rom/dos/createnewproc.c. Perhaps there should
149 be a public function for this? */
150 ForeachNode(&me
->pr_LocalVars
, lv
)
157 copyLength
= strlen(lv
->lv_Node
.ln_Name
) + 1 + sizeof(struct LocalVar
);
159 newVar
= (struct LocalVar
*)AllocVec(copyLength
, MEMF_PUBLIC
);
166 memcpy(newVar
, lv
, copyLength
);
167 newVar
->lv_Node
.ln_Name
= (char *)newVar
+ sizeof(struct LocalVar
);
168 newVar
->lv_Value
= AllocMem(lv
->lv_Len
, MEMF_PUBLIC
);
170 if (newVar
->lv_Value
== NULL
)
177 memcpy(newVar
->lv_Value
, lv
->lv_Value
, lv
->lv_Len
);
182 *old_vars
= me
->pr_LocalVars
;
183 me
->pr_LocalVars
= l
;
185 l
.mlh_Head
->mln_Pred
= (struct MinNode
*)&me
->pr_LocalVars
.mlh_Head
;
186 l
.mlh_TailPred
->mln_Succ
= (struct MinNode
*)&me
->pr_LocalVars
.mlh_Tail
;
191 /* Restores the old env var's list content */
192 static void restore_vars(struct MinList
*old_vars
)
194 struct Process
*me
= (struct Process
*)FindTask(NULL
);
196 free_vars(&me
->pr_LocalVars
);
197 me
->pr_LocalVars
= *old_vars
;
200 /* taken from rom/dos/createnewproc.c. */
201 static void free_vars(struct MinList
*vars
)
203 struct LocalVar
*varNode
;
204 struct Node
*tempNode
;
206 ForeachNodeSafe(vars
, varNode
, tempNode
)
208 FreeMem(varNode
->lv_Value
, varNode
->lv_Len
);
209 Remove((struct Node
*)varNode
);
214 /* setenv("PATH", current_cli_path, 1) */
215 static void update_PATH(void)
223 #define PE(x) ((PathEntry *)(BADDR(x)))
225 UBYTE aname
[PATH_MAX
]; /* PATH_MAX ought to be enough; it would be too complicated
226 handling aname dynamically (thanks to our sucky dos.library). */
230 struct CommandLineInterface
*cli
= Cli();
232 /* No cli, no luck. */
238 cur
= PE(cli
->cli_CommandDir
);
247 if (NameFromLock(cur
->lock
, aname
, sizeof(aname
)) == DOSFALSE
)
250 D(bug("aname = %s\n", aname
));
252 uname
= __path_a2u((const char *)aname
);
255 uname_len
= strlen(uname
);
257 D(bug("uname = %s\n", uname
));
259 new_PATH
= realloc(PATH
, PATH_len
+ uname_len
+ 1);
264 memcpy(PATH
+ PATH_len
, uname
, uname_len
);
265 PATH_len
+= uname_len
;
266 PATH
[PATH_len
++] = ':';
268 D(bug("PATH_len = %d, PATH = %.*s\n", PATH_len
, PATH_len
, PATH
));
273 PATH
[PATH_len
? (PATH_len
- 1) : 0] = '\0';
275 setenv("PATH", PATH
, 1);