forked from Mirrors/wine-wine
661 lines
15 KiB
C
661 lines
15 KiB
C
/*
|
|
* Utility routines
|
|
*
|
|
* Copyright 1998 Bertho A. Stultiens
|
|
* Copyright 2002 Ove Kaaven
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include "wine/port.h"
|
|
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
|
|
#include "widl.h"
|
|
#include "utils.h"
|
|
#include "parser.h"
|
|
|
|
#define CURRENT_LOCATION { input_name ? input_name : "stdin", line_number, parser_text }
|
|
|
|
static const int want_near_indication = 0;
|
|
|
|
static void make_print(char *str)
|
|
{
|
|
while(*str)
|
|
{
|
|
if(!isprint(*str))
|
|
*str = ' ';
|
|
str++;
|
|
}
|
|
}
|
|
|
|
static void generic_msg(const loc_info_t *loc_info, const char *s, const char *t, va_list ap)
|
|
{
|
|
fprintf(stderr, "%s:%d: %s: ", loc_info->input_name, loc_info->line_number, t);
|
|
vfprintf(stderr, s, ap);
|
|
|
|
if (want_near_indication)
|
|
{
|
|
char *cpy;
|
|
if(loc_info->near_text)
|
|
{
|
|
cpy = xstrdup(loc_info->near_text);
|
|
make_print(cpy);
|
|
fprintf(stderr, " near '%s'", cpy);
|
|
free(cpy);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void error_loc(const char *s, ...)
|
|
{
|
|
loc_info_t cur_loc = CURRENT_LOCATION;
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
generic_msg(&cur_loc, s, "error", ap);
|
|
va_end(ap);
|
|
exit(1);
|
|
}
|
|
|
|
/* yyerror: yacc assumes this is not newline terminated. */
|
|
void parser_error(const char *s)
|
|
{
|
|
error_loc("%s\n", s);
|
|
}
|
|
|
|
void error_loc_info(const loc_info_t *loc_info, const char *s, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
generic_msg(loc_info, s, "error", ap);
|
|
va_end(ap);
|
|
exit(1);
|
|
}
|
|
|
|
int parser_warning(const char *s, ...)
|
|
{
|
|
loc_info_t cur_loc = CURRENT_LOCATION;
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
generic_msg(&cur_loc, s, "warning", ap);
|
|
va_end(ap);
|
|
return 0;
|
|
}
|
|
|
|
void error(const char *s, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
fprintf(stderr, "error: ");
|
|
vfprintf(stderr, s, ap);
|
|
va_end(ap);
|
|
exit(2);
|
|
}
|
|
|
|
void warning(const char *s, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
fprintf(stderr, "warning: ");
|
|
vfprintf(stderr, s, ap);
|
|
va_end(ap);
|
|
}
|
|
|
|
void warning_loc_info(const loc_info_t *loc_info, const char *s, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
generic_msg(loc_info, s, "warning", ap);
|
|
va_end(ap);
|
|
}
|
|
|
|
void chat(const char *s, ...)
|
|
{
|
|
if(debuglevel & DEBUGLEVEL_CHAT)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, s);
|
|
fprintf(stderr, "chat: ");
|
|
vfprintf(stderr, s, ap);
|
|
va_end(ap);
|
|
}
|
|
}
|
|
|
|
char *dup_basename(const char *name, const char *ext)
|
|
{
|
|
int namelen;
|
|
int extlen = strlen(ext);
|
|
char *base;
|
|
char *slash;
|
|
|
|
if(!name)
|
|
name = "widl.tab";
|
|
|
|
slash = strrchr(name, '/');
|
|
if (!slash)
|
|
slash = strrchr(name, '\\');
|
|
|
|
if (slash)
|
|
name = slash + 1;
|
|
|
|
namelen = strlen(name);
|
|
|
|
/* +6 for later extension (strlen("_r.rgs")) and +1 for '\0' */
|
|
base = xmalloc(namelen +6 +1);
|
|
strcpy(base, name);
|
|
if(!strcasecmp(name + namelen-extlen, ext))
|
|
{
|
|
base[namelen - extlen] = '\0';
|
|
}
|
|
return base;
|
|
}
|
|
|
|
size_t widl_getline(char **linep, size_t *lenp, FILE *fp)
|
|
{
|
|
char *line = *linep;
|
|
size_t len = *lenp;
|
|
size_t n = 0;
|
|
|
|
if (!line)
|
|
{
|
|
len = 64;
|
|
line = xmalloc(len);
|
|
}
|
|
|
|
while (fgets(&line[n], len - n, fp))
|
|
{
|
|
n += strlen(&line[n]);
|
|
if (line[n - 1] == '\n')
|
|
break;
|
|
else if (n == len - 1)
|
|
{
|
|
len *= 2;
|
|
line = xrealloc(line, len);
|
|
}
|
|
}
|
|
|
|
*linep = line;
|
|
*lenp = len;
|
|
return n;
|
|
}
|
|
|
|
void *xmalloc(size_t size)
|
|
{
|
|
void *res;
|
|
|
|
assert(size > 0);
|
|
res = malloc(size);
|
|
if(res == NULL)
|
|
{
|
|
error("Virtual memory exhausted.\n");
|
|
}
|
|
memset(res, 0x55, size);
|
|
return res;
|
|
}
|
|
|
|
|
|
void *xrealloc(void *p, size_t size)
|
|
{
|
|
void *res;
|
|
|
|
assert(size > 0);
|
|
res = realloc(p, size);
|
|
if(res == NULL)
|
|
{
|
|
error("Virtual memory exhausted.\n");
|
|
}
|
|
return res;
|
|
}
|
|
|
|
char *strmake( const char* fmt, ... )
|
|
{
|
|
int n;
|
|
size_t size = 100;
|
|
va_list ap;
|
|
|
|
for (;;)
|
|
{
|
|
char *p = xmalloc( size );
|
|
va_start( ap, fmt );
|
|
n = vsnprintf( p, size, fmt, ap );
|
|
va_end( ap );
|
|
if (n == -1) size *= 2;
|
|
else if ((size_t)n >= size) size = n + 1;
|
|
else return p;
|
|
free( p );
|
|
}
|
|
}
|
|
|
|
char *xstrdup(const char *str)
|
|
{
|
|
char *s;
|
|
|
|
assert(str != NULL);
|
|
s = xmalloc(strlen(str)+1);
|
|
return strcpy(s, str);
|
|
}
|
|
|
|
int strendswith(const char* str, const char* end)
|
|
{
|
|
int l = strlen(str);
|
|
int m = strlen(end);
|
|
return l >= m && strcmp(str + l - m, end) == 0;
|
|
}
|
|
|
|
/*******************************************************************
|
|
* buffer management
|
|
*
|
|
* Function for writing to a memory buffer.
|
|
*/
|
|
|
|
int byte_swapped = 0;
|
|
unsigned char *output_buffer;
|
|
size_t output_buffer_pos;
|
|
size_t output_buffer_size;
|
|
|
|
static struct resource
|
|
{
|
|
unsigned char *data;
|
|
size_t size;
|
|
} resources[16];
|
|
static unsigned int nb_resources;
|
|
|
|
static void check_output_buffer_space( size_t size )
|
|
{
|
|
if (output_buffer_pos + size >= output_buffer_size)
|
|
{
|
|
output_buffer_size = max( output_buffer_size * 2, output_buffer_pos + size );
|
|
output_buffer = xrealloc( output_buffer, output_buffer_size );
|
|
}
|
|
}
|
|
|
|
void init_output_buffer(void)
|
|
{
|
|
output_buffer_size = 1024;
|
|
output_buffer_pos = 0;
|
|
output_buffer = xmalloc( output_buffer_size );
|
|
}
|
|
|
|
void flush_output_buffer( const char *name )
|
|
{
|
|
int fd = open( name, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666 );
|
|
if (fd == -1) error( "Error creating %s\n", name );
|
|
if (write( fd, output_buffer, output_buffer_pos ) != output_buffer_pos)
|
|
error( "Error writing to %s\n", name );
|
|
close( fd );
|
|
free( output_buffer );
|
|
}
|
|
|
|
static inline void put_resource_id( const char *str )
|
|
{
|
|
if (str[0] != '#')
|
|
{
|
|
while (*str)
|
|
{
|
|
unsigned char ch = *str++;
|
|
put_word( toupper(ch) );
|
|
}
|
|
put_word( 0 );
|
|
}
|
|
else
|
|
{
|
|
put_word( 0xffff );
|
|
put_word( atoi( str + 1 ));
|
|
}
|
|
}
|
|
|
|
void add_output_to_resources( const char *type, const char *name )
|
|
{
|
|
size_t data_size = output_buffer_pos;
|
|
size_t header_size = 5 * sizeof(unsigned int) + 2 * sizeof(unsigned short);
|
|
|
|
assert( nb_resources < ARRAY_SIZE( resources ));
|
|
|
|
if (type[0] != '#') header_size += (strlen( type ) + 1) * sizeof(unsigned short);
|
|
else header_size += 2 * sizeof(unsigned short);
|
|
if (name[0] != '#') header_size += (strlen( name ) + 1) * sizeof(unsigned short);
|
|
else header_size += 2 * sizeof(unsigned short);
|
|
|
|
header_size = (header_size + 3) & ~3;
|
|
align_output( 4 );
|
|
check_output_buffer_space( header_size );
|
|
resources[nb_resources].size = header_size + output_buffer_pos;
|
|
memmove( output_buffer + header_size, output_buffer, output_buffer_pos );
|
|
|
|
output_buffer_pos = 0;
|
|
put_dword( data_size ); /* ResSize */
|
|
put_dword( header_size ); /* HeaderSize */
|
|
put_resource_id( type ); /* ResType */
|
|
put_resource_id( name ); /* ResName */
|
|
align_output( 4 );
|
|
put_dword( 0 ); /* DataVersion */
|
|
put_word( 0 ); /* Memory options */
|
|
put_word( 0 ); /* Language */
|
|
put_dword( 0 ); /* Version */
|
|
put_dword( 0 ); /* Characteristics */
|
|
|
|
resources[nb_resources++].data = output_buffer;
|
|
init_output_buffer();
|
|
}
|
|
|
|
void flush_output_resources( const char *name )
|
|
{
|
|
int fd;
|
|
unsigned int i;
|
|
|
|
/* all output must have been saved with add_output_to_resources() first */
|
|
assert( !output_buffer_pos );
|
|
|
|
put_dword( 0 ); /* ResSize */
|
|
put_dword( 32 ); /* HeaderSize */
|
|
put_word( 0xffff ); /* ResType */
|
|
put_word( 0x0000 );
|
|
put_word( 0xffff ); /* ResName */
|
|
put_word( 0x0000 );
|
|
put_dword( 0 ); /* DataVersion */
|
|
put_word( 0 ); /* Memory options */
|
|
put_word( 0 ); /* Language */
|
|
put_dword( 0 ); /* Version */
|
|
put_dword( 0 ); /* Characteristics */
|
|
|
|
fd = open( name, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666 );
|
|
if (fd == -1) error( "Error creating %s\n", name );
|
|
if (write( fd, output_buffer, output_buffer_pos ) != output_buffer_pos)
|
|
error( "Error writing to %s\n", name );
|
|
for (i = 0; i < nb_resources; i++)
|
|
{
|
|
if (write( fd, resources[i].data, resources[i].size ) != resources[i].size)
|
|
error( "Error writing to %s\n", name );
|
|
free( resources[i].data );
|
|
}
|
|
close( fd );
|
|
nb_resources = 0;
|
|
free( output_buffer );
|
|
}
|
|
|
|
void put_data( const void *data, size_t size )
|
|
{
|
|
check_output_buffer_space( size );
|
|
memcpy( output_buffer + output_buffer_pos, data, size );
|
|
output_buffer_pos += size;
|
|
}
|
|
|
|
void put_byte( unsigned char val )
|
|
{
|
|
check_output_buffer_space( 1 );
|
|
output_buffer[output_buffer_pos++] = val;
|
|
}
|
|
|
|
void put_word( unsigned short val )
|
|
{
|
|
if (byte_swapped) val = (val << 8) | (val >> 8);
|
|
put_data( &val, sizeof(val) );
|
|
}
|
|
|
|
void put_dword( unsigned int val )
|
|
{
|
|
if (byte_swapped)
|
|
val = ((val << 24) | ((val << 8) & 0x00ff0000) | ((val >> 8) & 0x0000ff00) | (val >> 24));
|
|
put_data( &val, sizeof(val) );
|
|
}
|
|
|
|
void put_qword( unsigned int val )
|
|
{
|
|
if (byte_swapped)
|
|
{
|
|
put_dword( 0 );
|
|
put_dword( val );
|
|
}
|
|
else
|
|
{
|
|
put_dword( val );
|
|
put_dword( 0 );
|
|
}
|
|
}
|
|
|
|
/* pointer-sized word */
|
|
void put_pword( unsigned int val )
|
|
{
|
|
if (pointer_size == 8) put_qword( val );
|
|
else put_dword( val );
|
|
}
|
|
|
|
void put_str( int indent, const char *format, ... )
|
|
{
|
|
int n;
|
|
va_list args;
|
|
|
|
check_output_buffer_space( 4 * indent );
|
|
memset( output_buffer + output_buffer_pos, ' ', 4 * indent );
|
|
output_buffer_pos += 4 * indent;
|
|
|
|
for (;;)
|
|
{
|
|
size_t size = output_buffer_size - output_buffer_pos;
|
|
va_start( args, format );
|
|
n = vsnprintf( (char *)output_buffer + output_buffer_pos, size, format, args );
|
|
va_end( args );
|
|
if (n == -1) size *= 2;
|
|
else if ((size_t)n >= size) size = n + 1;
|
|
else
|
|
{
|
|
output_buffer_pos += n;
|
|
return;
|
|
}
|
|
check_output_buffer_space( size );
|
|
}
|
|
}
|
|
|
|
void align_output( unsigned int align )
|
|
{
|
|
size_t size = align - (output_buffer_pos % align);
|
|
|
|
if (size == align) return;
|
|
check_output_buffer_space( size );
|
|
memset( output_buffer + output_buffer_pos, 0, size );
|
|
output_buffer_pos += size;
|
|
}
|
|
|
|
/* xorshift* random generator. Source: https://en.wikipedia.org/wiki/Xorshift#xorshift* */
|
|
static uint64_t pseudorandom_state = 1;
|
|
static const char* pseudorandom_source = NULL;
|
|
static size_t pseudorandom_source_size = 0;
|
|
|
|
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);
|
|
|
|
/* add deebug info */
|
|
if ((result != -1) && (getenv("SOURCE_RANDOM_DEBUG") != NULL))
|
|
{
|
|
debug_filename = strmake("%s.random_debug", file);
|
|
debug_file = fopen(debug_filename, "wb");
|
|
fwrite(pseudorandom_source, 1, pseudorandom_source_size, debug_file);
|
|
fclose(debug_file);
|
|
free(debug_filename);
|
|
}
|
|
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]);
|
|
}
|
|
|
|
/* add debug if required */
|
|
if (getenv("SOURCE_RANDOM_DEBUG") != NULL)
|
|
{
|
|
/* gen field */
|
|
pseudorandom_source_size = env_size + argv_size + sizeof(generator_state);
|
|
buffer_pos = malloc(sizeof(char) * pseudorandom_source_size);
|
|
pseudorandom_source = buffer_pos;
|
|
|
|
/* copy data into it */
|
|
memcpy(buffer_pos, env_data, env_size);
|
|
buffer_pos += env_size;
|
|
memcpy(buffer_pos, argv_data, argv_size);
|
|
buffer_pos += argv_size;
|
|
memcpy(buffer_pos, &generator_state, sizeof(generator_state));
|
|
}
|
|
|
|
/* set state and free rest */
|
|
free(env_data);
|
|
free(argv_data);
|
|
pseudorandom_state = generator_state;
|
|
}
|