beanstalkd/sunos.c

#include <stdint.h>
#include <fcntl.h>
#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <poll.h>
#include <port.h>
#include "dat.h"

static int portfd;

int
sockinit(void)
{
    portfd = port_create();
    if (portfd == -1) {
        twarn("port_create");
        return -1;
    }
    return 0;
}


int
sockwant(Socket *s, int rw)
{
    int events = 0;

    if (rw) {
        switch (rw) {
        case 'r':
            events |= POLLIN;
            break;
        case 'w':
            events |= POLLOUT;
            break;
        }
    }

    events |= POLLPRI;

    if (!s->added && !rw) {
        return 0;
    } else if (!s->added && rw) {
        s->added = 1;
        return port_associate(portfd, PORT_SOURCE_FD, s->fd, events, (void *)s);
    } else if (!rw) {
        return port_dissociate(portfd, PORT_SOURCE_FD, s->fd);
    } else {
        port_dissociate(portfd, PORT_SOURCE_FD, s->fd);
        return port_associate(portfd, PORT_SOURCE_FD, s->fd, events, (void *)s);
    }
}


int
socknext(Socket **s, int64 timeout)
{
    int r;
    uint_t n = 1;
    struct port_event pe;
    struct timespec ts;

    ts.tv_sec = timeout / 1000000000;
    ts.tv_nsec = timeout % 1000000000;
    r = port_getn(portfd, &pe, 1, &n, &ts);
    if (r == -1 && errno != ETIME && errno != EINTR) {
        twarn("port_getn");
        return -1;
    }

    if (r == 0) {
        *s = pe.portev_user;
        if (pe.portev_events & POLLHUP) {
            return 'h';
        } else if (pe.portev_events & POLLIN) {
            if (sockwant(*s, 'r') == -1) {
                return -1;
            }
            return 'r';
        } else if (pe.portev_events & POLLOUT) {
            if (sockwant(*s, 'w') == -1) {
                return -1;
            }
            return 'w';
        }
    }

    return 0;
}