mirror of
https://github.com/GOSTSec/sgminer
synced 2025-03-13 06:01:03 +00:00
Win32 threading and longpoll keepalive fixes.
This commit is contained in:
Ycros
parent
5f360394c0
commit
d37c3fe56b
42
main.c
42
main.c
@ -1014,7 +1014,7 @@ void kill_work(void)
|
||||
|
||||
/* Kill the watchdog thread */
|
||||
thr = &thr_info[watchdog_thr_id];
|
||||
pthread_cancel(thr->pth);
|
||||
pthread_cancel(*thr->pth);
|
||||
|
||||
/* Stop the mining threads*/
|
||||
for (i = 0; i < mining_threads; i++) {
|
||||
@ -1023,14 +1023,14 @@ void kill_work(void)
|
||||
continue;
|
||||
tq_freeze(thr->q);
|
||||
/* No need to check if this succeeds or not */
|
||||
pthread_cancel(thr->pth);
|
||||
pthread_cancel(*thr->pth);
|
||||
}
|
||||
|
||||
/* Stop the others */
|
||||
thr = &thr_info[stage_thr_id];
|
||||
pthread_cancel(thr->pth);
|
||||
pthread_cancel(*thr->pth);
|
||||
thr = &thr_info[longpoll_thr_id];
|
||||
pthread_cancel(thr->pth);
|
||||
pthread_cancel(*thr->pth);
|
||||
|
||||
wc = calloc(1, sizeof(*wc));
|
||||
if (unlikely(!wc)) {
|
||||
@ -2736,7 +2736,7 @@ static void stop_longpoll(void)
|
||||
struct thr_info *thr = &thr_info[longpoll_thr_id];
|
||||
|
||||
tq_freeze(thr->q);
|
||||
pthread_cancel(thr->pth);
|
||||
pthread_cancel(*thr->pth);
|
||||
have_longpoll = false;
|
||||
}
|
||||
|
||||
@ -2746,10 +2746,10 @@ static void start_longpoll(void)
|
||||
{
|
||||
struct thr_info *thr = &thr_info[longpoll_thr_id];
|
||||
|
||||
tq_thaw(thr->q);
|
||||
if (unlikely(pthread_create(&thr->pth, NULL, longpoll_thread, thr)))
|
||||
tq_thaw(thr->q);
|
||||
if (unlikely(thr_info_create(thr, NULL, longpoll_thread, thr)))
|
||||
quit(1, "longpoll thread create failed");
|
||||
pthread_detach(thr->pth);
|
||||
pthread_detach(*thr->pth);
|
||||
}
|
||||
|
||||
static void restart_longpoll(void)
|
||||
@ -2764,7 +2764,7 @@ static void reinit_cputhread(int thr_id)
|
||||
struct thr_info *thr = &thr_info[thr_id];
|
||||
|
||||
tq_freeze(thr->q);
|
||||
if (!(pthread_cancel(thr->pth)) && pthread_join(thr->pth, NULL)) {
|
||||
if (!(pthread_cancel(*thr->pth)) && pthread_join(*thr->pth, NULL)) {
|
||||
applog(LOG_ERR, "Failed to pthread_join in reinit_cputhread");
|
||||
goto failed_out;
|
||||
}
|
||||
@ -2774,7 +2774,7 @@ static void reinit_cputhread(int thr_id)
|
||||
|
||||
thread_reportin(thr);
|
||||
|
||||
if (unlikely(pthread_create(&thr->pth, NULL, miner_thread, thr))) {
|
||||
if (unlikely(thr_info_create(thr, NULL, miner_thread, thr))) {
|
||||
applog(LOG_ERR, "thread %d create failed", thr_id);
|
||||
goto failed_out;
|
||||
}
|
||||
@ -2792,7 +2792,7 @@ static void reinit_gputhread(int thr_id)
|
||||
char name[256];
|
||||
|
||||
tq_freeze(thr->q);
|
||||
if (!(pthread_cancel(thr->pth)) && pthread_join(thr->pth, NULL)) {
|
||||
if (!(pthread_cancel(*thr->pth)) && pthread_join(*thr->pth, NULL)) {
|
||||
applog(LOG_ERR, "Failed to pthread_join in reinit_gputhread");
|
||||
goto failed_out;
|
||||
}
|
||||
@ -2809,7 +2809,7 @@ static void reinit_gputhread(int thr_id)
|
||||
|
||||
thread_reportin(thr);
|
||||
|
||||
if (unlikely(pthread_create(&thr->pth, NULL, gpuminer_thread, thr))) {
|
||||
if (unlikely(thr_info_create(thr, NULL, gpuminer_thread, thr))) {
|
||||
applog(LOG_ERR, "thread %d create failed", thr_id);
|
||||
goto failed_out;
|
||||
}
|
||||
@ -3272,7 +3272,7 @@ int main (int argc, char *argv[])
|
||||
quit(1, "Failed to tq_new");
|
||||
|
||||
/* start work I/O thread */
|
||||
if (pthread_create(&thr->pth, NULL, workio_thread, thr))
|
||||
if (thr_info_create(thr, NULL, workio_thread, thr))
|
||||
quit(1, "workio thread create failed");
|
||||
|
||||
/* init longpoll thread info */
|
||||
@ -3303,9 +3303,9 @@ int main (int argc, char *argv[])
|
||||
if (!thr->q)
|
||||
quit(1, "Failed to tq_new");
|
||||
/* start stage thread */
|
||||
if (pthread_create(&thr->pth, NULL, stage_thread, thr))
|
||||
if (thr_info_create(thr, NULL, stage_thread, thr))
|
||||
quit(1, "stage thread create failed");
|
||||
pthread_detach(thr->pth);
|
||||
pthread_detach(*thr->pth);
|
||||
|
||||
/* Flag the work as ready forcing the mining threads to wait till we
|
||||
* actually put something into the queue */
|
||||
@ -3374,7 +3374,7 @@ int main (int argc, char *argv[])
|
||||
|
||||
thread_reportin(thr);
|
||||
|
||||
if (unlikely(pthread_create(&thr->pth, NULL, gpuminer_thread, thr)))
|
||||
if (unlikely(thr_info_create(thr, NULL, gpuminer_thread, thr)))
|
||||
quit(1, "thread %d create failed", i);
|
||||
i++;
|
||||
}
|
||||
@ -3402,7 +3402,7 @@ int main (int argc, char *argv[])
|
||||
|
||||
thread_reportin(thr);
|
||||
|
||||
if (unlikely(pthread_create(&thr->pth, NULL, miner_thread, thr)))
|
||||
if (unlikely(thr_info_create(thr, NULL, miner_thread, thr)))
|
||||
quit(1, "thread %d create failed", i);
|
||||
}
|
||||
|
||||
@ -3414,7 +3414,7 @@ int main (int argc, char *argv[])
|
||||
watchdog_thr_id = mining_threads + 2;
|
||||
thr = &thr_info[watchdog_thr_id];
|
||||
/* start wakeup thread */
|
||||
if (pthread_create(&thr->pth, NULL, watchdog_thread, NULL))
|
||||
if (thr_info_create(thr, NULL, watchdog_thread, NULL))
|
||||
quit(1, "wakeup thread create failed");
|
||||
|
||||
/* Restart count as it will be wrong till all threads are started */
|
||||
@ -3435,12 +3435,12 @@ int main (int argc, char *argv[])
|
||||
/* Create curses input thread for keyboard input */
|
||||
input_thr_id = mining_threads + 4;
|
||||
thr = &thr_info[input_thr_id];
|
||||
if (pthread_create(&thr->pth, NULL, input_thread, thr))
|
||||
if (thr_info_create(thr, NULL, input_thread, thr))
|
||||
quit(1, "input thread create failed");
|
||||
pthread_detach(thr->pth);
|
||||
pthread_detach(*thr->pth);
|
||||
|
||||
/* main loop - simply wait for workio thread to exit */
|
||||
pthread_join(thr_info[work_thr_id].pth, NULL);
|
||||
pthread_join(*thr_info[work_thr_id].pth, NULL);
|
||||
applog(LOG_INFO, "workio thread dead, exiting.");
|
||||
|
||||
gettimeofday(&total_tv_end, NULL);
|
||||
|
||||
10
miner.h
10
miner.h
@ -29,7 +29,11 @@
|
||||
#ifdef HAVE_ALLOCA_H
|
||||
# include <alloca.h>
|
||||
#elif defined __GNUC__
|
||||
# define alloca __builtin_alloca
|
||||
# ifndef WIN32
|
||||
# define alloca __builtin_alloca
|
||||
# else
|
||||
# include <malloc.h>
|
||||
# endif
|
||||
#elif defined _AIX
|
||||
# define alloca __alloca
|
||||
#elif defined _MSC_VER
|
||||
@ -145,13 +149,15 @@ struct cgpu_info {
|
||||
|
||||
struct thr_info {
|
||||
int id;
|
||||
pthread_t pth;
|
||||
pthread_t *pth;
|
||||
struct thread_q *q;
|
||||
struct cgpu_info *cgpu;
|
||||
struct timeval last;
|
||||
bool getwork;
|
||||
};
|
||||
|
||||
extern inline int thr_info_create(struct thr_info *thr, pthread_attr_t *attr, void *(*start) (void *), void *arg);
|
||||
|
||||
static inline uint32_t swab32(uint32_t v)
|
||||
{
|
||||
#ifdef WANT_BUILTIN_BSWAP
|
||||
|
||||
48
util.c
48
util.c
@ -24,8 +24,13 @@
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/tcp.h>
|
||||
#ifndef WIN32
|
||||
# include <sys/socket.h>
|
||||
# include <netinet/tcp.h>
|
||||
#else
|
||||
# include <winsock2.h>
|
||||
# include <mstcpip.h>
|
||||
#endif
|
||||
#include "miner.h"
|
||||
#include "elist.h"
|
||||
|
||||
@ -251,18 +256,34 @@ int json_rpc_call_sockopt_cb(void *userdata, curl_socket_t fd, curlsocktype purp
|
||||
int tcp_keepcnt = 5;
|
||||
int tcp_keepidle = 120;
|
||||
int tcp_keepintvl = 120;
|
||||
|
||||
#ifndef WIN32
|
||||
|
||||
if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive)))
|
||||
if (unlikely(setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive))))
|
||||
return 1;
|
||||
|
||||
if (setsockopt(fd, SOL_TCP, TCP_KEEPCNT, &tcp_keepcnt, sizeof(tcp_keepcnt)))
|
||||
if (unlikely(setsockopt(fd, SOL_TCP, TCP_KEEPCNT, &tcp_keepcnt, sizeof(tcp_keepcnt))))
|
||||
return 1;
|
||||
|
||||
if (setsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &tcp_keepidle, sizeof(tcp_keepidle)))
|
||||
if (unlikely(setsockopt(fd, SOL_TCP, TCP_KEEPIDLE, &tcp_keepidle, sizeof(tcp_keepidle))))
|
||||
return 1;
|
||||
|
||||
if (setsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &tcp_keepintvl, sizeof(tcp_keepintvl)))
|
||||
if (unlikely(setsockopt(fd, SOL_TCP, TCP_KEEPINTVL, &tcp_keepintvl, sizeof(tcp_keepintvl))))
|
||||
return 1;
|
||||
|
||||
#else
|
||||
|
||||
struct tcp_keepalive vals;
|
||||
vals.onoff = 1;
|
||||
vals.keepalivetime = tcp_keepidle * 1000;
|
||||
vals.keepaliveinterval = tcp_keepintvl * 1000;
|
||||
|
||||
DWORD outputBytes;
|
||||
|
||||
if (unlikely(WSAIoctl(fd, SIO_KEEPALIVE_VALS, &vals, sizeof(vals), NULL, 0, &outputBytes, NULL, NULL)))
|
||||
return 1;
|
||||
|
||||
#endif
|
||||
|
||||
return 0;
|
||||
}
|
||||
@ -647,3 +668,18 @@ out:
|
||||
return rval;
|
||||
}
|
||||
|
||||
inline int thr_info_create(struct thr_info *thr, pthread_attr_t *attr, void *(*start) (void *), void *arg)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
thr->pth = malloc(sizeof(pthread_t));
|
||||
ret = pthread_create(thr->pth, attr, start, arg);
|
||||
|
||||
if (unlikely(ret)) {
|
||||
free(thr->pth);
|
||||
thr->pth = 0;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user