aboutsummaryrefslogtreecommitdiffstats
path: root/src/main_notify.c
blob: a358a7053b2ed9629fab038d910e87ccb5e3b915 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/* the Music Player Daemon (MPD)
 * Copyright (C) 2003-2007 Warren Dukes <warren.dukes@gmail.com>
 * Copyright (C) 2008 Eric Wong <normalperson@yhbt.net>
 *
 * This project's homepage is: http://www.musicpd.org
 *
 * 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "main_notify.h"
#include "notify.h"
#include "utils.h"
#include "ioops.h"
#include "gcc.h"
#include "log.h"

static struct ioOps main_notify_IO;
static int main_pipe[2];
static pthread_t main_task;
static pthread_cond_t main_wakeup = PTHREAD_COND_INITIALIZER;
static pthread_mutex_t main_wakeup_mutex = PTHREAD_MUTEX_INITIALIZER;
static pthread_mutex_t select_mutex = PTHREAD_MUTEX_INITIALIZER;

static int ioops_fdset(fd_set * rfds,
                       mpd_unused fd_set * wfds, mpd_unused fd_set * efds)
{
	FD_SET(main_pipe[0], rfds);
	return main_pipe[0];
}

static void consume_pipe(void)
{
	char buffer[2];
	ssize_t r = read(main_pipe[0], buffer, sizeof(buffer));

	if (r < 0 && errno != EAGAIN && errno != EINTR)
		FATAL("error reading from pipe: %s\n", strerror(errno));
}

static int ioops_consume(int fd_count, fd_set * rfds,
                         mpd_unused fd_set * wfds, mpd_unused fd_set * efds)
{
	if (FD_ISSET(main_pipe[0], rfds)) {
		consume_pipe();
		FD_CLR(main_pipe[0], rfds);
		fd_count--;
	}
	return fd_count;
}

void init_main_notify(void)
{
	if (pipe(main_pipe) < 0)
		FATAL("Couldn't open pipe: %s", strerror(errno));
	if (set_nonblocking(main_pipe[0]) < 0)
		FATAL("Couldn't set non-blocking on main_notify fd: %s",
		      strerror(errno));
	if (set_nonblocking(main_pipe[1]) < 0)
		FATAL("Couldn't set non-blocking on main_notify fd: %s",
		      strerror(errno));
	main_notify_IO.fdset = ioops_fdset;
	main_notify_IO.consume = ioops_consume;
	registerIO(&main_notify_IO);
	main_task = pthread_self();
}

static int wakeup_via_pipe(void)
{
	int ret = pthread_mutex_trylock(&select_mutex);
	if (ret == EBUSY) {
		ssize_t w = write(main_pipe[1], "", 1);
		if (w < 0 && errno != EAGAIN && errno != EINTR)
			FATAL("error writing to pipe: %s\n",
			      strerror(errno));
		return 1;
	} else {
		pthread_mutex_unlock(&select_mutex);
		return 0;
	}
}

static void wakeup_via_cond(void)
{
	int ret = pthread_mutex_trylock(&main_wakeup_mutex);

	if (ret == EBUSY)
		return; /* nope, no need to wakeup at all */
	pthread_cond_broadcast(&main_wakeup);
	pthread_mutex_unlock(&main_wakeup_mutex);
}

void wakeup_main_task(void)
{
	assert(!pthread_equal(main_task, pthread_self()));

	if (!wakeup_via_pipe())
		wakeup_via_cond();
}

void main_notify_lock(void)
{
	assert(pthread_equal(main_task, pthread_self()));
	pthread_mutex_lock(&select_mutex);
}

void main_notify_unlock(void)
{
	assert(pthread_equal(main_task, pthread_self()));
	pthread_mutex_unlock(&select_mutex);
}

void wait_main_task(void)
{
	assert(pthread_equal(main_task, pthread_self()));

	pthread_mutex_lock(&main_wakeup_mutex);
	pthread_cond_wait(&main_wakeup, &main_wakeup_mutex);
	pthread_mutex_unlock(&main_wakeup_mutex);
}