281 lines
6.7 KiB
C
281 lines
6.7 KiB
C
/*
|
|
* Copyright (c) 2007, Cameron Rich
|
|
*
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
* * Neither the name of the axTLS project nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
|
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
/**
|
|
* @file os_port.c
|
|
*
|
|
* OS specific functions.
|
|
*/
|
|
#include <time.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <stdarg.h>
|
|
#include "os_port.h"
|
|
#include <stdio.h>
|
|
#include "sockets.h"
|
|
|
|
static int memory_buf[400];
|
|
static char enable = 1;
|
|
static int nb_entries = 0;
|
|
static int nb_alloc = 0;
|
|
|
|
void disable_memory_buf(void)
|
|
{
|
|
enable = 0;
|
|
}
|
|
void enable_memory_buf(void)
|
|
{
|
|
enable = 1;
|
|
}
|
|
void init_memory_buf(void)
|
|
{
|
|
for(int i = 0; i < 400; i += 2)
|
|
{
|
|
memory_buf[i] = -1;
|
|
memory_buf[i+1] = 0;
|
|
}
|
|
}
|
|
void print_buf_stats(void)
|
|
{
|
|
if(enable)
|
|
{
|
|
int used = 0;
|
|
for(int i = 1; i < 400; i += 2)
|
|
used += memory_buf[i];
|
|
printf("%d\n", used);
|
|
}
|
|
}
|
|
|
|
void print_all_buf_stats(void)
|
|
{
|
|
int used = 0;
|
|
for(int i = 1; i < 400; i += 2)
|
|
used += memory_buf[i];
|
|
printf("used: %d bytes\n", used);
|
|
|
|
for(int i = 0; i < 400; i += 2)
|
|
if(memory_buf[i] != -1)
|
|
printf("ptr:%X, size:%d\n", memory_buf[i], memory_buf[i+1]);
|
|
}
|
|
|
|
static void add_entry(void *x, size_t s, const char* f, const int l)
|
|
{
|
|
nb_entries++;
|
|
for(int i = 0; i < 400; i += 2)
|
|
{
|
|
if(memory_buf[i] == -1)
|
|
{
|
|
if(enable)
|
|
printf("new ptr:%X, size:%d at %s:%d\n", x, s, f, l);
|
|
memory_buf[i] = (int)(x);
|
|
memory_buf[i+1] = s;
|
|
return;
|
|
}
|
|
}
|
|
if(enable)
|
|
printf("No space left in buffer\n");
|
|
}
|
|
|
|
static void remove_entry(void *x, const char* f, const int l)
|
|
{
|
|
nb_entries--;
|
|
for(int i = 0; i < 400; i += 2)
|
|
{
|
|
if(memory_buf[i] == (int)(x))
|
|
{
|
|
if(enable)
|
|
printf("free ptr:%X, size:%d at %s:%d\n", memory_buf[i], memory_buf[i+1], f, l);
|
|
memory_buf[i] = -1;
|
|
memory_buf[i+1] = 0;
|
|
return;
|
|
}
|
|
}
|
|
if(enable)
|
|
printf("not found\n");
|
|
}
|
|
|
|
#ifdef MBED
|
|
/**
|
|
* gettimeofday() not in mbed
|
|
*/
|
|
EXP_FUNC void STDCALL gettimeofday(struct timeval* t, void* timezone)
|
|
{
|
|
t->tv_sec = time(NULL);
|
|
t->tv_usec = 0; /* 1sec precision only */
|
|
}
|
|
|
|
#endif
|
|
|
|
#ifdef WIN32
|
|
/**
|
|
* gettimeofday() not in Win32
|
|
*/
|
|
EXP_FUNC void STDCALL gettimeofday(struct timeval* t, void* timezone)
|
|
{
|
|
#if defined(_WIN32_WCE)
|
|
t->tv_sec = time(NULL);
|
|
t->tv_usec = 0; /* 1sec precision only */
|
|
#else
|
|
struct _timeb timebuffer;
|
|
_ftime(&timebuffer);
|
|
t->tv_sec = (long)timebuffer.time;
|
|
t->tv_usec = 1000 * timebuffer.millitm; /* 1ms precision */
|
|
#endif
|
|
}
|
|
|
|
|
|
/**
|
|
* strcasecmp() not in Win32
|
|
*/
|
|
EXP_FUNC int STDCALL strcasecmp(const char *s1, const char *s2)
|
|
{
|
|
while (tolower(*s1) == tolower(*s2++))
|
|
{
|
|
if (*s1++ == '\0')
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return *(unsigned char *)s1 - *(unsigned char *)(s2 - 1);
|
|
}
|
|
|
|
|
|
EXP_FUNC int STDCALL getdomainname(char *buf, int buf_size)
|
|
{
|
|
HKEY hKey;
|
|
unsigned long datatype;
|
|
unsigned long bufferlength = buf_size;
|
|
|
|
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
|
TEXT("SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters"),
|
|
0, KEY_QUERY_VALUE, &hKey) != ERROR_SUCCESS)
|
|
return -1;
|
|
|
|
RegQueryValueEx(hKey, "Domain", NULL, &datatype, buf, &bufferlength);
|
|
RegCloseKey(hKey);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
#undef malloc
|
|
#undef realloc
|
|
#undef calloc
|
|
#undef free
|
|
|
|
static const char * out_of_mem_str = "out of memory";
|
|
static const char * file_open_str = "Could not open file \"%s\"";
|
|
|
|
/*
|
|
* Some functions that call display some error trace and then call abort().
|
|
* This just makes life much easier on embedded systems, since we're
|
|
* suffering major trauma...
|
|
*/
|
|
EXP_FUNC void * STDCALL ax_malloc(size_t s, const char* f, const int l)
|
|
{
|
|
if(enable)
|
|
printf("malloc\t");
|
|
|
|
void *x;
|
|
|
|
if ((x = malloc(s)) == NULL)
|
|
exit_now(out_of_mem_str);
|
|
add_entry(x,s, f, l);
|
|
print_buf_stats();
|
|
|
|
return x;
|
|
}
|
|
|
|
EXP_FUNC void * STDCALL ax_realloc(void *y, size_t s, const char* f, const int l)
|
|
{
|
|
if(enable)
|
|
printf("realloc\t");
|
|
|
|
void *x;
|
|
|
|
if ((x = realloc(y, s)) == NULL)
|
|
exit_now(out_of_mem_str);
|
|
remove_entry(y, f, l);
|
|
add_entry(x,s, f, l);
|
|
print_buf_stats();
|
|
return x;
|
|
}
|
|
|
|
EXP_FUNC void * STDCALL ax_calloc(size_t n, size_t s, const char* f, const int l)
|
|
{
|
|
if(enable)
|
|
printf("calloc\t");
|
|
void *x;
|
|
|
|
if ((x = calloc(n, s)) == NULL) {
|
|
exit_now(out_of_mem_str);
|
|
}
|
|
|
|
add_entry(x,n*s, f, l);
|
|
print_buf_stats();
|
|
return x;
|
|
}
|
|
|
|
EXP_FUNC void STDCALL ax_free(void *y, const char* f, const int l)
|
|
{
|
|
if(enable)
|
|
printf("free\t");
|
|
|
|
remove_entry(y, f, l);
|
|
print_buf_stats();
|
|
free(y);
|
|
}
|
|
/*
|
|
EXP_FUNC int STDCALL ax_open(const char *pathname, int flags)
|
|
{
|
|
int x;
|
|
|
|
if ((x = open(pathname, flags)) < 0)
|
|
exit_now(file_open_str, pathname);
|
|
|
|
return x;
|
|
}
|
|
*/
|
|
|
|
/**
|
|
* This is a call which will deliberately exit an application, but will
|
|
* display some information before dying.
|
|
*/
|
|
void exit_now(const char *format, ...)
|
|
{
|
|
va_list argp;
|
|
|
|
va_start(argp, format);
|
|
vfprintf(stderr, format, argp);
|
|
va_end(argp);
|
|
abort();
|
|
}
|
|
|