src/bin/lttng-relayd/cmd-generic.c

   1 /*
   2  * Copyright (C) 2013 - Julien Desfossez <jdesfossez@efficios.com>
   3  *                      David Goulet <dgoulet@efficios.com>
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License, version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc., 51
  16  * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #define _GNU_SOURCE
  20 #define _LGPL_SOURCE
  21 #include <assert.h>
  22
  23 #include <common/common.h>
  24
  25 #include "cmd-generic.h"
  26
  27 int cmd_recv(struct lttcomm_sock *sock, void *buf, size_t len)
  28 {
  29         int ret;
  30
  31         assert(sock);
  32         assert(buf);
  33
  34         ret = sock->ops->recvmsg(sock, buf, len, 0);
  35         if (ret < len) {
  36                 if (ret == 0) {
  37                         /* Orderly shutdown. Not necessary to print an error. */
  38                         DBG("Socket %d did an orderly shutdown", sock->fd);
  39                 } else {
  40                         ERR("Relay didn't receive valid add_stream struct size. "
  41                                         "Expected %zu, got %d", len, ret);
  42                 }
  43                 ret = -1;
  44                 goto error;
  45         }
  46
  47 error:
  48         return ret;
  49 }