URL https://opencores.org/ocsvn/or1k/or1k/trunk

Subversion Repositories or1k

[/] [or1k/] [trunk/] [or1ksim/] [vapi/] [vapi.c] - Blame information for rev 1767

Go to most recent revision | Details | Compare with Previous | View Log


/* vapi.c -- Verification API Interface
 
   Copyright (C) 2001, Marko Mlinar, markom@opencores.org
   Copyright (C) 2008 Embecosm Limited
 
   Contributor Jeremy Bennett <jeremy.bennett@embecosm.com>
 
   This file is part of OpenRISC 1000 Architectural Simulator.
 
   This program is free software; you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the Free
   Software Foundation; either version 3 of the License, or (at your option)
   any later version.
 
   This program is distributed in the hope that it will be useful, but WITHOUT
   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
   FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
   more details.
 
   You should have received a copy of the GNU General Public License along
   with this program.  If not, see <http://www.gnu.org/licenses/>. */
 
/* This program is commented throughout in a fashion suitable for processing
   with Doxygen. */
 
 
/* Autoconf and/or portability configuration */
#include "config.h"
#include "port.h"
 
/* System includes */
#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <errno.h>
#include <sys/poll.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <fcntl.h>
#include <netdb.h>
#include <netinet/tcp.h>
 
/* Package includes */
#include "sim-config.h"
#include "vapi.h"
 
 
static unsigned int serverIP = 0;
 
static unsigned int server_fd = 0;
static unsigned int nhandlers = 0;
 
static int tcp_level = 0;
 
static struct vapi_handler
{
  int fd;
  unsigned long base_id, num_ids;
  void (*read_func) (unsigned long, unsigned long, void *);
  void *priv_dat;
  struct vapi_handler *next;
  int temp;
} *vapi_handler = NULL;
 
/* Structure for polling, it is cached, that it doesn't have to be rebuilt each time */
static struct pollfd *fds = NULL;
static int nfds = 0;
 
/* Rebuilds the fds structures; see fds.  */
void
rebuild_fds ()
{
  struct vapi_handler *t;
  if (fds)
    free (fds);
  fds = (struct pollfd *) malloc (sizeof (struct pollfd) * (nhandlers + 1));
  if (!fds)
    {
      fprintf (stderr, "FATAL: Out of memory.\n");
      exit (1);
    }
 
  nfds = 0;
  fds[nfds].fd = server_fd;
  fds[nfds].events = POLLIN;
  fds[nfds++].revents = 0;
 
  for (t = vapi_handler; t; t = t->next)
    {
      if (t->fd)
        {
          t->temp = nfds;
          fds[nfds].fd = t->fd;
          fds[nfds].events = POLLIN;
          fds[nfds++].revents = 0;
        }
      else
        t->temp = -1;
    }
}
 
/* Determines whether a certain handler handles an ID */
static int
handler_fits_id (const struct vapi_handler *t, unsigned long id)
{
  return ((id >= t->base_id) && (id < t->base_id + t->num_ids));
}
 
/* Finds a handler with given ID, return it, NULL if not found.  */
static struct vapi_handler *
find_handler (unsigned long id)
{
  struct vapi_handler *t = vapi_handler;
  while (t && !handler_fits_id (t, id))
    t = t->next;
  return t;
}
 
/* Adds a handler with given id and returns it.  */
static struct vapi_handler *
add_handler (unsigned long base_id, unsigned long num_ids)
{
  struct vapi_handler **t = &vapi_handler;
  struct vapi_handler *tt;
  while ((*t))
    t = &(*t)->next;
  tt = (struct vapi_handler *) malloc (sizeof (struct vapi_handler));
  tt->next = NULL;
  tt->base_id = base_id;
  tt->num_ids = num_ids;
  tt->read_func = NULL;
  tt->priv_dat = NULL;
  tt->fd = 0;
  (*t) = tt;
  free (fds);
  fds = NULL;
  nhandlers++;
  rebuild_fds ();
  return tt;
}
 
void
vapi_write_log_file (VAPI_COMMAND command, unsigned long devid,
                     unsigned long data)
{
  if (!runtime.vapi.vapi_file)
    return;
  if (!config.vapi.hide_device_id && devid <= VAPI_MAX_DEVID)
    fprintf (runtime.vapi.vapi_file, "%04lx", devid);
  fprintf (runtime.vapi.vapi_file, "%1x%08lx\n", command, data);
}
 
static int
vapi_write_stream (int fd, void *buf, int len)
{
  int n;
  char *w_buf = (char *) buf;
  struct pollfd block;
 
  while (len)
    {
      if ((n = write (fd, w_buf, len)) < 0)
        {
          switch (errno)
            {
            case EWOULDBLOCK:   /* or EAGAIN */
              /* We've been called on a descriptor marked
                 for nonblocking I/O. We better simulate
                 blocking behavior. */
              block.fd = fd;
              block.events = POLLOUT;
              block.revents = 0;
              poll (&block, 1, -1);
              continue;
            case EINTR:
              continue;
            case EPIPE:
              close (fd);
              fd = 0;
              return -1;
            default:
              return -1;
            }
        }
      else
        {
          len -= n;
          w_buf += n;
        }
    }
  return 0;
}
 
