v86.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <unistd.h>
   4 #include <string.h>
   5 #include <errno.h>
   6 #include <time.h>
   7 #include <fcntl.h>
   8
   9 #include <sys/socket.h>
  10 #include <sys/poll.h>
  11
  12 #include <linux/netlink.h>
  13 #include <linux/rtnetlink.h>
  14
  15 #include <arpa/inet.h>
  16
  17 #include "v86.h"
  18
  19 static int need_exit;
  20 static __u32 seq;
  21
  22 static int netlink_send(int s, struct cn_msg *msg)
  23 {
  24         struct nlmsghdr *nlh;
  25         unsigned int size;
  26         int err;
  27         char buf[CONNECTOR_MAX_MSG_SIZE];
  28         struct cn_msg *m;
  29
  30         size = NLMSG_SPACE(sizeof(struct cn_msg) + msg->len);
  31
  32         nlh = (struct nlmsghdr *)buf;
  33         nlh->nlmsg_seq = seq++;
  34         nlh->nlmsg_pid = getpid();
  35         nlh->nlmsg_type = NLMSG_DONE;
  36         nlh->nlmsg_len = NLMSG_LENGTH(size - sizeof(*nlh));
  37         nlh->nlmsg_flags = 0;
  38
  39         m = NLMSG_DATA(nlh);
  40         memcpy(m, msg, sizeof(*m) + msg->len);
  41
  42         err = send(s, nlh, size, 0);
  43         if (err == -1)
  44                 ulog("Failed to send: %s [%d].\n", strerror(errno), errno);
  45
  46         return err;
  47 }
  48
  49 int req_exec(int s, struct cn_msg *msg)
  50 {
  51         struct uvesafb_task *tsk = (struct uvesafb_task*)(msg + 1);
  52         u8 *buf = (u8*)tsk + sizeof(struct uvesafb_task);
  53
  54         if (tsk->flags & TF_EXIT)
  55                 return 1;
  56
  57         v86_task(tsk, buf);
  58         netlink_send(s, msg);
  59
  60         return 0;
  61 }
  62
  63
  64 int main(int argc, char *argv[])
  65 {
  66         char buf[CONNECTOR_MAX_MSG_SIZE];
  67         int len, i, err = 0, s;
  68         struct nlmsghdr *reply;
  69         struct sockaddr_nl l_local;
  70         struct cn_msg *data;
  71         struct pollfd pfd;
  72
  73         s = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_CONNECTOR);
  74         if (s == -1) {
  75                 perror("socket");
  76                 return -1;
  77         }
  78
  79         l_local.nl_family = AF_NETLINK;
  80         l_local.nl_groups = 1 << (CN_IDX_V86D-1); /* bitmask of requested groups */
  81         l_local.nl_pid = 0;
  82
  83         if (bind(s, (struct sockaddr *)&l_local, sizeof(struct sockaddr_nl)) == -1) {
  84                 perror("bind");
  85                 close(s);
  86                 return -1;
  87         }
  88
  89         i = fork();
  90         if (i) {
  91                 exit(0);
  92         }
  93
  94         setsid();
  95         chdir("/");
  96
  97         openlog("v86d", 0, LOG_KERN);
  98
  99         if (v86_init())
 100                 return -1;
 101
 102         memset(buf, 0, sizeof(buf));
 103         pfd.fd = s;
 104
 105         while (!need_exit) {
 106                 pfd.events = POLLIN;
 107                 pfd.revents = 0;
 108                 switch (poll(&pfd, 1, -1)) {
 109                         case 0:
 110                                 need_exit = 1;
 111                                 continue;
 112                         case -1:
 113                                 if (errno != EINTR) {
 114                                         need_exit = 1;
 115                                         break;
 116                                 }
 117                                 continue;
 118                 }
 119
 120                 memset(buf, 0, sizeof(buf));
 121                 len = recv(s, buf, sizeof(buf), 0);
 122                 if (len == -1) {
 123                         perror("recv buf");
 124                         err = -1;
 125                         goto out;
 126                 }
 127
 128                 reply = (struct nlmsghdr *)buf;
 129
 130                 switch (reply->nlmsg_type) {
 131                 case NLMSG_ERROR:
 132                         ulog("Error message received.\n");
 133                         break;
 134
 135                 case NLMSG_DONE:
 136                         data = (struct cn_msg *)NLMSG_DATA(reply);
 137                         if (req_exec(s, data))
 138                                 goto out;
 139                         break;
 140                 default:
 141                         break;
 142                 }
 143         }
 144
 145 out:
 146         v86_cleanup();
 147
 148         closelog();
 149         close(s);
 150         return err;
 151 }