dispatch_cffd.c   [plain text]

#include <dispatch/dispatch.h>

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/param.h>
#include <sys/ucred.h>
#include <sys/mount.h>
#include <sys/queue.h>
#include <sys/errno.h>
#include <sys/types.h>
#include <sys/event.h>
#include <sys/time.h>

#include <CoreServices/CoreServices.h>

#include "dispatch_test.h"

int debug = 0;

#define DEBUG(...) do { \
		if (debug) fprintf(stderr, __VA_ARGS__); \
	} while(0);

#define assert_errno(str, expr) do { \
	if (!(expr)) { \
		fprintf(stderr, "%s: %s\n", (str), strerror(errno)); \
		exit(1); \
	} } while(0);

int
init_kqueue(void)
{
	int kq;
	int res;
	struct kevent ke;
	static struct timespec t0;

	kq = kqueue();
	assert_errno("kqueue", kq >= 0);
	
	EV_SET(&ke, 1, EVFILT_TIMER, EV_ADD, NOTE_SECONDS, 1, 0);
	
	res = kevent(kq, &ke, 1, NULL, 0, &t0);
	assert_errno("kevent", res == 0);

	return kq;
}

int
read_kevent(int kq)
{
	int res;
	struct kevent ke;
	//static struct timespec t0;

	res = kevent(kq, NULL, 0, &ke, 1, NULL);
	assert_errno("kevent", res >= 0);

	fprintf(stdout, "kevent.data = %ld\n", ke.data);

	return (res < 0);
}


static void
cffd_callback(CFFileDescriptorRef cffd,
	CFOptionFlags callBackTypes __attribute__((unused)),
	void *info __attribute__((unused)))
{
	int kq;

	kq = CFFileDescriptorGetNativeDescriptor(cffd);
	if (read_kevent(kq) == 0) {
		// ...
	}
 
	CFFileDescriptorEnableCallBacks(cffd, kCFFileDescriptorReadCallBack);
}

void
timer()
{
	dispatch_source_t ds;
	ds = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, dispatch_get_main_queue());
	assert(ds);
	dispatch_source_set_timer(ds, dispatch_time(0, 1*NSEC_PER_SEC), NSEC_PER_SEC, 0);
	dispatch_source_set_event_handler(ds, ^{
		printf("ping\n");
	});
	dispatch_resume(ds);
}

void
hangup()
{
	dispatch_source_t ds;
	ds = dispatch_source_create(DISPATCH_SOURCE_TYPE_SIGNAL, SIGHUP, 0, dispatch_get_main_queue());
	assert(ds);
	dispatch_source_set_event_handler(ds, ^{
		printf("hangup\n");
	});
	dispatch_resume(ds);
}

int
main(int argc __attribute__((unused)), char *argv[] __attribute__((unused)))
{
	int kq;
	CFFileDescriptorRef cffd;
	CFRunLoopSourceRef  rls;
	CFFileDescriptorContext ctx;

	test_start("CFFileDescriptor");

	signal(SIGHUP, SIG_IGN);

	kq = init_kqueue();

	memset(&ctx, 0, sizeof(CFFileDescriptorContext));
	cffd = CFFileDescriptorCreate(NULL, kq, 1, cffd_callback, &ctx);
	assert(cffd);
    
	rls = CFFileDescriptorCreateRunLoopSource(NULL, cffd, 0);
	assert(rls);
	CFRunLoopAddSource(CFRunLoopGetCurrent(), rls, kCFRunLoopDefaultMode);
	CFFileDescriptorEnableCallBacks(cffd, kCFFileDescriptorReadCallBack);

//	timer();
//	hangup();
	
	CFRunLoopRunInMode(kCFRunLoopDefaultMode, 10.0, false);

	test_stop();

	return 0;
}