Files @ 27eb6c6418f6
Branch filter:

Location: DistRen/src/common/protocol.c

binki
Handle resolving hostnames gracefully.
Fix support for hostname:port in remoteio.
Added timeout parameter to multiio_poll().
Fix stupid mistakes in distrenslave's read handler.
The server and client may not sit and play table tennis at last :-D.
/*
  Copyright 2010 Nathan Phillip Brink

  This file is a part of DistRen.

  DistRen is free software: you can redistribute it and/or modify
  it under the terms of the GNU Affero General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  DistRen 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 Affero General Public License for more details.

  You should have received a copy of the GNU Affero General Public License
  along with DistRen.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "protocol.h"
#include "remoteio.h"

#include <malloc.h>
#include <stdio.h>
#include <string.h>

#define DISTREN_REQUEST_MAGIC ((uint32_t)0x32423434)

int distren_request_new(struct distren_request **req, uint32_t len, enum distren_request_type type)
{
  struct distren_request *newreq;

  newreq = malloc(sizeof(struct distren_request));
  if(!newreq)
    {
      (*req) = NULL;
      return 1;
    }

  newreq->magic = DISTREN_REQUEST_MAGIC;
  newreq->len = len;
  newreq->type = type;

  (*req) = newreq;
  return 0;
}

int distren_request_send(struct remoteio *rem, struct distren_request *req, void *data)
{
  void *packet;
  size_t len;
  int write_err;

  if(req->magic != DISTREN_REQUEST_MAGIC)
    fprintf(stderr, "distren_request_send got a bad req\n");

  len = sizeof(struct distren_request) + req->len;

  packet = malloc(len);
  if(!packet)
    {
      fprintf(stderr, "Error allocating memory for packet\n");
      return 1;
    }
  memcpy(packet, req, sizeof(struct distren_request));
  memcpy(packet + sizeof(struct distren_request), data, req->len);

  write_err = remoteio_write(rem, packet, len);

  free(packet);

  return 0;
}

int distren_request_new_fromdata(struct distren_request **req, void *data, size_t len)
{
  struct distren_request *newreq;

#if 0
  size_t counter;
  uint32_t debugtmp;
#endif

  if(len < sizeof(struct distren_request))
    return 1;

  if( ((struct distren_request *)data)->magic != DISTREN_REQUEST_MAGIC )
    {
      fprintf(stderr, "packet doesn't match magic stuffs");
#if 0
      fputs("\n\tmagic=`", stderr);
      debugtmp = DISTREN_REQUEST_MAGIC;
      for(counter = 0; counter < sizeof(uint32_t); counter ++)
	putc(((char *)&debugtmp)[counter], stderr);
      fputs("'\n\t`", stderr);
      for(counter = 0; counter < sizeof(struct distren_request); counter ++)
	putc(((char *)data)[counter], stderr);
      fputs("'\n", stderr);
#endif
      return 1;
    }

  newreq = malloc(sizeof(struct distren_request));
  if(!newreq)
    {
      fprintf(stderr, "OOM\n");
      return 1;
    }

  memcpy(newreq, data, sizeof(struct distren_request));
  (*req) = newreq;
  return 0;
}

int distren_request_free(struct distren_request *req)
{
  free(req);
  return 0;
}