civetweb/src/timer.inl

#if !defined(MAX_TIMERS)
#define MAX_TIMERS MAX_WORKER_THREADS
#endif

typedef int (*taction)(void *arg);

struct ttimer {
	double time;
	double period;
	taction action;
	void *arg;
};

struct ttimers {
	pthread_t threadid;               /* Timer thread ID */
	pthread_mutex_t mutex;            /* Protects timer lists */
	struct ttimer timers[MAX_TIMERS]; /* List of timers */
	unsigned timer_count;             /* Current size of timer list */
};

static int timer_add(struct mg_context *ctx, double next_time, double period,
                     int is_relative, taction action, void *arg) {
	unsigned u, v;
	int error = 0;
	struct timespec now;

	if (ctx->stop_flag) {
		return 0;
	}

	if (is_relative) {
		clock_gettime(CLOCK_MONOTONIC, &now);
		next_time += now.tv_sec;
		next_time += now.tv_nsec * 1.0E-9;
	}

	pthread_mutex_lock(&ctx->timers->mutex);
	if (ctx->timers->timer_count == MAX_TIMERS) {
		error = 1;
	} else {
		for (u = 0; u < ctx->timers->timer_count; u++) {
			if (ctx->timers->timers[u].time < next_time) {
				for (v = ctx->timers->timer_count; v > u; v--) {
					ctx->timers->timers[v] = ctx->timers->timers[v - 1];
				}
				break;
			}
		}
		ctx->timers->timers[u].time = next_time;
		ctx->timers->timers[u].period = period;
		ctx->timers->timers[u].action = action;
		ctx->timers->timers[u].arg = arg;
		ctx->timers->timer_count++;
	}
	pthread_mutex_unlock(&ctx->timers->mutex);
	return error;
}

static void timer_thread_run(void *thread_func_param) {
	struct mg_context *ctx = (struct mg_context *)thread_func_param;
	struct timespec now;
	double d;
	unsigned u;
	int re_schedule;
	struct ttimer t;

#if defined(HAVE_CLOCK_NANOSLEEP) /* Linux with librt */
	/* TODO */
	while (clock_nanosleep(CLOCK_MONOTONIC, TIMER_ABSTIME, &request,
	                       &request) == EINTR) { /*nop*/
		;
	}
#else
	clock_gettime(CLOCK_MONOTONIC, &now);
	d = (double)now.tv_sec + (double)now.tv_nsec * 1.0E-9;
	while (ctx->stop_flag == 0) {
		pthread_mutex_lock(&ctx->timers->mutex);
		if (ctx->timers->timer_count > 0 && d >= ctx->timers->timers[0].time) {
			t = ctx->timers->timers[0];
			for (u = 1; u < ctx->timers->timer_count; u++) {
				ctx->timers->timers[u - 1] = ctx->timers->timers[u];
			}
			ctx->timers->timer_count--;
			pthread_mutex_unlock(&ctx->timers->mutex);
			re_schedule = t.action(t.arg);
			if (re_schedule && (t.period > 0)) {
				timer_add(ctx, t.time + t.period, t.period, 0, t.action, t.arg);
			}
			continue;
		} else {
			pthread_mutex_unlock(&ctx->timers->mutex);
		}
		mg_sleep(1);
		clock_gettime(CLOCK_MONOTONIC, &now);
		d = (double)now.tv_sec + (double)now.tv_nsec * 1.0E-9;
	}
#endif
}

#ifdef _WIN32
static unsigned __stdcall timer_thread(void *thread_func_param) {
	timer_thread_run(thread_func_param);
	return 0;
}
#else
static void *timer_thread(void *thread_func_param) {
	timer_thread_run(thread_func_param);
	return NULL;
}
#endif /* _WIN32 */

static int timers_init(struct mg_context *ctx) {
	ctx->timers = (struct ttimers *)mg_calloc(sizeof(struct ttimers), 1);
	(void)pthread_mutex_init(&ctx->timers->mutex, NULL);

	/* Start timer thread */
	mg_start_thread_with_id(timer_thread, ctx, &ctx->timers->threadid);

	return 0;
}

static void timers_exit(struct mg_context *ctx) {
	if (ctx->timers) {
		(void)pthread_mutex_destroy(&ctx->timers->mutex);
		mg_free(ctx->timers);
	}
}