You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
410 lines
15 KiB
410 lines
15 KiB
/* |
|
* Copyright 2014 Con Kolivas |
|
* |
|
* This program is free software; you can redistribute it and/or modify it |
|
* under the terms of the GNU General Public License as published by the Free |
|
* Software Foundation; either version 3 of the License, or (at your option) |
|
* any later version. See COPYING for more details. |
|
*/ |
|
|
|
/* This file should contain all exported functions of libckpool */ |
|
|
|
#ifndef LIBCKPOOL_H |
|
#define LIBCKPOOL_H |
|
|
|
#include <errno.h> |
|
#include <jansson.h> |
|
#include <pthread.h> |
|
#include <signal.h> |
|
#include <stdbool.h> |
|
#include <stdint.h> |
|
#include <stdio.h> |
|
#include <string.h> |
|
#include <syslog.h> |
|
|
|
#if HAVE_BYTESWAP_H |
|
# include <byteswap.h> |
|
#endif |
|
|
|
#if HAVE_ENDIAN_H |
|
# include <endian.h> |
|
#elif HAVE_SYS_ENDIAN_H |
|
# include <sys/endian.h> |
|
#endif |
|
|
|
#ifndef bswap_16 |
|
#define bswap_16 __builtin_bswap16 |
|
#define bswap_32 __builtin_bswap32 |
|
#define bswap_64 __builtin_bswap64 |
|
#endif |
|
|
|
/* This assumes htobe32 is a macro in endian.h, and if it doesn't exist, then |
|
* htobe64 also won't exist */ |
|
#ifndef htobe32 |
|
# if __BYTE_ORDER == __LITTLE_ENDIAN |
|
# define htole16(x) (x) |
|
# define le16toh(x) (x) |
|
# define htole32(x) (x) |
|
# define htole64(x) (x) |
|
# define le32toh(x) (x) |
|
# define le64toh(x) (x) |
|
# define be32toh(x) bswap_32(x) |
|
# define be64toh(x) bswap_64(x) |
|
# define htobe16(x) bswap_16(x) |
|
# define htobe32(x) bswap_32(x) |
|
# define htobe64(x) bswap_64(x) |
|
# elif __BYTE_ORDER == __BIG_ENDIAN |
|
# define htole16(x) bswap_16(x) |
|
# define le16toh(x) bswap_16(x) |
|
# define htole32(x) bswap_32(x) |
|
# define le32toh(x) bswap_32(x) |
|
# define le64toh(x) bswap_64(x) |
|
# define htole64(x) bswap_64(x) |
|
# define be32toh(x) (x) |
|
# define be64toh(x) (x) |
|
# define htobe16(x) (x) |
|
# define htobe32(x) (x) |
|
# define htobe64(x) (x) |
|
# endif |
|
#endif |
|
|
|
#define unlikely(expr) (__builtin_expect(!!(expr), 0)) |
|
#define likely(expr) (__builtin_expect(!!(expr), 1)) |
|
#define __maybe_unused __attribute__((unused)) |
|
#define uninitialised_var(x) x = x |
|
|
|
typedef unsigned char uchar; |
|
|
|
typedef struct timeval tv_t; |
|
typedef struct timespec ts_t; |
|
|
|
static inline void swap_256(void *dest_p, const void *src_p) |
|
{ |
|
uint32_t *dest = dest_p; |
|
const uint32_t *src = src_p; |
|
|
|
dest[0] = src[7]; |
|
dest[1] = src[6]; |
|
dest[2] = src[5]; |
|
dest[3] = src[4]; |
|
dest[4] = src[3]; |
|
dest[5] = src[2]; |
|
dest[6] = src[1]; |
|
dest[7] = src[0]; |
|
} |
|
|
|
static inline void bswap_256(void *dest_p, const void *src_p) |
|
{ |
|
uint32_t *dest = dest_p; |
|
const uint32_t *src = src_p; |
|
|
|
dest[0] = bswap_32(src[7]); |
|
dest[1] = bswap_32(src[6]); |
|
dest[2] = bswap_32(src[5]); |
|
dest[3] = bswap_32(src[4]); |
|
dest[4] = bswap_32(src[3]); |
|
dest[5] = bswap_32(src[2]); |
|
dest[6] = bswap_32(src[1]); |
|
dest[7] = bswap_32(src[0]); |
|
} |
|
|
|
static inline void flip_32(void *dest_p, const void *src_p) |
|
{ |
|
uint32_t *dest = dest_p; |
|
const uint32_t *src = src_p; |
|
int i; |
|
|
|
for (i = 0; i < 8; i++) |
|
dest[i] = bswap_32(src[i]); |
|
} |
|
|
|
static inline void flip_80(void *dest_p, const void *src_p) |
|
{ |
|
uint32_t *dest = dest_p; |
|
const uint32_t *src = src_p; |
|
int i; |
|
|
|
for (i = 0; i < 20; i++) |
|
dest[i] = bswap_32(src[i]); |
|
} |
|
|
|
#define mutex_lock(_lock) _mutex_lock(_lock, __FILE__, __func__, __LINE__) |
|
#define mutex_unlock_noyield(_lock) _mutex_unlock_noyield(_lock, __FILE__, __func__, __LINE__) |
|
#define mutex_unlock(_lock) _mutex_unlock(_lock, __FILE__, __func__, __LINE__) |
|
#define mutex_trylock(_lock) _mutex_trylock(_lock, __FILE__, __func__, __LINE__) |
|
#define wr_lock(_lock) _wr_lock(_lock, __FILE__, __func__, __LINE__) |
|
#define wr_trylock(_lock) _wr_trylock(_lock, __FILE__, __func__, __LINE__) |
|
#define rd_lock(_lock) _rd_lock(_lock, __FILE__, __func__, __LINE__) |
|
#define rw_unlock(_lock) _rw_unlock(_lock, __FILE__, __func__, __LINE__) |
|
#define rd_unlock_noyield(_lock) _rd_unlock_noyield(_lock, __FILE__, __func__, __LINE__) |
|
#define wr_unlock_noyield(_lock) _wr_unlock_noyield(_lock, __FILE__, __func__, __LINE__) |
|
#define rd_unlock(_lock) _rd_unlock(_lock, __FILE__, __func__, __LINE__) |
|
#define wr_unlock(_lock) _wr_unlock(_lock, __FILE__, __func__, __LINE__) |
|
#define mutex_init(_lock) _mutex_init(_lock, __FILE__, __func__, __LINE__) |
|
#define rwlock_init(_lock) _rwlock_init(_lock, __FILE__, __func__, __LINE__) |
|
#define cond_init(_cond) _cond_init(_cond, __FILE__, __func__, __LINE__) |
|
|
|
#define cklock_init(_lock) _cklock_init(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_rlock(_lock) _ck_rlock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_ilock(_lock) _ck_ilock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_uilock(_lock) _ck_uilock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_ulock(_lock) _ck_ulock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_wlock(_lock) _ck_wlock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_dwlock(_lock) _ck_dwlock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_dwilock(_lock) _ck_dwilock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_dlock(_lock) _ck_dlock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_runlock(_lock) _ck_runlock(_lock, __FILE__, __func__, __LINE__) |
|
#define ck_wunlock(_lock) _ck_wunlock(_lock, __FILE__, __func__, __LINE__) |
|
|
|
#define ckalloc(len) _ckalloc(len, __FILE__, __func__, __LINE__) |
|
#define ckzalloc(len) _ckzalloc(len, __FILE__, __func__, __LINE__) |
|
|
|
#define dealloc(ptr) _dealloc((void *)&(ptr)) |
|
|
|
#define VASPRINTF(strp, fmt, ...) do { \ |
|
if (unlikely(vasprintf(strp, fmt, ##__VA_ARGS__) < 0)) \ |
|
quitfrom(1, __FILE__, __func__, __LINE__, "Failed to asprintf"); \ |
|
} while (0) |
|
|
|
#define ASPRINTF(strp, fmt, ...) do { \ |
|
if (unlikely(asprintf(strp, fmt, ##__VA_ARGS__) < 0)) \ |
|
quitfrom(1, __FILE__, __func__, __LINE__, "Failed to asprintf"); \ |
|
} while (0) |
|
|
|
void logmsg(int loglevel, const char *fmt, ...); |
|
|
|
#define LOGEMERG(fmt, ...) logmsg(LOG_EMERG, fmt, ##__VA_ARGS__) |
|
#define LOGALERT(fmt, ...) logmsg(LOG_ALERT, fmt, ##__VA_ARGS__) |
|
#define LOGCRIT(fmt, ...) logmsg(LOG_CRIT, fmt, ##__VA_ARGS__) |
|
#define LOGERR(fmt, ...) logmsg(LOG_ERR, fmt, ##__VA_ARGS__) |
|
#define LOGWARNING(fmt, ...) logmsg(LOG_WARNING, fmt, ##__VA_ARGS__) |
|
#define LOGNOTICE(fmt, ...) logmsg(LOG_NOTICE, fmt, ##__VA_ARGS__) |
|
#define LOGINFO(fmt, ...) logmsg(LOG_INFO, fmt, ##__VA_ARGS__) |
|
#define LOGDEBUG(fmt, ...) logmsg(LOG_DEBUG, fmt, ##__VA_ARGS__) |
|
|
|
#define IN_FMT_FFL " in %s %s():%d" |
|
#define quitfrom(status, _file, _func, _line, fmt, ...) do { \ |
|
if (fmt) { \ |
|
fprintf(stderr, fmt IN_FMT_FFL, ##__VA_ARGS__, _file, _func, _line); \ |
|
fprintf(stderr, "\n"); \ |
|
fflush(stderr); \ |
|
} \ |
|
exit(status); \ |
|
} while (0) |
|
|
|
#define quit(status, fmt, ...) do { \ |
|
if (fmt) { \ |
|
fprintf(stderr, fmt, ##__VA_ARGS__); \ |
|
fprintf(stderr, "\n"); \ |
|
fflush(stderr); \ |
|
} \ |
|
exit(status); \ |
|
} while (0) |
|
|
|
#define PAGESIZE (4096) |
|
|
|
/* ck locks, a write biased variant of rwlocks */ |
|
struct cklock { |
|
pthread_mutex_t mutex; |
|
pthread_rwlock_t rwlock; |
|
}; |
|
|
|
typedef struct cklock cklock_t; |
|
|
|
struct unixsock { |
|
int sockd; |
|
char *path; |
|
}; |
|
|
|
typedef struct unixsock unixsock_t; |
|
|
|
typedef struct proc_instance proc_instance_t; |
|
|
|
/* No error checking with these, make sure we know they're valid already! */ |
|
static inline void json_strcpy(char *buf, json_t *val, const char *key) |
|
{ |
|
strcpy(buf, json_string_value(json_object_get(val, key))); |
|
} |
|
|
|
static inline void json_dblcpy(double *dbl, json_t *val, const char *key) |
|
{ |
|
*dbl = json_real_value(json_object_get(val, key)); |
|
} |
|
|
|
static inline void json_uintcpy(uint32_t *u32, json_t *val, const char *key) |
|
{ |
|
*u32 = (uint32_t)json_integer_value(json_object_get(val, key)); |
|
} |
|
|
|
static inline void json_uint64cpy(uint64_t *u64, json_t *val, const char *key) |
|
{ |
|
*u64 = (uint64_t)json_integer_value(json_object_get(val, key)); |
|
} |
|
|
|
static inline void json_int64cpy(int64_t *i64, json_t *val, const char *key) |
|
{ |
|
*i64 = (int64_t)json_integer_value(json_object_get(val, key)); |
|
} |
|
|
|
static inline void json_intcpy(int *i, json_t *val, const char *key) |
|
{ |
|
*i = json_integer_value(json_object_get(val, key)); |
|
} |
|
|
|
static inline void json_strdup(char **buf, json_t *val, const char *key) |
|
{ |
|
*buf = strdup(json_string_value(json_object_get(val, key))); |
|
} |
|
|
|
static inline void json_set_string(json_t *val, const char *key, const char *str) |
|
{ |
|
json_object_set_new_nocheck(val, key, json_string(str)); |
|
} |
|
|
|
static inline void json_set_int(json_t *val, const char *key, int64_t integer) |
|
{ |
|
json_object_set_new_nocheck(val, key, json_integer(integer)); |
|
} |
|
|
|
static inline void json_set_double(json_t *val, const char *key, double real) |
|
{ |
|
json_object_set_new_nocheck(val, key, json_real(real)); |
|
} |
|
|
|
static inline void json_set_bool(json_t *val, const char *key, bool boolean) |
|
{ |
|
json_object_set_new_nocheck(val, key, json_boolean(boolean)); |
|
} |
|
|
|
void rename_proc(const char *name); |
|
void create_pthread(pthread_t *thread, void *(*start_routine)(void *), void *arg); |
|
void join_pthread(pthread_t thread); |
|
|
|
void _mutex_lock(pthread_mutex_t *lock, const char *file, const char *func, const int line); |
|
void _mutex_unlock_noyield(pthread_mutex_t *lock, const char *file, const char *func, const int line); |
|
void _mutex_unlock(pthread_mutex_t *lock, const char *file, const char *func, const int line); |
|
int _mutex_trylock(pthread_mutex_t *lock, __maybe_unused const char *file, __maybe_unused const char *func, __maybe_unused const int line); |
|
void _wr_lock(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
int _wr_trylock(pthread_rwlock_t *lock, __maybe_unused const char *file, __maybe_unused const char *func, __maybe_unused const int line); |
|
void _rd_lock(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _rw_unlock(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _rd_unlock_noyield(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _wr_unlock_noyield(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _rd_unlock(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _wr_unlock(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void _mutex_init(pthread_mutex_t *lock, const char *file, const char *func, const int line); |
|
void mutex_destroy(pthread_mutex_t *lock); |
|
void _rwlock_init(pthread_rwlock_t *lock, const char *file, const char *func, const int line); |
|
void rwlock_destroy(pthread_rwlock_t *lock); |
|
void _cond_init(pthread_cond_t *cond, const char *file, const char *func, const int line); |
|
|
|
void _cklock_init(cklock_t *lock, const char *file, const char *func, const int line); |
|
void cklock_destroy(cklock_t *lock); |
|
void _ck_rlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_ilock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_uilock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_ulock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_wlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_dwlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_dwilock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_dlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_runlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
void _ck_wunlock(cklock_t *lock, const char *file, const char *func, const int line); |
|
|
|
static inline bool sock_connecting(void) |
|
{ |
|
return errno == EINPROGRESS; |
|
} |
|
|
|
static inline bool sock_blocks(void) |
|
{ |
|
return (errno == EAGAIN || errno == EWOULDBLOCK); |
|
} |
|
static inline bool sock_timeout(void) |
|
{ |
|
return (errno == ETIMEDOUT); |
|
} |
|
static inline bool interrupted(void) |
|
{ |
|
return (errno == EINTR); |
|
} |
|
|
|
bool extract_sockaddr(char *url, char **sockaddr_url, char **sockaddr_port); |
|
void keep_sockalive(int fd); |
|
void noblock_socket(int fd); |
|
void block_socket(int fd); |
|
int bind_socket(char *url, char *port); |
|
int connect_socket(char *url, char *port); |
|
int write_socket(int fd, const void *buf, size_t nbyte); |
|
void empty_socket(int fd); |
|
void close_unix_socket(const int sockd, const char *server_path); |
|
int _open_unix_server(const char *server_path, const char *file, const char *func, const int line); |
|
#define open_unix_server(server_path) _open_unix_server(server_path, __FILE__, __func__, __LINE__) |
|
int _open_unix_client(const char *server_path, const char *file, const char *func, const int line); |
|
#define open_unix_client(server_path) _open_unix_client(server_path, __FILE__, __func__, __LINE__) |
|
int wait_read_select(int sockd, int timeout); |
|
int read_length(int sockd, void *buf, int len); |
|
char *_recv_unix_msg(int sockd, const char *file, const char *func, const int line); |
|
#define recv_unix_msg(sockd) _recv_unix_msg(sockd, __FILE__, __func__, __LINE__) |
|
int wait_write_select(int sockd, int timeout); |
|
int write_length(int sockd, const void *buf, int len); |
|
bool _send_unix_msg(int sockd, const char *buf, const char *file, const char *func, const int line); |
|
#define send_unix_msg(sockd, buf) _send_unix_msg(sockd, buf, __FILE__, __func__, __LINE__) |
|
bool _send_unix_data(int sockd, void *buf, uint32_t len, const char *file, const char *func, const int line); |
|
#define send_unix_data(sockd, buf, len) _send_unix_data(sockd, buf, len, __FILE__, __func__, __LINE__) |
|
|
|
const char *__json_array_string(json_t *val, unsigned int entry); |
|
char *json_array_string(json_t *val, unsigned int entry); |
|
json_t *json_object_dup(json_t *val, const char *entry); |
|
|
|
void align_len(size_t *len); |
|
void realloc_strcat(char **ptr, const char *s); |
|
void trail_slash(char **buf); |
|
void *_ckalloc(size_t len, const char *file, const char *func, const int line); |
|
void *_ckzalloc(size_t len, const char *file, const char *func, const int line); |
|
void _dealloc(void **ptr); |
|
void __bin2hex(void *vs, const void *vp, size_t len); |
|
void *bin2hex(const void *vp, size_t len); |
|
bool _hex2bin(void *p, const void *vhexstr, size_t len, const char *file, const char *func, const int line); |
|
#define hex2bin(p, vhexstr, len) _hex2bin(p, vhexstr, len, __FILE__, __func__, __LINE__) |
|
char *http_base64(const char *src); |
|
void b58tobin(char *b58bin, const char *b58); |
|
|
|
void address_to_pubkeytxn(char *pkh, const char *addr); |
|
int ser_number(uchar *s, int32_t val); |
|
int get_sernumber(uchar *s); |
|
bool fulltest(const uchar *hash, const uchar *target); |
|
|
|
void copy_tv(tv_t *dest, const tv_t *src); |
|
void ts_to_tv(tv_t *val, const ts_t *spec); |
|
void tv_to_ts(ts_t *spec, const tv_t *val); |
|
void us_to_tv(tv_t *val, int64_t us); |
|
void us_to_ts(ts_t *spec, int64_t us); |
|
void ms_to_ts(ts_t *spec, int64_t ms); |
|
void ms_to_tv(tv_t *val, int64_t ms); |
|
void tv_time(tv_t *tv); |
|
void ts_realtime(ts_t *ts); |
|
|
|
void cksleep_prepare_r(ts_t *ts); |
|
void nanosleep_abstime(ts_t *ts_end); |
|
void timeraddspec(ts_t *a, const ts_t *b); |
|
void cksleep_ms_r(ts_t *ts_start, int ms); |
|
void cksleep_us_r(ts_t *ts_start, int64_t us); |
|
void cksleep_ms(int ms); |
|
void cksleep_us(int64_t us); |
|
|
|
double us_tvdiff(tv_t *end, tv_t *start); |
|
int ms_tvdiff(tv_t *end, tv_t *start); |
|
double tvdiff(tv_t *end, tv_t *start); |
|
|
|
void decay_time(double *f, double fadd, double fsecs, double interval); |
|
void suffix_string(double val, char *buf, size_t bufsiz, int sigdigits); |
|
|
|
double le256todouble(const uchar *target); |
|
double diff_from_target(uchar *target); |
|
double diff_from_nbits(char *nbits); |
|
void target_from_diff(uchar *target, double diff); |
|
|
|
void gen_hash(uchar *data, uchar *hash, int len); |
|
|
|
#endif /* LIBCKPOOL_H */
|
|
|