second code style cleanup: void/void * usage, proper #includes. try to

[moira.git] / lib / mr_param.c

/* $Id $
 *
 * Deal with mrgdb (bleah!)
 *
 * Copyright (C) 1987-1998 by the Massachusetts Institute of Technology
 * For copying and distribution information, please see the file
 * <mit-copyright.h>.
 *
 */

#include <mit-copyright.h>
#include <moira.h>
#include "mr_private.h"

#include <sys/types.h>
#include <netinet/in.h>

#include <stdlib.h>
#include <string.h>

RCSID("$Header$");

/*
 * GDB operations to send and recieve RPC requests and replies.
 */

/*
 * This doesn't get called until after the actual buffered write completes.
 * In a non-preflattening version of this, this would then queue the
 * write of the next bunch of data.
 */

int mr_cont_send(OPERATION op, HALF_CONNECTION hcon, struct mr_params *arg)
{
  op->result = OP_SUCCESS;
  free(arg->mr_flattened);
  arg->mr_flattened = NULL;

  return OP_COMPLETE;
}

int mr_start_send(OPERATION op, HALF_CONNECTION hcon,
		  struct mr_params *arg)
{
  int i, len;
  unsigned int mr_size;
  int *argl;
  char *buf, *bp;

  /*
   * This should probably be split into several routines.
   * It could also probably be made more efficient (punting most
   * of the argument marshalling stuff) by doing I/O directly
   * from the strings.  Anyone for a scatter/gather mr_send_data?
   *
   * that would look a lot like the uio stuff in the kernel..  hmm.
   */

  /*
   * Marshall the entire data right now..
   * We are sending the version number,
   * total request size, request number,
   * argument count, and then each argument.
   * At least for now, each argument is a string, which is
   * sent as a count of bytes followed by the bytes
   * (including the trailing '\0'), padded
   * to a 32-bit boundary.
   */

  mr_size = 4 * sizeof(int32);

  argl = malloc(sizeof(int) * arg->mr_argc);

  /*
   * For each argument, figure out how much space is needed.
   */

  for (i = 0; i < arg->mr_argc; ++i)
    {
      if (arg->mr_argl)
	argl[i] = len = arg->mr_argl[i];
      else
	argl[i] = len = strlen(arg->mr_argv[i]) + 1;
      mr_size += sizeof(int32) + len;
      /* Round up to next 32-bit boundary.. */
      mr_size = sizeof(int32) * howmany(mr_size, sizeof(int32));
    }

  arg->mr_flattened = buf = malloc(mr_size);

  memset(arg->mr_flattened, 0, mr_size);

  arg->mr_size = mr_size;

  ((int32 *)buf)[0] = htonl(mr_size);
  ((int32 *)buf)[1] = htonl(arg->mr_version_no);
  ((int32 *)buf)[2] = htonl(arg->mr_procno);
  ((int32 *)buf)[3] = htonl(arg->mr_argc);

  /*
   * bp is a pointer into the point in the buffer to put
   * the next argument.
   */

  bp = (char *)(((int32 *)buf) + 4);

  for (i = 0; i < arg->mr_argc; ++i)
    {
      len = argl[i];
      *((int32 *)bp) = htonl(len);
      bp += sizeof(int32);
      memcpy(bp, arg->mr_argv[i], len);
      bp += sizeof(int32) * howmany(len, sizeof(int32));
    }
  op->fcn.cont = mr_cont_send;
  arg->mr_size = mr_size;

  free(argl);

  if (gdb_send_data(hcon, arg->mr_flattened, mr_size) == OP_COMPLETE)
    return mr_cont_send(op, hcon, arg);
  else
    return OP_RUNNING;
}

int mr_cont_recv(OPERATION op, HALF_CONNECTION hcon, mr_params **argp)
{
  int done = FALSE;
  char *cp;
  int *ip;
  int i;
  mr_params *arg = *argp;

  while (!done)
    {
      switch (arg->mr_state)
	{
	case S_RECV_START:
	  arg->mr_state = S_RECV_DATA;
	  if (gdb_receive_data(hcon, (caddr_t)&arg->mr_size,
			       sizeof(int32)) == OP_COMPLETE)
	    continue;
	  done = TRUE;
	  break;
	case S_RECV_DATA:
	  fflush(stdout);
	  /* Should validate that length is reasonable */
	  arg->mr_size = ntohl(arg->mr_size);
	  if (arg->mr_size > 65536)
	    return OP_CANCELLED;
	  arg->mr_flattened = malloc(arg->mr_size);
	  arg->mr_state = S_DECODE_DATA;
	  memcpy(arg->mr_flattened, (caddr_t)&arg->mr_size, sizeof(int32));

	  if (gdb_receive_data(hcon, arg->mr_flattened + sizeof(int32),
			       arg->mr_size - sizeof(int32))
	      == OP_COMPLETE)
	    continue;
	  done = TRUE;
	  break;
	case S_DECODE_DATA:
	  cp = arg->mr_flattened;
	  ip = (int *) cp;
	  /* we already got the overall length.. */
	  for (i = 1; i < 4; i++)
	    ip[i] = ntohl(ip[i]);
	  arg->mr_version_no = ip[1];
	  if (arg->mr_version_no != MR_VERSION_1 &&
	      arg->mr_version_no != MR_VERSION_2)
	    arg->mr_status = MR_VERSION_MISMATCH;
	  else arg->mr_status = ip[2];
	  arg->mr_argc = ip[3];
	  cp += 4 * sizeof(int);
	  arg->mr_argv = malloc(arg->mr_argc * sizeof(char *));
	  arg->mr_argl = malloc(arg->mr_argc * sizeof(int));

	  for (i = 0; i < arg->mr_argc; ++i)
	    {
	      u_short nlen = ntohl(*(int *)cp);
	      cp += sizeof(int32);
	      if (cp + nlen > arg->mr_flattened + arg->mr_size)
		{
		  free(arg->mr_flattened);
		  arg->mr_flattened = NULL;
		  return OP_CANCELLED;
		}
	      arg->mr_argv[i] = malloc(nlen);
	      memcpy(arg->mr_argv[i], cp, nlen);
	      arg->mr_argl[i] = nlen;
	      cp += sizeof(int32) * howmany(nlen, sizeof(int32));
	    }
	  free(arg->mr_flattened);
	  arg->mr_flattened = NULL;
	  return OP_COMPLETE;
	}
    }
  return OP_RUNNING;
}


int mr_start_recv(OPERATION op, HALF_CONNECTION hcon, struct mr_params **argp)
{
  mr_params *arg = *argp;
  if (!arg)
    {
      *argp = arg = malloc(sizeof(mr_params));
      arg->mr_argl = NULL;
      arg->mr_argv = NULL;
      arg->mr_flattened = NULL;
    }
  arg->mr_state = S_RECV_START;
  op->fcn.cont = mr_cont_recv;
  return mr_cont_recv(op, hcon, argp);
}

void mr_destroy_reply(mr_params *reply)
{
  int i;
  if (reply)
    {
      if (reply->mr_argl)
	free(reply->mr_argl);
      reply->mr_argl = NULL;
      if (reply->mr_flattened)
	free(reply->mr_flattened);
      reply->mr_flattened = NULL;
      if (reply->mr_argv)
	{
	  for (i = 0; i < reply->mr_argc; i++)
	    {
	      if (reply->mr_argv[i])
		free (reply->mr_argv[i]);
	      reply->mr_argv[i] = NULL;
	    }
	  free(reply->mr_argv);
	}
      reply->mr_argv = NULL;
      free(reply);
    }
}