From: Marko Semet Subject: [PATCH 2/7] winebuild: generate deterministic temporary filenames Message-Id: <20200624134539.AHZpk%marko@marko10-000.de> Date: Wed, 24 Jun 2020 15:45:39 +0200 Using xorshift* as pseudorandom number generator. It get initialised with the environment (sorted by name) and arguments. Signed-off-by: Marko Semet --- tools/winebuild/build.h | 6 ++ tools/winebuild/main.c | 1 + tools/winebuild/utils.c | 160 +++++++++++++++++++++++++++++++++++++++- 3 files changed, 165 insertions(+), 2 deletions(-) diff --git a/tools/winebuild/build.h b/tools/winebuild/build.h index c162888a03..1ff66445c1 100644 --- a/tools/winebuild/build.h +++ b/tools/winebuild/build.h @@ -27,6 +27,7 @@ # error You must include config.h to use this header #endif +#include #include #include #include @@ -352,6 +353,11 @@ extern void put_qword( unsigned int val ); extern void put_pword( unsigned int val ); extern void align_output( unsigned int align ); +extern uint64_t pseudorandom(uint64_t add_to_state); +extern char pseudorandom_char(void); +extern int pseudorandom_tempfile(char* file); +extern void init_random_generator(int argc, char **argv); + /* global variables */ extern int current_line; diff --git a/tools/winebuild/main.c b/tools/winebuild/main.c index efb86b705e..16796b4220 100644 --- a/tools/winebuild/main.c +++ b/tools/winebuild/main.c @@ -666,6 +666,7 @@ int main(int argc, char **argv) signal( SIGTERM, exit_on_signal ); signal( SIGINT, exit_on_signal ); + init_random_generator(argc, argv); argv = parse_options( argc, argv, spec ); atexit( cleanup ); /* make sure we remove the output file on exit */ diff --git a/tools/winebuild/utils.c b/tools/winebuild/utils.c index 2a1fc96092..e2313aab51 100644 --- a/tools/winebuild/utils.c +++ b/tools/winebuild/utils.c @@ -514,13 +514,13 @@ char *get_temp_file_name( const char *prefix, const char *suffix ) strcpy( name + (ext - prefix), ".XXXXXX" ); strcat( name, suffix ); - if ((fd = mkstemps( name, strlen(suffix) )) == -1) + if ((fd = pseudorandom_tempfile(name)) == -1) { strcpy( name, "/tmp/" ); memcpy( name + 5, basename, ext - basename ); strcpy( name + 5 + (ext - basename), ".XXXXXX" ); strcat( name, suffix ); - if ((fd = mkstemps( name, strlen(suffix) )) == -1) + if ((fd = pseudorandom_tempfile(name)) == -1) fatal_error( "could not generate a temp file\n" ); } @@ -1275,3 +1275,159 @@ const char *get_asm_string_section(void) default: return ".section .rodata"; } } + +/* xorshift* random generator. Source: https://en.wikipedia.org/wiki/Xorshift#xorshift* */ +static uint64_t pseudorandom_state = 1; + +static uint64_t pseudorandom_algo(uint64_t* generator_state, uint64_t add_to_state) +{ + uint64_t tmp = (*generator_state) + add_to_state; + tmp ^= tmp >> 12; + tmp ^= tmp << 25; + tmp ^= tmp >> 27; + (*generator_state) = tmp; + return tmp * UINT64_C(0x2545F4914F6CDD1D); +} + +uint64_t pseudorandom(uint64_t add_to_state) +{ + return pseudorandom_algo(&pseudorandom_state, add_to_state); +} + +char pseudorandom_char(void) +{ + char tmp; + tmp = (char) (pseudorandom(0) % 62); + if (tmp < 10) + { + return '0' + tmp; + } + else if (tmp < 36) + { + return 'A' + (tmp - 10); + } + else + { + return 'a' + (tmp - 36); + } +} + +int pseudorandom_tempfile(char* file) +{ + /* variables */ + int result; + + int counter; + char *filename_pos; + + char *debug_filename; + FILE *debug_file; + + /* replaces Xs */ + for (filename_pos = file; (*filename_pos) != 0; filename_pos++) + { + if ((*filename_pos) == 'X') + { + (*filename_pos) = pseudorandom_char(); + } + } + + /* try to open file */ + counter = 0; + do + { + result = open(file, O_CREAT | O_EXCL | O_RDWR, 0600); + if (result != -1) + { + break; + } + sleep(5); + counter++; + } while (counter <= 12); + return result; +} + +void init_random_generator(int argc, char **argv) +{ + /* variables */ + uint64_t generator_state; + + void *argv_data; + size_t argv_size; + + unsigned int counter; + void *env_data; + size_t env_size; + char **tmp_envs; + + char *buffer_pos; + char **env; + size_t length; + + unsigned int i; + size_t j; + + /* process environment */ + generator_state = 1; + env_size = 0; + { + /* sort environment variables */ + counter = 0; + for (env = environ; (*env) != NULL; env++) + { + counter++; + env_size += strlen(*env) + 1; + } + tmp_envs = (char**) malloc(sizeof(char*) * counter); + for (i = 0; i < counter; i++) + { + tmp_envs[i] = (char*) (environ[i]); + } + qsort(tmp_envs, counter, sizeof(char*), (int(*)(const void*, const void*)) &strcmp); + + /* and copy them */ + env_data = malloc(env_size * sizeof(char)); + buffer_pos = env_data; + for (i = 0; i < counter; i++) + { + length = strlen(tmp_envs[i]) + 1; + memcpy(buffer_pos, tmp_envs[i], length); + buffer_pos += length; + } + + /* add sorted environment to random generator */ + for (j = 0; j < env_size; j++) + { + pseudorandom_algo(&generator_state, ((unsigned char*) env_data)[j]); + } + + /* free temp environment */ + free(tmp_envs); + } + + /* copy argv */ + argv_size = 0; + for (i = 0; i < argc; i++) + { + argv_size += strlen(argv[i]) + 1; + } + argv_data = malloc(sizeof(char) * argv_size); + buffer_pos = argv_data; + for (i = 0; i < argc; i++) + { + length = strlen(argv[i]) + 1; + memcpy(buffer_pos, argv[i], length); + buffer_pos += length; + } + + /* add arguments to gnerator */ + for (j = 0; j < argv_size; j++) + { + pseudorandom_algo(&generator_state, ((unsigned char*) argv_data)[j]); + } + + /* set state and free rest */ + free(env_data); + free(argv_data); + pseudorandom_state = generator_state; +}