ckpool/src/ckpool.h

/*
 * 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.
 */

#ifndef CKPOOL_H
#define CKPOOL_H

#include "config.h"

#include <sys/file.h>

#include "libckpool.h"
#include "uthash.h"

struct ckpool_instance;
typedef struct ckpool_instance ckpool_t;

struct proc_instance {
	ckpool_t *ckp;
	unixsock_t us;
	char *processname;
	char *sockname;
	int pid;
	int (*process)(proc_instance_t *);
};

struct server_instance {
	/* Hash table data */
	UT_hash_handle hh;
	int id;

	char *url;
	char *auth;
	char *pass;
	connsock_t cs;

	void *data; // Private data
};

typedef struct server_instance server_instance_t;

struct ckpool_instance {
	/* Filename of config file */
	char *config;
	/* Kill old instance with same name */
	bool killold;
	/* Logging level */
	int loglevel;
	/* Main process name */
	char *name;
	/* Directory where sockets are created */
	char *socket_dir;
	/* Directory where logs are written */
	char *logdir;
	/* Logfile */
	FILE *logfp;
	int logfd;

	/* Process instance data of parent/child processes */
	proc_instance_t main;

	int proc_instances;
	proc_instance_t **children;

	proc_instance_t *generator;
	proc_instance_t *stratifier;
	proc_instance_t *connector;

	/* Threads of main process */
	pthread_t pth_listener;
	pthread_t pth_watchdog;

	/* Are we running as a proxy */
	bool proxy;

	/* Bitcoind data */
	int btcds;
	char **btcdurl;
	char **btcdauth;
	char **btcdpass;
	int blockpoll; // How frequently in ms to poll bitcoind for block updates

	/* Difficulty settings */
	int mindiff; // Default 1
	int startdiff; // Default 42

	/* Coinbase data */
	char *btcaddress; // Address to mine to
	char *btcsig; // Optional signature to add to coinbase

	/* Stratum options */
	server_instance_t **servers;
	char *serverurl; // URL to bind our server/proxy to
	int update_interval; // Seconds between stratum updates
	int chosen_server; // Chosen server for next connection

	/* Proxy options */
	int proxies;
	char **proxyurl;
	char **proxyauth;
	char **proxypass;
};

ckpool_t *global_ckp;

int process_exit(ckpool_t *ckp, proc_instance_t *pi, int ret);

#define ASPRINTF(strp, fmt, ...) do { \
	if (unlikely(asprintf(strp, fmt, ##__VA_ARGS__) < 0)) \
		quitfrom(1, __FILE__, __func__, __LINE__, "Failed to asprintf"); \
} while (0)

/* Log everything to the logfile, but display warnings on the console as well */
#define LOGMSG(_loglevel, fmt, ...) do { \
	if (global_ckp->loglevel >= _loglevel && fmt) { \
		struct tm *tm; \
		char *BUF = NULL; \
		time_t now_t; \
		int LOGFD = global_ckp->logfd; \
		\
		ASPRINTF(&BUF, fmt, ##__VA_ARGS__); \
		now_t = time(NULL); \
		tm = localtime(&now_t); \
		if (LOGFD) { \
			FILE *LOGFP = global_ckp->logfp; \
			\
			flock(LOGFD, LOCK_EX); \
			fprintf(LOGFP, "[%d-%02d-%02d %02d:%02d:%02d] %s", \
				tm->tm_year + 1900, \
				tm->tm_mon + 1, \
				tm->tm_mday, \
				tm->tm_hour, \
				tm->tm_min, \
				tm->tm_sec, \
				BUF); \
			if (_loglevel <= LOG_ERR) \
				fprintf(LOGFP, " with errno %d: %s", errno, strerror(errno)); \
			fprintf(LOGFP, "\n"); \
			fflush(LOGFP); \
			flock(LOGFD, LOCK_UN); \
		} \
		if (_loglevel <= LOG_WARNING) {\
			fprintf(stderr, "%s", BUF); \
			if (_loglevel <= LOG_ERR) \
				fprintf(stderr, " with errno %d: %s", errno, strerror(errno)); \
			fprintf(stderr, "\n"); \
			fflush(stderr); \
		} \
		free(BUF); \
	} \
} while (0)

#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)

#endif /* CKPOOL_H */