#include "mktmpdir.h"
#define PAR_TEMP "PAR_TEMP"
#ifdef O_BINARY
# define OPEN_O_BINARY O_BINARY
#else
# define OPEN_O_BINARY 0
#endif
#ifndef P_tmpdir
#define P_tmpdir "/tmp"
#endif
/* NOTE: The code below is #include'd both from a plain C program (boot.c)
* and our custom Perl interpreter (main.c). In the latter case,
* lstat() or stat() may be #define'd as calls into PerlIO and
* expect &PL_statbuf as second parameter, rather than a pointer
* to a struct stat. Try to distinguish these cases by checking
* whether PL_statbuf is defined. */
static int isWritableDir(const char* val)
{
#ifndef PL_statbuf
struct stat PL_statbuf;
#endif
return par_lstat(val, &PL_statbuf) == 0 &&
( S_ISDIR(PL_statbuf.st_mode) || S_ISLNK(PL_statbuf.st_mode) ) &&
access(val, W_OK) == 0;
}
#ifndef WIN32
/* check that:
* - val is a directory (and not a symlink)
* - val is owned by the user
* - val has mode 0700
*/
static int isSafeDir(const char* val)
{
#ifndef PL_statbuf
struct stat PL_statbuf;
#endif
return par_lstat(val, &PL_statbuf) == 0 &&
S_ISDIR(PL_statbuf.st_mode) &&
PL_statbuf.st_uid == getuid() &&
(PL_statbuf.st_mode & 0777) == 0700;
}
#endif
void par_setup_libpath( const char * stmpdir )
{
const char *key = NULL , *val = NULL;
int i;
char *ld_path_env = NULL;
/* NOTE: array is NULL terminated */
const char *ld_path_keys[] = {
"LD_LIBRARY_PATH", "LIBPATH", "LIBRARY_PATH",
"PATH", "DYLD_LIBRARY_PATH", "SHLIB_PATH", NULL
};
for ( i = 0 ; key = ld_path_keys[i]; i++ ) {
if ( (val = par_getenv(key)) == NULL || strlen(val) == 0 ) {
par_setenv(key, stmpdir);
}
else if ( !strstr(val, stmpdir) ) {
/* prepend stmpdir to (value of) environment variable */
ld_path_env = malloc(
strlen(stmpdir) + strlen(path_sep) + strlen(val) + 1);
sprintf(
ld_path_env, "%s%s%s",
stmpdir, path_sep, val);
par_setenv(key, ld_path_env);
}
}
}
char *par_mktmpdir ( char **argv ) {
int i;
const char *tmpdir = NULL;
const char *key = NULL , *val = NULL;
/* NOTE: all arrays below are NULL terminated */
const char *temp_dirs[] = {
P_tmpdir,
#ifdef WIN32
"C:\\TEMP",
#endif
".", NULL };
const char *temp_keys[] = { "PAR_TMPDIR", "TMPDIR", "TEMPDIR",
"TEMP", "TMP", NULL };
const char *user_keys[] = { "USER", "USERNAME", NULL };
const char *subdirbuf_prefix = "par-";
const char *subdirbuf_suffix = "";
char *progname = NULL, *username = NULL;
char *stmpdir = NULL, *top_tmpdir = NULL;
int f, j, k, stmp_len = 0;
char sha1[41];
SHA_INFO sha_info;
unsigned char buf[32768];
unsigned char sha_data[20];
if ( (val = par_getenv(PAR_TEMP)) && strlen(val) ) {
par_setup_libpath(val);
return strdup(val);
}
#ifdef WIN32
{
DWORD buflen = MAXPATHLEN;
username = malloc(MAXPATHLEN);
GetUserName((LPTSTR)username, &buflen);
// FIXME this is uncondifionally overwritten below - WTF?
}
#endif
/* Determine username */
username = get_username_from_getpwuid();
if ( !username ) { /* fall back to env vars */
for ( i = 0 ; username == NULL && (key = user_keys[i]); i++) {
if ( (val = par_getenv(key)) && strlen(val) )
username = strdup(val);
}
}
if ( username == NULL )
username = "SYSTEM";
/* sanitize username: encode all bytes as 2 hex digits */
{
char *hexname = malloc(2 * strlen(username) + 1);
char *u, *h;
for ( u = username, h = hexname ; *u != '\0' ; u++, h += 2)
sprintf(h, "%02x", *(unsigned char*)u);
username = hexname;
}
/* Try temp environment variables */
for ( i = 0 ; tmpdir == NULL && (key = temp_keys[i]); i++ ) {
if ( (val = par_getenv(key)) && strlen(val) && isWritableDir(val) ) {
tmpdir = strdup(val);
break;
}
}
#ifdef WIN32
/* Try the windows temp directory */
if ( tmpdir == NULL && (val = par_getenv("WinDir")) && strlen(val) ) {
char* buf = malloc(strlen(val) + 5 + 1);
sprintf(buf, "%s\\temp", val);
if (isWritableDir(buf)) {
tmpdir = buf;
} else {
free(buf);
}
}
#endif
/* Try default locations */
for ( i = 0 ; tmpdir == NULL && (val = temp_dirs[i]) && strlen(val) ; i++ ) {
if ( isWritableDir(val) ) {
tmpdir = strdup(val);
}
}
/* "$TEMP/par-$USER" */
stmp_len =
strlen(tmpdir) +
strlen(subdirbuf_prefix) +
strlen(username) +
strlen(subdirbuf_suffix) + 1024;
/* stmpdir is what we are going to return;
top_tmpdir is the top $TEMP/par-$USER, needed to build stmpdir.
NOTE: We need 2 buffers because snprintf() can't write to a buffer
it is also reading from. */
top_tmpdir = malloc( stmp_len );
sprintf(top_tmpdir, "%s%s%s%s", tmpdir, dir_sep, subdirbuf_prefix, username);
#ifdef WIN32
_mkdir(top_tmpdir); /* FIXME bail if error (other than EEXIST) */
#else
{
if (mkdir(top_tmpdir, 0700) == -1 && errno != EEXIST) {
fprintf(stderr, "%s: creation of private subdirectory %s failed (errno=%i)\n",
argv[0], top_tmpdir, errno);
return NULL;
}
if (!isSafeDir(top_tmpdir)) {
fprintf(stderr, "%s: private subdirectory %s is unsafe (please remove it and retry your operation)\n",
argv[0], top_tmpdir);
return NULL;
}
}
#endif
stmpdir = malloc( stmp_len );
/* Doesn't really work - XXX */
val = par_getenv( "PATH" );
if (val != NULL)
progname = par_findprog(argv[0], strdup(val));
if (progname == NULL)
progname = argv[0];
/* If invoked as "/usr/bin/parl foo.par myscript.pl" then progname should
* be ".../parl", and we don't want to base our checksum on that, but
* rather on "foo.par".
*/
{
#ifdef WIN32
#define STREQ(a,b) (strcasecmp(a,b) == 0)
#else
#define STREQ(a,b) (strcmp(a,b) == 0)
#endif
int prog_len = strlen(progname);
int parl_len = strlen(PARL_EXE);
if (prog_len >= parl_len
&& STREQ(progname + prog_len - parl_len, PARL_EXE)
&& (prog_len == parl_len || progname[prog_len - parl_len - 1] == dir_sep[0])
&& argv[1]
&& strlen(argv[1]) >= 4
&& STREQ(argv[1] + strlen(argv[1]) - 4, ".par"))
progname = argv[1];
#undef STREQ
}
if ( !par_env_clean() && (f = open( progname, O_RDONLY | OPEN_O_BINARY ))) {
lseek(f, -18, 2);
read(f, buf, 6);
if(buf[0] == 0 && buf[1] == 'C' && buf[2] == 'A' && buf[3] == 'C' && buf[4] == 'H' && buf[5] == 'E') {
/* pre-computed cache_name in this file */
/* "$TEMP/par-$USER/cache-$cache_name" */
lseek(f, -58, 2);
read(f, buf, 41);
sprintf(
stmpdir,
"%s%scache-%s%s",
top_tmpdir, dir_sep, buf, subdirbuf_suffix
);
}
else {
/* "$TEMP/par-$USER/cache-$SHA1" */
lseek(f, 0, 0);
sha_init( &sha_info );
while( ( j = read( f, buf, sizeof( buf ) ) ) > 0 )
{
sha_update( &sha_info, buf, j );
}
close( f );
sha_final( sha_data, &sha_info );
for( k = 0; k < 20; k++ )
{
sprintf( sha1+k*2, "%02x", sha_data[k] );
}
sha1[40] = '\0';
sprintf(
stmpdir,
"%s%scache-%s%s",
top_tmpdir, dir_sep, sha1, subdirbuf_suffix
);
}
}
else {
int i = 0;
/* "$TEMP/par-$USER/temp-$PID" */
par_setenv("PAR_CLEAN", "1");
sprintf(
stmpdir,
"%s%stemp-%u%s",
top_tmpdir, dir_sep, getpid(), subdirbuf_suffix
);
/* Ensure we pick an unused directory each time. If the directory
already exists when we try to create it, bump a counter and try
"$TEMP/par-$USER/temp-$PID-$i". This will guard against cases where
a prior invocation crashed leaving garbage in a temp directory that
might interfere. */
while (my_mkdir(stmpdir, 0700) == -1 && errno == EEXIST) {
sprintf(
stmpdir,
"%s%stemp-%u-%u%s",
top_tmpdir, dir_sep, getpid(), ++i, subdirbuf_suffix
);
}
}
free(top_tmpdir);
/* set dynamic loading path */
par_setenv(PAR_TEMP, stmpdir);
par_setup_libpath( stmpdir );
return stmpdir;
}
#ifdef WIN32
static void par_rmtmpdir ( char *stmpdir ) {
struct _finddata_t cur_file;
int subsub_len;
char *subsubdir;
char *slashdot;
long hFile;
int tries = 0;
HMODULE dll;
if ((stmpdir == NULL) || !strlen(stmpdir)) return;
subsub_len = strlen(stmpdir) + 258;
subsubdir = malloc( subsub_len );
sprintf(subsubdir, "%s\\*.*", stmpdir);
hFile = _findfirst( subsubdir, &cur_file );
if ( hFile == -1 ) return;
do {
if (!strstr(cur_file.name, "\\")) {
sprintf(subsubdir, "%s\\%s", stmpdir, cur_file.name);
}
else {
sprintf(subsubdir, "%s", cur_file.name);
}
if (!(slashdot = strstr(subsubdir, "\\.")) || (strcmp(slashdot,"\\.") && strcmp(slashdot,"\\.."))) {
if ((cur_file.attrib & _A_SUBDIR)) {
par_rmtmpdir( subsubdir );
}
else {
dll = GetModuleHandle(cur_file.name);
tries = 0;
while ( _unlink(subsubdir) && ( tries++ < 10 ) ) {
if ( dll ) FreeLibrary(dll);
};
}
}
} while ( _findnext( hFile, &cur_file ) == 0 );
_findclose(hFile);
_rmdir(stmpdir);
}
#else
static void par_rmtmpdir ( char *stmpdir ) {
DIR *partmp_dirp;
Direntry_t *dp;
char *subsubdir = NULL;
int subsub_len;
struct stat stbuf;
/* remove temporary PAR directory */
if (!stmpdir || !*stmpdir) return;
partmp_dirp = opendir(stmpdir);
if ( partmp_dirp == NULL ) return;
while ( ( dp = readdir(partmp_dirp) ) != NULL ) {
if ( strcmp (dp->d_name, ".") != 0 && strcmp (dp->d_name, "..") != 0 )
{
subsub_len = strlen(stmpdir) + 1 + strlen(dp->d_name) + 1;
subsubdir = malloc( subsub_len);
sprintf(subsubdir, "%s/%s", stmpdir, dp->d_name);
if (stat(subsubdir, &stbuf) != -1 && S_ISDIR(stbuf.st_mode)) {
par_rmtmpdir(subsubdir);
}
else {
unlink(subsubdir);
}
free(subsubdir);
subsubdir = NULL;
}
}
closedir(partmp_dirp);
rmdir(stmpdir);
}
#endif
void par_cleanup (char *stmpdir) {
char *dirname = par_dirname(stmpdir);
char *basename = par_basename(dirname);
if ( par_env_clean() && stmpdir != NULL && strlen(stmpdir)) {
if ( strstr(basename, "par-") == basename ) {
par_rmtmpdir(stmpdir);
/* Don't try to remove dirname because this will introduce a race
with other applications that are trying to start. */
}
}
}