rabbitmq-c/tools/publish.c

// Copyright 2007 - 2021, Alan Antonuk and the rabbitmq-c contributors.
// SPDX-License-Identifier: mit

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

#include "common.h"

#define MAX_LINE_LENGTH 1024 * 32

static void do_publish(amqp_connection_state_t conn, char *exchange,
                       char *routing_key, amqp_basic_properties_t *props,
                       amqp_bytes_t body) {
  int res = amqp_basic_publish(conn, 1, cstring_bytes(exchange),
                               cstring_bytes(routing_key), 0, 0, props, body);
  die_amqp_error(res, "basic.publish");
}

int main(int argc, const char **argv) {
  amqp_connection_state_t conn;
  static char *exchange = NULL;
  static char *routing_key = NULL;
  static char *content_type = NULL;
  static char *content_encoding = NULL;
  static char **headers = NULL;
  static char *reply_to = NULL;
  static char *body = NULL;
  amqp_basic_properties_t props;
  amqp_bytes_t body_bytes;
  static int delivery = 1; /* non-persistent by default */
  static int line_buffered = 0;
  static char **pos;

  struct poptOption options[] = {
      INCLUDE_OPTIONS(connect_options),
      {"exchange", 'e', POPT_ARG_STRING, &exchange, 0,
       "the exchange to publish to", "exchange"},
      {"routing-key", 'r', POPT_ARG_STRING, &routing_key, 0,
       "the routing key to publish with", "routing key"},
      {"persistent", 'p', POPT_ARG_VAL, &delivery, 2,
       "use the persistent delivery mode", NULL},
      {"content-type", 'C', POPT_ARG_STRING, &content_type, 0,
       "the content-type for the message", "content type"},
      {"reply-to", 't', POPT_ARG_STRING, &reply_to, 0,
       "the replyTo to use for the message", "reply to"},
      {"line-buffered", 'l', POPT_ARG_VAL, &line_buffered, 2,
       "treat each line from standard in as a separate message", NULL},
      {"content-encoding", 'E', POPT_ARG_STRING, &content_encoding, 0,
       "the content-encoding for the message", "content encoding"},
      {"header", 'H', POPT_ARG_ARGV, &headers, 0,
       "set a message header (may be specified multiple times)",
       "\"key: value\""},
      {"body", 'b', POPT_ARG_STRING, &body, 0, "specify the message body",
       "body"},
      POPT_AUTOHELP{NULL, '\0', 0, NULL, 0, NULL, NULL}};

  process_all_options(argc, argv, options);

  if (!exchange && !routing_key) {
    fprintf(stderr, "neither exchange nor routing key specified\n");
    return 1;
  }

  memset(&props, 0, sizeof props);
  props._flags = AMQP_BASIC_DELIVERY_MODE_FLAG;
  props.delivery_mode = delivery;

  if (content_type) {
    props._flags |= AMQP_BASIC_CONTENT_TYPE_FLAG;
    props.content_type = amqp_cstring_bytes(content_type);
  }

  if (content_encoding) {
    props._flags |= AMQP_BASIC_CONTENT_ENCODING_FLAG;
    props.content_encoding = amqp_cstring_bytes(content_encoding);
  }

  if (reply_to) {
    props._flags |= AMQP_BASIC_REPLY_TO_FLAG;
    props.reply_to = amqp_cstring_bytes(reply_to);
  }

  if (headers) {
    int num = 0;
    for (pos = headers; *pos; pos++) {
      num++;
    }

    if (num > 0) {
      amqp_table_t *table = &props.headers;
      table->num_entries = num;
      table->entries = calloc(num, sizeof(amqp_table_entry_t));
      if (table->entries == NULL) {
        fprintf(stderr, "Memory allocation failed\n");
        return 1;
      }
      int i = 0;
      for (pos = headers; *pos; pos++) {
        char *colon = strchr(*pos, ':');
        if (colon) {
          *colon++ = '\0';
          while (*colon == ' ') colon++;
          table->entries[i].key = amqp_cstring_bytes(*pos);
          table->entries[i].value.kind = AMQP_FIELD_KIND_UTF8;
          table->entries[i].value.value.bytes = amqp_cstring_bytes(colon);
          i++;
        } else {
          fprintf(stderr,
                  "Ignored header definition missing ':' delimiter in \"%s\"\n",
                  *pos);
        }
      }
      props._flags |= AMQP_BASIC_HEADERS_FLAG;
    }
  }

  conn = make_connection();

  if (body) {
    body_bytes = amqp_cstring_bytes(body);
  } else {
    if (line_buffered) {
      body_bytes.bytes = (char *)malloc(MAX_LINE_LENGTH);
      if (body_bytes.bytes == NULL) {
        fprintf(stderr, "Memory allocation failed\n");
        return 1;
      }
      while (fgets(body_bytes.bytes, MAX_LINE_LENGTH, stdin)) {
        body_bytes.len = strlen(body_bytes.bytes);
        do_publish(conn, exchange, routing_key, &props, body_bytes);
      }
    } else {
      body_bytes = read_all(0);
    }
  }

  if (!line_buffered) {
    do_publish(conn, exchange, routing_key, &props, body_bytes);
  }

  if (props.headers.num_entries > 0) {
    free(props.headers.entries);
  }

  if (!body) {
    free(body_bytes.bytes);
  }

  close_connection(conn);
  return 0;
}