1
0
mirror of https://github.com/0intro/wmii synced 2024-12-18 01:23:56 +03:00
wmii/libixp2/server.c

203 lines
4.9 KiB
C

/*
* (C)opyright MMIV-MMV Anselm R. Garbe <garbeam at gmail dot com>
* See LICENSE file for license details.
*/
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include "ixp.h"
#include "cext.h"
static IXPConn zero_conn = { -1, 0, 0, 0 };
static unsigned char msg[IXP_MAX_MSG];
static IXPConn *
next_free_conn(IXPServer * s)
{
int i;
for(i = 0; i < IXP_MAX_CONN; i++)
if(s->conn[i].fd < 0)
return &s->conn[i];
return nil;
}
static void
prepare_select(IXPServer * s)
{
int i;
FD_ZERO(&s->rd);
for(i = 0; i < IXP_MAX_CONN; i++) {
if(s->conn[i].fd >= 0) {
if(s->maxfd < s->conn[i].fd)
s->maxfd = s->conn[i].fd;
if(s->conn[i].read)
FD_SET(s->conn[i].fd, &s->rd);
}
}
}
void
ixp_server_rm_conn(IXPServer * s, IXPConn * c)
{
if(!c->dont_close) {
shutdown(c->fd, SHUT_RDWR);
close(c->fd);
}
if(s->freeconn)
s->freeconn(s, c);
*c = zero_conn;
}
static IXPConn *
init_conn(IXPConn * c, int fd, int dont_close,
void (*read) (IXPServer *, IXPConn *))
{
*c = zero_conn;
c->fd = fd;
c->dont_close = dont_close;
c->read = read;
return c;
}
IXPConn *
ixp_server_add_conn(IXPServer * s, int fd, int dont_close,
void (*read) (IXPServer *, IXPConn *))
{
IXPConn *c = next_free_conn(s);
if(!c)
return nil;
return init_conn(c, fd, dont_close, read);
}
static void
handle_conns(IXPServer * s)
{
int i;
for(i = 0; i < IXP_MAX_CONN; i++) {
if(s->conn[i].fd >= 0) {
if(FD_ISSET(s->conn[i].fd, &s->rd) && s->conn[i].read)
/* call back read handler */
s->conn[i].read(s, &s->conn[i]);
}
}
}
static void
server_client_read(IXPServer * s, IXPConn * c)
{
unsigned int i, msize;
s->errstr = 0;
if(!(msize = ixp_recv_message(c->fd, msg, IXP_MAX_MSG, &s->errstr))) {
ixp_server_rm_conn(s, c);
return;
}
fprintf(stderr, "msize=%d\n", msize);
if((msize = ixp_msg_to_fcall(msg, IXP_MAX_MSG, &s->fcall))) {
for(i = 0; s->funcs && s->funcs[i].id; i++) {
if(s->funcs[i].id == s->fcall.id) {
if(s->funcs[i].tfunc(s, c) == -1)
break;
msize = ixp_fcall_to_msg(&s->fcall, msg, s->fcall.maxmsg);
fprintf(stderr, "msize=%d\n", msize);
if(ixp_send_message(c->fd, msg, msize, &s->errstr) !=
msize)
break;
return;
}
}
}
fprintf(stderr, "function id=%d\n", s->fcall.id);
if(!s->errstr)
s->errstr = "function not supported";
s->fcall.id = RERROR;
cext_strlcpy(s->fcall.errstr, s->errstr, sizeof(s->fcall.errstr));
msize = ixp_fcall_to_msg(&s->fcall, msg, IXP_MAX_MSG);
if(ixp_send_message(c->fd, msg, msize, &s->errstr) != msize)
ixp_server_rm_conn(s, c);
}
static void
server_read(IXPServer * s, IXPConn * c)
{
int fd;
IXPConn *new = next_free_conn(s);
if(new && ((fd = ixp_accept_sock(c->fd)) >= 0))
init_conn(new, fd, 0, server_client_read);
}
void
ixp_server_loop(IXPServer * s)
{
int r;
s->running = 1;
s->errstr = 0;
/* main loop */
while(s->running) {
prepare_select(s);
r = select(s->maxfd + 1, &s->rd, 0, 0, 0);
if(r == -1 && errno == EINTR)
continue;
if(r < 0) {
s->errstr = "fatal select error";
break; /* allow cleanups in IXP using app */
} else if(r > 0)
handle_conns(s);
}
}
int
ixp_server_tversion(IXPServer * s, IXPConn * c)
{
fprintf(stderr, "got version %s (%s) %d (%d)\n", s->fcall.version,
IXP_VERSION, s->fcall.maxmsg, IXP_MAX_MSG);
if(strncmp(s->fcall.version, IXP_VERSION, strlen(IXP_VERSION))) {
s->errstr = "9P versions differ";
return -1;
} else if(s->fcall.maxmsg > IXP_MAX_MSG)
s->fcall.maxmsg = IXP_MAX_MSG;
s->fcall.id = RVERSION;
return 0;
}
int
ixp_server_init(IXPServer * s, char *sockfile, IXPTFunc * funcs,
void (*freeconn) (IXPServer *, IXPConn *))
{
int fd, i;
s->funcs = funcs;
s->freeconn = freeconn;
s->errstr = 0;
if(!sockfile) {
s->errstr = "no socket file provided or invalid directory";
return -1;
}
if((fd = ixp_create_sock(sockfile, &s->errstr)) < 0)
return -1;
for(i = 0; i < IXP_MAX_CONN; i++)
s->conn[i] = zero_conn;
ixp_server_add_conn(s, fd, 0, server_read);
return 0;
}
void
ixp_server_deinit(IXPServer * s)
{
int i;
/* shut down server */
for(i = 0; i < IXP_MAX_CONN; i++)
if(s->conn[i].fd >= 0)
ixp_server_rm_conn(s, &s->conn[i]);
}