static int
vapi_read_stream (int fd, void *buf, int len)
{
  int n;
  char *r_buf = (char *) buf;
  struct pollfd block;
 
  while (len)
    {
      if ((n = read (fd, r_buf, len)) < 0)
        {
          switch (errno)
            {
            case EWOULDBLOCK:   /* or EAGAIN */
              /* We've been called on a descriptor marked
                 for nonblocking I/O. We better simulate
                 blocking behavior. */
              block.fd = fd;
              block.events = POLLIN;
              block.revents = 0;
              poll (&block, 1, -1);
              continue;
            case EINTR:
              continue;
            default:
              return -1;
            }
        }
      else if (n == 0)
        {
          close (fd);
          fd = 0;
          return -1;
        }
      else
        {
          len -= n;
          r_buf += n;
        }
    }
  return 0;
}
 
/* Added by CZ 24/05/01 */
int
get_server_socket (const char *name, const char *proto, int port)
{
  struct servent *service;
  struct protoent *protocol;
  struct sockaddr_in sa;
  struct hostent *hp;
  int sockfd;
  socklen_t len;
  char myname[256];
  int flags;
  char sTemp[256];
 
  /* First, get the protocol number of TCP */
  if (!(protocol = getprotobyname (proto)))
    {
      sprintf (sTemp, "Unable to load protocol \"%s\"", proto);
      perror (sTemp);
      return 0;
    }
  tcp_level = protocol->p_proto;        /* Save for later */
 
  /* If we weren't passed a non standard port, get the port
     from the services directory. */
  if (!port)
    {
      if ((service = getservbyname (name, protocol->p_name)))
        port = ntohs (service->s_port);
    }
 
  /* Create the socket using the TCP protocol */
  if ((sockfd = socket (PF_INET, SOCK_STREAM, protocol->p_proto)) < 0)
    {
      perror ("Unable to create socket");
      return 0;
    }
 
  flags = 1;
  if (setsockopt
      (sockfd, SOL_SOCKET, SO_REUSEADDR, (const char *) &flags,
       sizeof (int)) < 0)
    {
      sprintf (sTemp, "Can not set SO_REUSEADDR option on socket %d", sockfd);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
 
  /* The server should also be non blocking. Get the current flags. */
  if (fcntl (sockfd, F_GETFL, &flags) < 0)
    {
      sprintf (sTemp, "Unable to get flags for socket %d", sockfd);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
 
  /* Set the nonblocking flag */
  if (fcntl (sockfd, F_SETFL, flags | O_NONBLOCK) < 0)
    {
      sprintf (sTemp, "Unable to set flags for socket %d to value 0x%08x",
               sockfd, flags | O_NONBLOCK);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
 
  /* Find out what our address is */
  memset (&sa, 0, sizeof (struct sockaddr_in));
  gethostname (myname, sizeof (myname));
  if (!(hp = gethostbyname (myname)))
    {
      perror ("Unable to read hostname");
      close (sockfd);
      return 0;
    }
 
  /* Bind our socket to the appropriate address */
  sa.sin_family = hp->h_addrtype;
  sa.sin_port = htons (port);
  if (bind (sockfd, (struct sockaddr *) &sa, sizeof (struct sockaddr_in)) < 0)
    {
      sprintf (sTemp, "Unable to bind socket %d to port %d", sockfd, port);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
  serverIP = sa.sin_addr.s_addr;
  len = sizeof (struct sockaddr_in);
  if (getsockname (sockfd, (struct sockaddr *) &sa, &len) < 0)
    {
      sprintf (sTemp, "Unable to get socket information for socket %d",
               sockfd);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
  runtime.vapi.server_port = ntohs (sa.sin_port);
 
  /* Set the backlog to 1 connections */
  if (listen (sockfd, 1) < 0)
    {
      sprintf (sTemp, "Unable to set backlog on socket %d to %d", sockfd, 1);
      perror (sTemp);
      close (sockfd);
      return 0;
    }
 
  return sockfd;
}
 
static void
server_request ()
{
  struct sockaddr_in sa;
  struct sockaddr *addr = (struct sockaddr *) &sa;
  socklen_t len = sizeof (struct sockaddr_in);
  int fd = accept (server_fd, addr, &len);
  int on_off = 0;                /* Turn off Nagel's algorithm on the socket */
  int flags;
  char sTemp[256];
 
  if (fd < 0)
    {
      /* This is valid, because a connection could have started,
         and then terminated due to a protocol error or user
         initiation before the accept could take place. */
      if (errno != EWOULDBLOCK && errno != EAGAIN)
        {
          perror ("accept");
          close (server_fd);
          server_fd = 0;
          runtime.vapi.enabled = 0;
          serverIP = 0;
        }
      return;
    }
 
  if (fcntl (fd, F_GETFL, &flags) < 0)
    {
      sprintf (sTemp, "Unable to get flags for vapi socket %d", fd);
      perror (sTemp);
      close (fd);
      return;
    }
 
  if (fcntl (fd, F_SETFL, flags | O_NONBLOCK) < 0)
    {
      sprintf (sTemp,
               "Unable to set flags for vapi socket %d to value 0x%08x", fd,
               flags | O_NONBLOCK);
      perror (sTemp);
      close (fd);
      return;
    }
 
  if (setsockopt (fd, tcp_level, TCP_NODELAY, &on_off, sizeof (int)) < 0)
    {
      sprintf (sTemp,
               "Unable to disable Nagel's algorithm for socket %d.\nsetsockopt",
               fd);
      perror (sTemp);
      close (fd);
      return;
    }
 
  /* Install new handler */
  {
    unsigned long id;
    struct vapi_handler *t;
    if (vapi_read_stream (fd, &id, sizeof (id)))
      {
        perror ("Cannot get id");
        close (fd);
        return;
      }
    t = find_handler (id);
    if (t)
      {
        if (t->fd)
          {
            fprintf (stderr,
                     "WARNING: Test with id %lx already connected. Ignoring.\n",
                     id);
            close (fd);
            return;
          }
        else
          {
            t->fd = fd;
            rebuild_fds ();
          }
      }
    else
      {
        fprintf (stderr,
                 "WARNING: Test with id %lx not registered. Ignoring.\n", id);
        close (fd);             /* kill the connection */
        return;
      }
    if (config.sim.verbose)
      PRINTF ("\nConnection with test (id %lx) established.\n", id);
  }
}
 
static int
write_packet (unsigned long id, unsigned long data)
{
  struct vapi_handler *t = find_handler (id);
  if (!t || !t->fd)
    return 1;
  id = htonl (id);
  if (vapi_write_stream (t->fd, &id, sizeof (id)) < 0)
    return 1;
  data = htonl (data);
  if (vapi_write_stream (t->fd, &data, sizeof (data)) < 0)
    return 1;
  return 0;
}
 
static int
read_packet (int fd, unsigned long *id, unsigned long *data)
{
  if (fd <= 0)
    return 1;
  if (vapi_read_stream (fd, id, sizeof (unsigned long)) < 0)
    return 1;
  *id = ntohl (*id);
  if (vapi_read_stream (fd, data, sizeof (unsigned long)) < 0)
    return 1;
  *data = ntohl (*data);
  return 0;
}
 
static void
vapi_request (struct vapi_handler *t)
{
  unsigned long id, data;
 
  if (read_packet (t->fd, &id, &data))
    {
      if (t->fd > 0)
        {
          perror ("vapi read");
          close (t->fd);
          t->fd = 0;
          rebuild_fds ();
        }
      return;
    }
 
  vapi_write_log_file (VAPI_COMMAND_REQUEST, id, data);
 
  /* This packet may be for another handler */
  if (!handler_fits_id (t, id))
    t = find_handler (id);
  if (!t || !t->read_func)
    fprintf (stderr,
             "WARNING: Received packet for undefined id %08lx, data %08lx\n",
             id, data);
  else
    t->read_func (id, data, t->priv_dat);
}
 
void
vapi_check ()
{
  struct vapi_handler *t;
 
  if (!server_fd || !fds)
    {
      fprintf (stderr, "FATAL: Unable to maintain VAPI server.\n");
      exit (1);
    }
 
  /* Handle everything in queue. */
  while (1)
    {
      switch (poll (fds, nfds, 0))
        {
        case -1:
          if (errno == EINTR)
            continue;
          perror ("poll");
          if (server_fd)
            close (server_fd);
          runtime.vapi.enabled = 0;
          serverIP = 0;
          return;
        case 0:          /* Nothing interesting going on */
          return;
        default:
          /* Handle the vapi ports first. */
          for (t = vapi_handler; t; t = t->next)
            if (t->temp >= 0 && fds[t->temp].revents)
              vapi_request (t);
 
          if (fds[0].revents)
            {
              if (fds[0].revents & POLLIN)
                server_request ();
              else
                {               /* Error Occurred */
                  fprintf (stderr,
                           "Received flags 0x%08x on server. Shutting down.\n",
                           fds[0].revents);
                  if (server_fd)
                    close (server_fd);
                  server_fd = 0;
                  runtime.vapi.enabled = 0;
                  serverIP = 0;
                }
            }
          break;
        }                       /* End of switch statement */
    }                           /* End of while statement */
}
 
/* Inits the VAPI, according to sim-config */
int
vapi_init ()
{
  nhandlers = 0;
  vapi_handler = NULL;
  if (!runtime.vapi.enabled)
    return 0;                    /* Nothing to do */
 
  runtime.vapi.server_port = config.vapi.server_port;
  if (!runtime.vapi.server_port)
    {
      fprintf (stderr, "WARNING: server_port = 0, shutting down VAPI\n");
      runtime.vapi.enabled = 0;
      return 1;
    }
  if ((server_fd =
       get_server_socket ("or1ksim", "tcp", runtime.vapi.server_port)))
    PRINTF ("VAPI Server started on port %d\n", runtime.vapi.server_port);
  else
    {
      perror ("Connection");
      return 1;
    }
 
  rebuild_fds ();
 
  if ((runtime.vapi.vapi_file = fopen (config.vapi.vapi_fn, "wt+")) == NULL)
    fprintf (stderr, "WARNING: cannot open VAPI log file\n");
 
  return 0;
}
 
/* Closes the VAPI */
void
vapi_done ()
{
  int i;
  struct vapi_handler *t = vapi_handler;
 
  for (i = 0; i < nfds; i++)
    if (fds[i].fd)
      close (fds[i].fd);
  server_fd = 0;
  runtime.vapi.enabled = 0;
  serverIP = 0;
  free (fds);
  fds = 0;
  if (runtime.vapi.vapi_file)
    {
      /* Mark end of simulation */
      vapi_write_log_file (VAPI_COMMAND_END, 0, 0);
      fclose (runtime.vapi.vapi_file);
    }
 
  while (vapi_handler)
    {
      t = vapi_handler;
      vapi_handler = vapi_handler->next;
      free (t);
    }
}
 
/* Installs a vapi handler for one VAPI id */
void
vapi_install_handler (unsigned long id,
                      void (*read_func) (unsigned long, unsigned long,
                                         void *), void *dat)
{
  vapi_install_multi_handler (id, 1, read_func, dat);
}
 
/* Installs a vapi handler for many VAPI id */
void
vapi_install_multi_handler (unsigned long base_id, unsigned long num_ids,
                            void (*read_func) (unsigned long, unsigned long,
                                               void *), void *dat)
{
  struct vapi_handler *tt;
 
  if (read_func == NULL)
    {
      struct vapi_handler **t = &vapi_handler;
      while ((*t) && !handler_fits_id (*t, base_id))
        t = &(*t)->next;
      if (!t)
        {
          fprintf (stderr, "Cannot uninstall VAPI read handler from id %lx\n",
                   base_id);
          exit (1);
        }
      tt = *t;
      (*t) = (*t)->next;
      free (tt);
      nhandlers--;
    }
  else
    {
      tt = find_handler (base_id);
      if (!tt)
        {
          tt = add_handler (base_id, num_ids);
          tt->read_func = read_func;
          tt->priv_dat = dat;
        }
      else
        {
          tt->read_func = read_func;
          tt->priv_dat = dat;
          rebuild_fds ();
        }
    }
}
 
/* Returns number of unconnected handles.  */
int
vapi_num_unconnected (int printout)
{
  struct vapi_handler *t = vapi_handler;
  int numu = 0;
  for (; t; t = t->next)
    {
      if (!t->fd)
        {
          numu++;
          if (printout)
            {
              if (t->num_ids == 1)
                PRINTF (" 0x%lx", t->base_id);
              else
                PRINTF (" 0x%lx..0x%lx", t->base_id,
                        t->base_id + t->num_ids - 1);
            }
        }
    }
  return numu;
}
 
/* Sends a packet to specified test */
void
vapi_send (unsigned long id, unsigned long data)
{
  vapi_write_log_file (VAPI_COMMAND_SEND, id, data);
  write_packet (id, data);
}
 
/*
int main ()
{
  runtime.vapi.enabled = 1;
  config.vapi.server_port = 9999;
  vapi_init ();
  while (1) {
    vapi_check();
    usleep(1);
  }
  vapi_done ();
}*/
 
/*---------------------------------------------------[ VAPI configuration ]---*/
 
static void
vapi_enabled (union param_val val, void *dat)
{
  config.vapi.enabled = val.int_val;
}
 
 
/*---------------------------------------------------------------------------*/
/*!Set the VAPI server port
 
   Ensure the value chosen is valid
 
   @param[in] val  The value to use
   @param[in] dat  The config data structure (not used here)                 */
/*---------------------------------------------------------------------------*/
static void
vapi_server_port (union param_val val, void *dat)
{
  if ((val.int_val < 1) || (val.int_val > 65535))
    {
      fprintf (stderr, "Warning: invalid VAPI port specified: ignored\n");
    }
  else
    {
      config.vapi.server_port = val.int_val;
    }
}       /* vapi_server_port() */
 
 
static void
vapi_log_enabled (union param_val val, void *dat)
{
  config.vapi.log_enabled = val.int_val;
}
 
static void
vapi_hide_device_id (union param_val val, void *dat)
{
  config.vapi.hide_device_id = val.int_val;
}
 
 
/*---------------------------------------------------------------------------*/
/*!Set the log file
 
   Free any existing string.
 
   @param[in] val  The value to use
   @param[in] dat  The config data structure (not used here)                 */
/*---------------------------------------------------------------------------*/
static void
vapi_log_fn (union param_val  val,
             void            *dat)
{
  if (NULL != config.vapi.vapi_fn)
    {
      free (config.vapi.vapi_fn);
    }
 
  config.vapi.vapi_fn = strdup (val.str_val);
 
}       /* vapi_log_fn() */
 
 
void
reg_vapi_sec (void)
{
  struct config_section *sec = reg_config_sec ("VAPI", NULL, NULL);
 
  reg_config_param (sec, "enabled", paramt_int, vapi_enabled);
  reg_config_param (sec, "server_port", paramt_int, vapi_server_port);
  reg_config_param (sec, "log_enabled", paramt_int, vapi_log_enabled);
  reg_config_param (sec, "hide_device_id", paramt_int, vapi_hide_device_id);
  reg_config_param (sec, "vapi_log_file", paramt_str, vapi_log_fn);
  reg_config_param (sec, "vapi_log_fn", paramt_str, vapi_log_fn);
}

Line No.	Rev	Author	Line
1	293	markom	`/* vapi.c -- Verification API Interface`
2	1748	jeremybenn
3	293	markom	`Copyright (C) 2001, Marko Mlinar, markom@opencores.org`
4	1748	jeremybenn	`Copyright (C) 2008 Embecosm Limited`
5
6			`Contributor Jeremy Bennett <jeremy.bennett@embecosm.com>`
7
8			`This file is part of OpenRISC 1000 Architectural Simulator.`
9
10			`This program is free software; you can redistribute it and/or modify it`
11			`under the terms of the GNU General Public License as published by the Free`
12			`Software Foundation; either version 3 of the License, or (at your option)`
13			`any later version.`
14
15			`This program is distributed in the hope that it will be useful, but WITHOUT`
16			`ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or`
17			`FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for`
18			`more details.`
19
20			`You should have received a copy of the GNU General Public License along`
21			`with this program. If not, see <http://www.gnu.org/licenses/>. */`
22	293	markom
23	1748	jeremybenn	`/* This program is commented throughout in a fashion suitable for processing`
24			`with Doxygen. */`
25	293	markom
26
27	1748	jeremybenn	`/* Autoconf and/or portability configuration */`
28	336	markom	`#include "config.h"`
29	1748	jeremybenn	`#include "port.h"`
30	293	markom
31	1748	jeremybenn	`/* System includes */`
32	293	markom	`#include <stdlib.h>`
33			`#include <unistd.h>`
34	1748	jeremybenn	`#include <stdio.h>`
35	293	markom	`#include <errno.h>`
36			`#include <sys/poll.h>`
37	1751	jeremybenn	`#include <sys/socket.h>`
38			`#include <netinet/in.h>`
39	293	markom	`#include <fcntl.h>`
40			`#include <netdb.h>`
41			`#include <netinet/tcp.h>`
42
43	1748	jeremybenn	`/* Package includes */`
44	336	markom	`#include "sim-config.h"`
45	439	erez	`#include "vapi.h"`
46	293	markom
47	1492	nogj
48	304	markom	`static unsigned int serverIP = 0;`
49	293	markom
50	304	markom	`static unsigned int server_fd = 0;`
51			`static unsigned int nhandlers = 0;`
52
53	293	markom	`static int tcp_level = 0;`
54
55	1748	jeremybenn	`static struct vapi_handler`
56			`{`
57	304	markom	`int fd;`
58	477	erez	`unsigned long base_id, num_ids;`
59	1748	jeremybenn	`void (read_func) (unsigned long, unsigned long, void );`
60	1366	nogj	`void *priv_dat;`
61	293	markom	`struct vapi_handler *next;`
62	304	markom	`int temp;`
63	442	erez	`} *vapi_handler = NULL;`
64	293	markom
65	304	markom	`/* Structure for polling, it is cached, that it doesn't have to be rebuilt each time */`
66			`static struct pollfd *fds = NULL;`
67			`static int nfds = 0;`
68
69			`/* Rebuilds the fds structures; see fds. */`
70	1748	jeremybenn	`void`
71			`rebuild_fds ()`
72			`{`
73	304	markom	`struct vapi_handler *t;`
74			`if (fds)`
75			`free (fds);`
76			`fds = (struct pollfd ) malloc (sizeof (struct pollfd) (nhandlers + 1));`
77	1748	jeremybenn	`if (!fds)`
78			`{`
79			`fprintf (stderr, "FATAL: Out of memory.\n");`
80			`exit (1);`
81			`}`
82
83	304	markom	`nfds = 0;`
84			`fds[nfds].fd = server_fd;`
85			`fds[nfds].events = POLLIN;`
86			`fds[nfds++].revents = 0;`
87
88	1748	jeremybenn	`for (t = vapi_handler; t; t = t->next)`
89			`{`
90			`if (t->fd)`
91			`{`
92			`t->temp = nfds;`
93			`fds[nfds].fd = t->fd;`
94			`fds[nfds].events = POLLIN;`
95			`fds[nfds++].revents = 0;`
96			`}`
97			`else`
98			`t->temp = -1;`
99			`}`
100	304	markom	`}`
101
102	477	erez	`/* Determines whether a certain handler handles an ID */`
103	1748	jeremybenn	`static int`
104			`handler_fits_id (const struct vapi_handler *t, unsigned long id)`
105			`{`
106	477	erez	`return ((id >= t->base_id) && (id < t->base_id + t->num_ids));`
107			`}`
108
109	304	markom	`/* Finds a handler with given ID, return it, NULL if not found. */`
110	1748	jeremybenn	`static struct vapi_handler *`
111			`find_handler (unsigned long id)`
112			`{`
113	304	markom	`struct vapi_handler *t = vapi_handler;`
114	477	erez	`while (t && !handler_fits_id (t, id))`
115	304	markom	`t = t->next;`
116			`return t;`
117			`}`
118
119			`/* Adds a handler with given id and returns it. */`
120	1748	jeremybenn	`static struct vapi_handler *`
121			`add_handler (unsigned long base_id, unsigned long num_ids)`
122			`{`
123	304	markom	`struct vapi_handler **t = &vapi_handler;`
124			`struct vapi_handler *tt;`
125			`while ((*t))`
126			`t = &(*t)->next;`
127	1748	jeremybenn	`tt = (struct vapi_handler *) malloc (sizeof (struct vapi_handler));`
128	304	markom	`tt->next = NULL;`
129	477	erez	`tt->base_id = base_id;`
130			`tt->num_ids = num_ids;`
131	304	markom	`tt->read_func = NULL;`
132	1366	nogj	`tt->priv_dat = NULL;`
133	304	markom	`tt->fd = 0;`
134			`(*t) = tt;`
135			`free (fds);`
136			`fds = NULL;`
137			`nhandlers++;`
138			`rebuild_fds ();`
139			`return tt;`
140			`}`
141
142	1748	jeremybenn	`void`
143			`vapi_write_log_file (VAPI_COMMAND command, unsigned long devid,`
144			`unsigned long data)`
145	419	erez	`{`
146	439	erez	`if (!runtime.vapi.vapi_file)`
147			`return;`
148	477	erez	`if (!config.vapi.hide_device_id && devid <= VAPI_MAX_DEVID)`
149	1308	phoenix	`fprintf (runtime.vapi.vapi_file, "%04lx", devid);`
150			`fprintf (runtime.vapi.vapi_file, "%1x%08lx\n", command, data);`
151	419	erez	`}`
152
153	1748	jeremybenn	`static int`
154			`vapi_write_stream (int fd, void *buf, int len)`
155	304	markom	`{`
156			`int n;`
157	1748	jeremybenn	`char w_buf = (char ) buf;`
158	304	markom	`struct pollfd block;`
159
160	1748	jeremybenn	`while (len)`
161			`{`
162			`if ((n = write (fd, w_buf, len)) < 0)`
163			`{`
164			`switch (errno)`
165			`{`
166			`case EWOULDBLOCK: /* or EAGAIN */`
167			`/* We've been called on a descriptor marked`
168			`for nonblocking I/O. We better simulate`
169			`blocking behavior. */`
170			`block.fd = fd;`
171			`block.events = POLLOUT;`
172			`block.revents = 0;`
173			`poll (&block, 1, -1);`
174			`continue;`
175			`case EINTR:`
176			`continue;`
177			`case EPIPE:`
178			`close (fd);`
179			`fd = 0;`
180			`return -1;`
181			`default:`
182			`return -1;`
183			`}`
184			`}`
185			`else`
186			`{`
187			`len -= n;`
188			`w_buf += n;`
189			`}`
190	304	markom	`}`
191			`return 0;`
192			`}`
193
194	1748	jeremybenn	`static int`
195			`vapi_read_stream (int fd, void *buf, int len)`
196	304	markom	`{`
197			`int n;`
198	1748	jeremybenn	`char r_buf = (char ) buf;`
199	304	markom	`struct pollfd block;`
200
201	1748	jeremybenn	`while (len)`
202			`{`
203			`if ((n = read (fd, r_buf, len)) < 0)`
204			`{`
205			`switch (errno)`
206			`{`
207			`case EWOULDBLOCK: /* or EAGAIN */`
208			`/* We've been called on a descriptor marked`
209			`for nonblocking I/O. We better simulate`
210			`blocking behavior. */`
211			`block.fd = fd;`
212			`block.events = POLLIN;`
213			`block.revents = 0;`
214			`poll (&block, 1, -1);`
215			`continue;`
216			`case EINTR:`
217			`continue;`
218			`default:`
219			`return -1;`
220			`}`
221			`}`
222			`else if (n == 0)`
223			`{`
224			`close (fd);`
225			`fd = 0;`
226			`return -1;`
227			`}`
228			`else`
229			`{`
230			`len -= n;`
231			`r_buf += n;`
232			`}`
233	304	markom	`}`
234			`return 0;`
235			`}`
236
237	293	markom	`/* Added by CZ 24/05/01 */`
238	1748	jeremybenn	`int`
239			`get_server_socket (const char name, const char proto, int port)`
240	293	markom	`{`
241			`struct servent *service;`
242			`struct protoent *protocol;`
243			`struct sockaddr_in sa;`
244	1748	jeremybenn	`struct hostent *hp;`
245	293	markom	`int sockfd;`
246	1557	nogj	`socklen_t len;`
247	293	markom	`char myname[256];`
248			`int flags;`
249			`char sTemp[256];`
250
251			`/* First, get the protocol number of TCP */`
252	1748	jeremybenn	`if (!(protocol = getprotobyname (proto)))`
253			`{`
254			`sprintf (sTemp, "Unable to load protocol \"%s\"", proto);`
255			`perror (sTemp);`
256			`return 0;`
257			`}`
258			`tcp_level = protocol->p_proto; /* Save for later */`
259	293	markom
260			`/* If we weren't passed a non standard port, get the port`
261			`from the services directory. */`
262	1748	jeremybenn	`if (!port)`
263			`{`
264			`if ((service = getservbyname (name, protocol->p_name)))`
265			`port = ntohs (service->s_port);`
266			`}`
267	336	markom
268	293	markom	`/* Create the socket using the TCP protocol */`
269	1748	jeremybenn	`if ((sockfd = socket (PF_INET, SOCK_STREAM, protocol->p_proto)) < 0)`
270			`{`
271			`perror ("Unable to create socket");`
272			`return 0;`
273			`}`
274	336	markom
275	293	markom	`flags = 1;`
276	1748	jeremybenn	`if (setsockopt`
277			`(sockfd, SOL_SOCKET, SO_REUSEADDR, (const char *) &flags,`
278			`sizeof (int)) < 0)`
279			`{`
280			`sprintf (sTemp, "Can not set SO_REUSEADDR option on socket %d", sockfd);`
281			`perror (sTemp);`
282			`close (sockfd);`
283			`return 0;`
284			`}`
285	293	markom
286			`/* The server should also be non blocking. Get the current flags. */`
287	1748	jeremybenn	`if (fcntl (sockfd, F_GETFL, &flags) < 0)`
288			`{`
289			`sprintf (sTemp, "Unable to get flags for socket %d", sockfd);`
290			`perror (sTemp);`
291			`close (sockfd);`
292			`return 0;`
293			`}`
294	293	markom
295			`/* Set the nonblocking flag */`
296	1748	jeremybenn	`if (fcntl (sockfd, F_SETFL, flags \| O_NONBLOCK) < 0)`
297			`{`
298			`sprintf (sTemp, "Unable to set flags for socket %d to value 0x%08x",`
299			`sockfd, flags \| O_NONBLOCK);`
300			`perror (sTemp);`
301			`close (sockfd);`
302			`return 0;`
303			`}`
304	293	markom
305			`/* Find out what our address is */`
306	1748	jeremybenn	`memset (&sa, 0, sizeof (struct sockaddr_in));`
307			`gethostname (myname, sizeof (myname));`
308			`if (!(hp = gethostbyname (myname)))`
309			`{`
310			`perror ("Unable to read hostname");`
311			`close (sockfd);`
312			`return 0;`
313			`}`
314
315	293	markom	`/* Bind our socket to the appropriate address */`
316			`sa.sin_family = hp->h_addrtype;`
317	1748	jeremybenn	`sa.sin_port = htons (port);`
318			`if (bind (sockfd, (struct sockaddr *) &sa, sizeof (struct sockaddr_in)) < 0)`
319			`{`
320			`sprintf (sTemp, "Unable to bind socket %d to port %d", sockfd, port);`
321			`perror (sTemp);`
322			`close (sockfd);`
323			`return 0;`
324			`}`
325	293	markom	`serverIP = sa.sin_addr.s_addr;`
326	1748	jeremybenn	`len = sizeof (struct sockaddr_in);`
327			`if (getsockname (sockfd, (struct sockaddr *) &sa, &len) < 0)`
328			`{`
329			`sprintf (sTemp, "Unable to get socket information for socket %d",`
330			`sockfd);`
331			`perror (sTemp);`
332			`close (sockfd);`
333			`return 0;`
334			`}`
335			`runtime.vapi.server_port = ntohs (sa.sin_port);`
336	293	markom
337			`/* Set the backlog to 1 connections */`
338	1748	jeremybenn	`if (listen (sockfd, 1) < 0)`
339			`{`
340			`sprintf (sTemp, "Unable to set backlog on socket %d to %d", sockfd, 1);`
341			`perror (sTemp);`
342			`close (sockfd);`
343			`return 0;`
344			`}`
345	293	markom
346			`return sockfd;`
347			`}`
348
349	1748	jeremybenn	`static void`
350			`server_request ()`
351	293	markom	`{`
352			`struct sockaddr_in sa;`
353	1748	jeremybenn	`struct sockaddr addr = (struct sockaddr ) &sa;`
354			`socklen_t len = sizeof (struct sockaddr_in);`
355			`int fd = accept (server_fd, addr, &len);`
356			`int on_off = 0; /* Turn off Nagel's algorithm on the socket */`
357	293	markom	`int flags;`
358			`char sTemp[256];`
359
360	1748	jeremybenn	`if (fd < 0)`
361			`{`
362			`/* This is valid, because a connection could have started,`
363			`and then terminated due to a protocol error or user`
364			`initiation before the accept could take place. */`
365			`if (errno != EWOULDBLOCK && errno != EAGAIN)`
366			`{`
367			`perror ("accept");`
368			`close (server_fd);`
369			`server_fd = 0;`
370			`runtime.vapi.enabled = 0;`
371			`serverIP = 0;`
372			`}`
373			`return;`
374	336	markom	`}`
375	293	markom
376	1748	jeremybenn	`if (fcntl (fd, F_GETFL, &flags) < 0)`
377			`{`
378			`sprintf (sTemp, "Unable to get flags for vapi socket %d", fd);`
379			`perror (sTemp);`
380			`close (fd);`
381			`return;`
382			`}`
383	293	markom
384	1748	jeremybenn	`if (fcntl (fd, F_SETFL, flags \| O_NONBLOCK) < 0)`
385			`{`
386			`sprintf (sTemp,`
387			`"Unable to set flags for vapi socket %d to value 0x%08x", fd,`
388			`flags \| O_NONBLOCK);`
389			`perror (sTemp);`
390			`close (fd);`
391			`return;`
392			`}`
393
394			`if (setsockopt (fd, tcp_level, TCP_NODELAY, &on_off, sizeof (int)) < 0)`
395			`{`
396			`sprintf (sTemp,`
397			`"Unable to disable Nagel's algorithm for socket %d.\nsetsockopt",`
398			`fd);`
399			`perror (sTemp);`
400			`close (fd);`
401			`return;`
402			`}`
403
404	304	markom	`/* Install new handler */`
405			`{`
406			`unsigned long id;`
407			`struct vapi_handler *t;`
408	1748	jeremybenn	`if (vapi_read_stream (fd, &id, sizeof (id)))`
409			`{`
410			`perror ("Cannot get id");`
411			`close (fd);`
412			`return;`
413			`}`
414	304	markom	`t = find_handler (id);`
415	1748	jeremybenn	`if (t)`
416			`{`
417			`if (t->fd)`
418			`{`
419			`fprintf (stderr,`
420			`"WARNING: Test with id %lx already connected. Ignoring.\n",`
421			`id);`
422			`close (fd);`
423			`return;`
424			`}`
425			`else`
426			`{`
427			`t->fd = fd;`
428			`rebuild_fds ();`
429			`}`
430	355	markom	`}`
431	1748	jeremybenn	`else`
432			`{`
433			`fprintf (stderr,`
434			`"WARNING: Test with id %lx not registered. Ignoring.\n", id);`
435			`close (fd); /* kill the connection */`
436			`return;`
437			`}`
438			`if (config.sim.verbose)`
439	1308	phoenix	`PRINTF ("\nConnection with test (id %lx) established.\n", id);`
440	293	markom	`}`
441			`}`
442
443	1748	jeremybenn	`static int`
444			`write_packet (unsigned long id, unsigned long data)`
445			`{`
446	304	markom	`struct vapi_handler *t = find_handler (id);`
447	1748	jeremybenn	`if (!t \|\| !t->fd)`
448	304	markom	`return 1;`
449	293	markom	`id = htonl (id);`
450	1748	jeremybenn	`if (vapi_write_stream (t->fd, &id, sizeof (id)) < 0)`
451	293	markom	`return 1;`
452			`data = htonl (data);`
453	1748	jeremybenn	`if (vapi_write_stream (t->fd, &data, sizeof (data)) < 0)`
454	293	markom	`return 1;`
455			`return 0;`
456	442	erez	`}`
457	293	markom
458	1748	jeremybenn	`static int`
459			`read_packet (int fd, unsigned long id, unsigned long data)`
460	442	erez	`{`
461	1748	jeremybenn	`if (fd <= 0)`
462	293	markom	`return 1;`
463	1748	jeremybenn	`if (vapi_read_stream (fd, id, sizeof (unsigned long)) < 0)`
464	293	markom	`return 1;`
465	442	erez	`id = ntohl (id);`
466	1748	jeremybenn	`if (vapi_read_stream (fd, data, sizeof (unsigned long)) < 0)`
467	304	markom	`return 1;`
468	442	erez	`data = ntohl (data);`
469	293	markom	`return 0;`
470	304	markom	`}`
471	293	markom
472	1748	jeremybenn	`static void`
473			`vapi_request (struct vapi_handler *t)`
474	442	erez	`{`
475			`unsigned long id, data;`
476	1748	jeremybenn
477			`if (read_packet (t->fd, &id, &data))`
478			`{`
479			`if (t->fd > 0)`
480			`{`
481			`perror ("vapi read");`
482			`close (t->fd);`
483			`t->fd = 0;`
484			`rebuild_fds ();`
485			`}`
486			`return;`
487	293	markom	`}`
488
489	442	erez	`vapi_write_log_file (VAPI_COMMAND_REQUEST, id, data);`
490
491			`/* This packet may be for another handler */`
492	477	erez	`if (!handler_fits_id (t, id))`
493	442	erez	`t = find_handler (id);`
494			`if (!t \|\| !t->read_func)`
495	1748	jeremybenn	`fprintf (stderr,`
496			`"WARNING: Received packet for undefined id %08lx, data %08lx\n",`
497			`id, data);`
498	293	markom	`else`
499	1748	jeremybenn	`t->read_func (id, data, t->priv_dat);`
500	293	markom	`}`

Browse

Tools

Subversion Repositories or1k

[/] [or1k/] [trunk/] [or1ksim/] [vapi/] [vapi.c] - Blame information for rev 1767