libmp/src/mptp/server.c

#include "../../include/error.h"
#include "../../include/mptp.h"
#include "../../include/util.h"

#include <errno.h>
#include <stdbool.h>
#include <string.h>
#include <sys/socket.h>

int lm_mptp_server_listen(char *addr, uint16_t port) {
  struct sockaddr saddr;
  int             sock;

  bzero(&saddr, sizeof(saddr));

  if ((sock = lm_mptp_socket(addr, port, &saddr)) < 0)
    return -1;

  if (bind(sock, &saddr, sizeof(struct sockaddr)) < 0) {
    lm_mptp_close(sock);
    lm_error_set(LM_ERR_MPTPBindFail, strerror(errno));
    return -1;
  }

  if(listen(sock, SOMAXCONN) < 0){
    lm_mptp_close(sock);
    lm_error_set(LM_ERR_MPTPListenFail, strerror(errno));
    return -1;
  }

  return sock;
}

int lm_mptp_server_accept(int sock, struct sockaddr *addr){
  socklen_t sl = sizeof(struct sockaddr);
  int s = -1;

  if((s = accept(sock, addr, &sl)) < 0){
    lm_error_set(LM_ERR_MPTPAcceptFail, strerror(errno));
    s = -1;
  }

  if(!lm_mptp_socket_opts(s)){
    close(s);
    s = -1;
  }

  return s;
}

void lm_mptp_server_close(int sock){
  close(sock); // you didn't see that comming, did you?
}

bool lm_mptp_server_verify(lm_mptp_t *packet) {
  if (!lm_mptp_verify(packet)){
    pdebug(__func__, "failed to verify the packet: %s", lm_strerror());
    return false;
  }

  if (!MPTP_IS_REQUEST(packet)) {
    pdebug(__func__, "MPTP packet is not request");
    lm_error_set(LM_ERR_MPTPNotRequest);
    return false;
  }

  if (!MPTP_IS_LAST(packet)) {
    pdebug(__func__, "MPTP packet is not the last");
    lm_error_set(LM_ERR_MPTPNotLast);
    return false;
  }

  return true;
}

bool lm_mptp_server_recv(int sock, lm_mptp_t *packet) {
  if (NULL == packet) {
    lm_error_set(LM_ERR_ArgNULL);
    return false;
  }

  lm_mptp_free(packet);

  if(!lm_mptp_recv(sock, packet)){
    pdebug(__func__, "failed to receive the packet: %s", lm_strerror());
    return false; // error set by function
  }

  return true;
}

bool lm_mptp_server_send(int sock, lm_mptp_t *packet) {
  if (NULL == packet) {
    lm_error_set(LM_ERR_ArgNULL);
    lm_mptp_free(packet);
    return false;
  }

  if (MPTP_FLAGS_VERSION(packet) != MPTP_VERSION_SUPPORTED) {
    lm_error_set(LM_ERR_MPTPBadVersion);
    lm_mptp_free(packet);
    return false;
  }

  if (packet->header.data_size > MPTP_DATA_MAX) {
    lm_error_set(LM_ERR_MPTPBadData);
    lm_mptp_free(packet);
    return false;
  }

  if (packet->header.host_size != 0) {
    lm_error_set(LM_ERR_MPTPBadHost);
    lm_mptp_free(packet);
    return false;
  }

  char      buffer[
    sizeof(packet->header) + 
    packet->header.host_size + 
    packet->header.path_size + 
    packet->header.data_size
  ];
  ssize_t   total = sizeof(buffer), used = 0, buflen = total;
  bool ret = false;

  copy_to_buffer(buffer, &packet->header, sizeof(packet->header), &total, &used);
  copy_to_buffer(buffer, packet->host, packet->header.host_size, &total, &used);
  copy_to_buffer(buffer, packet->path, packet->header.path_size, &total, &used);
  copy_to_buffer(buffer, packet->data, packet->header.data_size, &total, &used);
  
  packet->header.flags = htons(packet->header.flags);
  //packet->header.host_size = htons(packet->header.host_size);
  //packet->header.path_size = htons(packet->header.path_size);
  packet->header.data_size = htons(packet->header.data_size);

  if (send(sock, buffer, buflen, MSG_MORE) < 0) {
    lm_error_set(LM_ERR_MPTPSendFail);
    goto end;
  }

  pdebug(__func__, "printing the packet dump (%lu bytes)", buflen);
  pdebug_binary(buffer, buflen);
  ret = true;

end:
  lm_mptp_free(packet);
  return ret;
}