source: TI05-delivery/trunk/src/bbftp-client-3.2.0/bbftpc/ndg_client.c @ 1395

/**
 * NDG python embedded bbftp daemon module.
 *
 * @author Stephen Pascoe
 *
 * Copyright (C) 2006 CCLRC & NERC
 *
 * This software may be distributed under the terms of the Q Public Licence, version 1.0 or later.
 *
 */

#include <ndg_client.h>
#include <stdio.h>

/*
 * Higher level message transfer functions.
 *
 */

/**
 * Send a message of variable length to the client.
 *
 * This function sends the length of the message in a header thus the server
 * does not need to know what length message to expect.
 *
 * @param buffer a pointer to the message buffer.
 * @param length the number of bytes to send.
 * @param logmessage is filled with the error message on error.
 * @return 0 if OK, -1 if error.
 */
int ndg_client_message_send(char *buffer, int length, char *logmessage) {
  char ctrl[NDG_MESSAGE_LEN];

  sprintf(ctrl, "NDG-msg: %i", length);
  if (bbftp_private_send(ctrl, NDG_MESSAGE_LEN, logmessage) == -1) {
    return -1;
  }

  if (bbftp_private_send(buffer, length, logmessage) == -1) {
    return -1;
  }

  return 0;
}

/**
 * Receive a message of variable length from the client.
 *
 * The message length is sent in a separate header.
 * The message is guaranteed to be NULL terminated.  This is done by
 * allocating length+1 bytes and filling them with \0.
 *
 * @param buffer is set to newly allocated message buffer.
 * @param length is set to the length of the message.
 * @param logmessage is filled with the logmessage on error.
 * @return 0 if OK, -1 if error.
 */
int ndg_client_message_recv(char **buffer, int *length, char *logmessage) {
  char ctrl[NDG_MESSAGE_LEN];

  if (bbftp_private_recv(ctrl, NDG_MESSAGE_LEN, logmessage) == -1) {
    return -1;
  }
  if (sscanf(ctrl, "NDG-msg: %i", length) != 1) {
    sprintf(logmessage, "ndg_message_recv ctrl error: %40s", ctrl);
    return -1;
  }

  if ((*buffer = (char *)calloc(*length+1, sizeof(char))) == NULL) {
    sprintf(logmessage, "ngd_message_recv malloc error");
    return -1;
  }

  if (bbftp_private_recv(*buffer, *length, logmessage) == -1) {
    free(*buffer); *buffer = NULL;
    return -1;
  }

  return 0;
}