Commit 5a30956

#include <sys/socket.h>

#ifndef PYPY

#define PYOSINPUTHOOK_REPETITIVE 1

#endif

static void

_stdin_callback(CFReadStreamRef stream, CFStreamEventType eventType, void* info)

{

CFRunLoopRef runloop = info;

CFRunLoopStop(runloop);

}

static int sigint_fd = -1;

static void _sigint_handler(int sig)

{

const char c = 'i';

write(sigint_fd, &c, 1);

}

static void _sigint_callback(CFSocketRef s,

CFSocketCallBackType type,

CFDataRef address,

const void * data,

void *info)

{

char c;

int* interrupted = info;

CFSocketNativeHandle handle = CFSocketGetNative(s);

CFRunLoopRef runloop = CFRunLoopGetCurrent();

read(handle, &c, 1);

*interrupted = 1;

CFRunLoopStop(runloop);

}

static CGEventRef _eventtap_callback(

CGEventTapProxy proxy, CGEventType type, CGEventRef event, void *refcon)

{

CFRunLoopRef runloop = refcon;

CFRunLoopStop(runloop);

return event;

}

static int wait_for_stdin(void)

{

int interrupted = 0;

const UInt8 buffer[] = "/dev/fd/0";

const CFIndex n = (CFIndex)strlen((char*)buffer);

CFRunLoopRef runloop = CFRunLoopGetCurrent();

CFURLRef url = CFURLCreateFromFileSystemRepresentation(kCFAllocatorDefault,

buffer,

n,

false);

CFReadStreamRef stream = CFReadStreamCreateWithFile(kCFAllocatorDefault,

url);

CFRelease(url);

CFReadStreamOpen(stream);

#ifdef PYOSINPUTHOOK_REPETITIVE

if (!CFReadStreamHasBytesAvailable(stream))

/* This is possible because of how PyOS_InputHook is called from Python */

#endif

{

int error;

int channel[2];

CFSocketRef sigint_socket = NULL;

PyOS_sighandler_t py_sigint_handler = NULL;

CFStreamClientContext clientContext = {0, NULL, NULL, NULL, NULL};

clientContext.info = runloop;

CFReadStreamSetClient(stream,

kCFStreamEventHasBytesAvailable,

_stdin_callback,

&clientContext);

CFReadStreamScheduleWithRunLoop(stream, runloop, kCFRunLoopDefaultMode);

error = socketpair(AF_UNIX, SOCK_STREAM, 0, channel);

if (!error) {

CFSocketContext context;

context.version = 0;

context.info = &interrupted;

context.retain = NULL;

context.release = NULL;

context.copyDescription = NULL;

fcntl(channel[0], F_SETFL, O_WRONLY | O_NONBLOCK);

sigint_socket = CFSocketCreateWithNative(

kCFAllocatorDefault,

channel[1],

kCFSocketReadCallBack,

_sigint_callback,

&context);

if (sigint_socket) {

CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(

kCFAllocatorDefault, sigint_socket, 0);

CFRelease(sigint_socket);

if (source) {

CFRunLoopAddSource(runloop, source, kCFRunLoopDefaultMode);

CFRelease(source);

sigint_fd = channel[0];

py_sigint_handler = PyOS_setsig(SIGINT, _sigint_handler);

}

}

}

NSEvent* event;

while (true) {

while (true) {

event = [NSApp nextEventMatchingMask: NSEventMaskAny

untilDate: [NSDate distantPast]

inMode: NSDefaultRunLoopMode

dequeue: YES];

if (!event) { break; }

[NSApp sendEvent: event];

}

CFRunLoopRun();

if (interrupted || CFReadStreamHasBytesAvailable(stream)) { break; }

}

if (py_sigint_handler) { PyOS_setsig(SIGINT, py_sigint_handler); }

CFReadStreamUnscheduleFromRunLoop(

stream, runloop, kCFRunLoopCommonModes);

if (sigint_socket) { CFSocketInvalidate(sigint_socket); }

if (!error) {

close(channel[0]);

close(channel[1]);

}

}

CFReadStreamClose(stream);

CFRelease(stream);

if (interrupted) {

errno = EINTR;

raise(SIGINT);

return -1;

}

return 1;

}

- (void)launch:(NSNotification*)notification;

PyOS_InputHook = wait_for_stdin;

WindowServerConnectionManager* connectionManager = [WindowServerConnectionManager sharedManager];

NSWorkspace* workspace = [NSWorkspace sharedWorkspace];

NSNotificationCenter* notificationCenter = [workspace notificationCenter];

[notificationCenter addObserver: connectionManager

selector: @selector(launch:)

name: NSWorkspaceDidLaunchApplicationNotification

object: nil];

location: NSMakePoint(0, 0)

int error;

int interrupted = 0;

int channel[2];

CFSocketRef sigint_socket = NULL;

PyOS_sighandler_t py_sigint_handler = NULL;

CFRunLoopRef runloop = CFRunLoopGetCurrent();

error = pipe(channel);

if (!error) {

CFSocketContext context = {0, NULL, NULL, NULL, NULL};

fcntl(channel[1], F_SETFL, O_WRONLY | O_NONBLOCK);

context.info = &interrupted;

sigint_socket = CFSocketCreateWithNative(kCFAllocatorDefault,

channel[0],

kCFSocketReadCallBack,

_sigint_callback,

&context);

if (sigint_socket) {

CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(

kCFAllocatorDefault, sigint_socket, 0);

CFRelease(sigint_socket);

if (source) {

CFRunLoopAddSource(runloop, source, kCFRunLoopDefaultMode);

CFRelease(source);

sigint_fd = channel[1];

py_sigint_handler = PyOS_setsig(SIGINT, _sigint_handler);

}

}

else

close(channel[0]);

}

if (py_sigint_handler) { PyOS_setsig(SIGINT, py_sigint_handler); }

if (sigint_socket) { CFSocketInvalidate(sigint_socket); }

if (!error) { close(channel[1]); }

if (interrupted) { raise(SIGINT); }

- (void)launch:(NSNotification*)notification

{

CFRunLoopRef runloop;

CFMachPortRef port;

CFRunLoopSourceRef source;

NSDictionary* dictionary = [notification userInfo];

if (![[dictionary valueForKey:@"NSApplicationName"]

localizedCaseInsensitiveContainsString:@"python"])

return;

NSNumber* psnLow = [dictionary valueForKey: @"NSApplicationProcessSerialNumberLow"];

NSNumber* psnHigh = [dictionary valueForKey: @"NSApplicationProcessSerialNumberHigh"];

ProcessSerialNumber psn;

psn.highLongOfPSN = [psnHigh intValue];

psn.lowLongOfPSN = [psnLow intValue];

runloop = CFRunLoopGetCurrent();

port = CGEventTapCreateForPSN(&psn,

kCGHeadInsertEventTap,

kCGEventTapOptionListenOnly,

kCGEventMaskForAllEvents,

&_eventtap_callback,

runloop);

source = CFMachPortCreateRunLoopSource(kCFAllocatorDefault,

port,

0);

CFRunLoopAddSource(runloop, source, kCFRunLoopDefaultMode);

CFRelease(port);

}