mirror of
https://github.com/cuberite/libdeflate.git
synced 2025-08-04 02:06:31 -04:00
7ba3155c17
Avoid confusion with the GNU extension 'program_invocation_name', which is described by 'man 3 program_invocation_name'. The GNU version isn't supposed to be exposed without defining _GNU_SOURCE, which we don't in any of the relevant files, but it's best to avoid any confusion.
177 lines
4.6 KiB
C
177 lines
4.6 KiB
C
/*
|
|
* prog_util.h - utility functions for programs
|
|
*
|
|
* Copyright 2016 Eric Biggers
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person
|
|
* obtaining a copy of this software and associated documentation
|
|
* files (the "Software"), to deal in the Software without
|
|
* restriction, including without limitation the rights to use,
|
|
* copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following
|
|
* conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
* OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#ifndef PROGRAMS_PROG_UTIL_H
|
|
#define PROGRAMS_PROG_UTIL_H
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include "libdeflate.h"
|
|
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "../common/common_defs.h"
|
|
|
|
#ifdef __GNUC__
|
|
# define _printf(str_idx, args_idx) \
|
|
__attribute__((format(printf, str_idx, args_idx)))
|
|
#else
|
|
# define _printf(str_idx, args_idx)
|
|
#endif
|
|
|
|
#ifdef _MSC_VER
|
|
/*
|
|
* Old versions (e.g. VS2010) of MSC have stdint.h but not the C99 header
|
|
* inttypes.h. Work around this by defining the PRI* macros ourselves.
|
|
*/
|
|
# define PRIu8 "hhu"
|
|
# define PRIu16 "hu"
|
|
# define PRIu32 "u"
|
|
# define PRIu64 "llu"
|
|
# define PRIi8 "hhi"
|
|
# define PRIi16 "hi"
|
|
# define PRIi32 "i"
|
|
# define PRIi64 "lli"
|
|
# define PRIx8 "hhx"
|
|
# define PRIx16 "hx"
|
|
# define PRIx32 "x"
|
|
# define PRIx64 "llx"
|
|
#else
|
|
# include <inttypes.h>
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
|
|
/*
|
|
* Definitions for Windows builds. Mainly, 'tchar' is defined to be the 2-byte
|
|
* 'wchar_t' type instead of 'char'. This is the only "easy" way I know of to
|
|
* get full Unicode support on Windows...
|
|
*/
|
|
|
|
#include <wchar.h>
|
|
int wmain(int argc, wchar_t **argv);
|
|
# define tmain wmain
|
|
# define tchar wchar_t
|
|
# define _T(text) L##text
|
|
# define T(text) _T(text)
|
|
# define TS "ls"
|
|
# define TC "lc"
|
|
# define tmemcpy wmemcpy
|
|
# define topen _wopen
|
|
# define tstrchr wcschr
|
|
# define tstrcmp wcscmp
|
|
# define tstrlen wcslen
|
|
# define tstrrchr wcsrchr
|
|
# define tstrtoul wcstoul
|
|
# define tstrxcmp wcsicmp
|
|
# define tunlink _wunlink
|
|
# define tutimbuf __utimbuf64
|
|
# define tutime _wutime64
|
|
# define tstat _wstat64
|
|
# define tfstat _fstat64
|
|
# define stat_t struct _stat64
|
|
# ifdef _MSC_VER
|
|
# define STDIN_FILENO 0
|
|
# define STDOUT_FILENO 1
|
|
# define STDERR_FILENO 2
|
|
# define S_ISREG(m) (((m) & S_IFMT) == S_IFREG)
|
|
# define S_ISDIR(m) (((m) & S_IFMT) == S_IFDIR)
|
|
# endif
|
|
|
|
#else /* _WIN32 */
|
|
|
|
/* Standard definitions for everyone else */
|
|
|
|
# define tmain main
|
|
# define tchar char
|
|
# define T(text) text
|
|
# define TS "s"
|
|
# define TC "c"
|
|
# define tmemcpy memcpy
|
|
# define topen open
|
|
# define tstrchr strchr
|
|
# define tstrcmp strcmp
|
|
# define tstrlen strlen
|
|
# define tstrrchr strrchr
|
|
# define tstrtoul strtoul
|
|
# define tstrxcmp strcmp
|
|
# define tunlink unlink
|
|
# define tutimbuf utimbuf
|
|
# define tutime utime
|
|
# define tstat stat
|
|
# define tfstat fstat
|
|
# define stat_t struct stat
|
|
|
|
#endif /* !_WIN32 */
|
|
|
|
extern const tchar *prog_invocation_name;
|
|
|
|
void _printf(1, 2) msg(const char *fmt, ...);
|
|
void _printf(1, 2) msg_errno(const char *fmt, ...);
|
|
|
|
void *xmalloc(size_t size);
|
|
|
|
void begin_program(tchar *argv[]);
|
|
|
|
struct file_stream {
|
|
int fd;
|
|
tchar *name;
|
|
bool is_standard_stream;
|
|
void *mmap_token;
|
|
void *mmap_mem;
|
|
size_t mmap_size;
|
|
};
|
|
|
|
int xopen_for_read(const tchar *path, bool symlink_ok,
|
|
struct file_stream *strm);
|
|
int xopen_for_write(const tchar *path, bool force, struct file_stream *strm);
|
|
int map_file_contents(struct file_stream *strm, u64 size);
|
|
|
|
ssize_t xread(struct file_stream *strm, void *buf, size_t count);
|
|
int full_write(struct file_stream *strm, const void *buf, size_t count);
|
|
|
|
int xclose(struct file_stream *strm);
|
|
|
|
int parse_compression_level(tchar opt_char, const tchar *arg);
|
|
|
|
struct libdeflate_compressor *alloc_compressor(int level);
|
|
struct libdeflate_decompressor *alloc_decompressor(void);
|
|
|
|
/* tgetopt.c */
|
|
|
|
extern tchar *toptarg;
|
|
extern int toptind, topterr, toptopt;
|
|
|
|
int tgetopt(int argc, tchar *argv[], const tchar *optstring);
|
|
|
|
#endif /* PROGRAMS_PROG_UTIL_H */
|