winebuild: generate deterministic temporary filenames

Using xorshift* as pseudorandom number generator. It get initialised
with the environment (sorted by name) and arguments.

Signed-off-by: Marko Semet <marko@marko10-000.de>
feature/deterministic_pub1
Marko Semet 2020-06-23 20:25:04 +02:00
parent 60eed6ef13
commit f2d9cae0f1
3 changed files with 165 additions and 2 deletions

View File

@ -27,6 +27,7 @@
# error You must include config.h to use this header
#endif
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
@ -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;

View File

@ -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 */

View File

@ -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;
}