177 lines
4.5 KiB
C
177 lines
4.5 KiB
C
/******************************************************************************/
|
|
/* SlunkCrypt, by LoRd_MuldeR <MuldeR2@GMX.de> */
|
|
/* This work has been released under the CC0 1.0 Universal license! */
|
|
/******************************************************************************/
|
|
|
|
#ifdef _WIN32
|
|
# define WIN32_LEAN_AND_MEAN 1
|
|
# define _CRT_SECURE_NO_WARNINGS 1
|
|
#else
|
|
# define _GNU_SOURCE 1
|
|
#endif
|
|
|
|
#include "utils.h"
|
|
#include <slunkcrypt.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <signal.h>
|
|
|
|
/* Platform support */
|
|
#ifdef _WIN32
|
|
# include <Windows.h>
|
|
# include <io.h>
|
|
# include <fcntl.h>
|
|
# define STAT_T struct _stati64
|
|
# define FSTAT(X,Y) _fstati64((X),(Y))
|
|
# define FILENO(X) _fileno((X))
|
|
# define S_IFMT _S_IFMT
|
|
# define S_IFDIR _S_IFDIR
|
|
# define S_IFIFO _S_IFIFO
|
|
# ifndef _O_U8TEXT
|
|
# define _O_U8TEXT 0x40000
|
|
# endif
|
|
#else
|
|
# if defined(__USE_LARGEFILE64) && (__USE_LARGEFILE64)
|
|
# define STAT_T struct stat64
|
|
# define FSTAT(X,Y) fstat64((X),(Y))
|
|
# else
|
|
# define STAT_T struct stat
|
|
# define FSTAT(X,Y) fstat((X),(Y))
|
|
# endif
|
|
# define FILENO(X) fileno((X))
|
|
#endif
|
|
|
|
// ==========================================================================
|
|
// Terminal initialization
|
|
// ==========================================================================
|
|
|
|
/* CRT imports */
|
|
#if defined(_WIN32) && !defined(_DLL)
|
|
extern char *const _acmdln;
|
|
extern wchar_t *const _wcmdln;
|
|
#endif
|
|
|
|
void init_terminal(void)
|
|
{
|
|
#ifdef _WIN32
|
|
SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX);
|
|
_setmode(_fileno(stdin), _O_BINARY);
|
|
_setmode(_fileno(stderr), _O_U8TEXT);
|
|
#ifndef _DLL
|
|
if (_acmdln) slunkcrypt_bzero(_acmdln, strlen(_acmdln) * sizeof(char));
|
|
if (_wcmdln) slunkcrypt_bzero(_wcmdln, wcslen(_wcmdln) * sizeof(wchar_t));
|
|
#endif
|
|
#endif
|
|
}
|
|
|
|
// ==========================================================================
|
|
// Signal handling
|
|
// ==========================================================================
|
|
|
|
void setup_signal_handler(const int signo, signal_handler_t* const handler)
|
|
{
|
|
#ifdef _WIN32
|
|
signal(signo, handler);
|
|
#else
|
|
struct sigaction act;
|
|
act.sa_handler = handler;
|
|
sigemptyset(&act.sa_mask);
|
|
act.sa_flags = 0;
|
|
sigaction(signo, &act, NULL);
|
|
#endif
|
|
}
|
|
|
|
// ==========================================================================
|
|
// Character set conversion
|
|
// ==========================================================================
|
|
|
|
char* CHR_to_utf8(const CHR*const input)
|
|
{
|
|
#ifdef _WIN32
|
|
char* buffer;
|
|
DWORD buffer_size = 0U, result = 0U;
|
|
|
|
buffer_size = WideCharToMultiByte(CP_UTF8, 0, input, -1, NULL, 0, NULL, NULL);
|
|
if (buffer_size < 1U)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
buffer = (char*)malloc(sizeof(char) * buffer_size);
|
|
if (!buffer)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
result = WideCharToMultiByte(CP_UTF8, 0, input, -1, (LPSTR)buffer, buffer_size, NULL, NULL);
|
|
if ((result > 0U) && (result <= buffer_size))
|
|
{
|
|
return buffer;
|
|
}
|
|
|
|
free(buffer);
|
|
return NULL;
|
|
#else
|
|
return strdup(input);
|
|
#endif
|
|
}
|
|
|
|
// ==========================================================================
|
|
// Byte-order support
|
|
// ==========================================================================
|
|
|
|
#if defined(__BYTE_ORDER) && defined(__BIG_ENDIAN) && (__BYTE_ORDER == __BIG_ENDIAN)
|
|
# define BIG_ENDIAN_BYTE_ORDER 1
|
|
#elif defined(__BYTE_ORDER__) && defined(__ORDER_BIG_ENDIAN__) && (__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__)
|
|
# define BIG_ENDIAN_BYTE_ORDER 1
|
|
#else
|
|
# define BIG_ENDIAN_BYTE_ORDER 0
|
|
#endif
|
|
|
|
uint64_t swap_bytes_u64(const uint64_t value)
|
|
{
|
|
#if BIG_ENDIAN_BYTE_ORDER
|
|
return
|
|
(((value) >> 56) & 0x00000000000000FF) | (((value) >> 40) & 0x000000000000FF00) |
|
|
(((value) >> 24) & 0x0000000000FF0000) | (((value) >> 8) & 0x00000000FF000000) |
|
|
(((value) << 8) & 0x000000FF00000000) | (((value) << 24) & 0x0000FF0000000000) |
|
|
(((value) << 40) & 0x00FF000000000000) | (((value) << 56) & 0xFF00000000000000);
|
|
#else
|
|
return value; /*nothing to do*/
|
|
#endif
|
|
}
|
|
|
|
// ==========================================================================
|
|
// File functions
|
|
// ==========================================================================
|
|
|
|
uint64_t get_file_size(FILE* const file)
|
|
{
|
|
STAT_T stat;
|
|
if (FSTAT(FILENO(file), &stat) == 0)
|
|
{
|
|
const uint16_t file_type = stat.st_mode & S_IFMT;
|
|
if ((file_type != S_IFDIR) && (file_type != S_IFIFO))
|
|
{
|
|
const int64_t ssize = stat.st_size;
|
|
return (ssize >= 0) ? ((uint64_t)ssize) : 0U;
|
|
}
|
|
return 0U;
|
|
}
|
|
return UINT64_MAX;
|
|
}
|
|
|
|
const CHR* get_file_name(const CHR* path)
|
|
{
|
|
const CHR* ptr;
|
|
while ((ptr = STRRCHR(path, T('/'))))
|
|
{
|
|
path = ptr + 1U;
|
|
}
|
|
while ((ptr = STRRCHR(path, T('\\'))))
|
|
{
|
|
path = ptr + 1U;
|
|
}
|
|
return path;
|
|
}
|