gotd/repo_imsg.c

   1 /*
   2  * Copyright (c) 2022 Stefan Sperling <stsp@openbsd.org>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include "got_compat.h"
  18
  19 #include <sys/types.h>
  20 #include <sys/queue.h>
  21 #include <sys/uio.h>
  22
  23 #include <event.h>
  24 #include <limits.h>
  25 #include <stdint.h>
  26 #include <stdio.h>
  27 #include <string.h>
  28
  29 #include <imsg.h>
  30
  31 #include "got_error.h"
  32 #include "got_object.h"
  33 #include "got_path.h"
  34
  35 #include "got_lib_hash.h"
  36
  37 #include "gotd.h"
  38 #include "log.h"
  39
  40 void
  41 gotd_imsg_send_ack(struct got_object_id *id, struct imsgbuf *ibuf,
  42     uint32_t peerid, pid_t pid)
  43 {
  44         const struct got_error *err = NULL;
  45         struct gotd_imsg_ack iack;
  46         char hex[SHA1_DIGEST_STRING_LENGTH];
  47
  48         if (log_getverbose() > 0 &&
  49             got_sha1_digest_to_str(id->sha1, hex, sizeof(hex)))
  50                 log_debug("sending ACK for %s", hex);
  51
  52         memset(&iack, 0, sizeof(iack));
  53         memcpy(iack.object_id, id->sha1, SHA1_DIGEST_LENGTH);
  54
  55         if (imsg_compose(ibuf, GOTD_IMSG_ACK, peerid, pid, -1,
  56             &iack, sizeof(iack)) == -1) {
  57                 err = got_error_from_errno("imsg_compose ACK");
  58                 goto done;
  59         }
  60
  61         err = gotd_imsg_flush(ibuf);
  62 done:
  63         if (err)
  64                 log_warnx("sending ACK: %s", err->msg);
  65 }
  66
  67 void
  68 gotd_imsg_send_nak(struct got_object_id *id, struct imsgbuf *ibuf,
  69     uint32_t peerid, pid_t pid)
  70 {
  71         const struct got_error *err = NULL;
  72         struct gotd_imsg_nak inak;
  73         char hex[SHA1_DIGEST_STRING_LENGTH];
  74
  75         if (log_getverbose() > 0 &&
  76             got_sha1_digest_to_str(id->sha1, hex, sizeof(hex)))
  77                 log_debug("sending NAK for %s", hex);
  78
  79         memset(&inak, 0, sizeof(inak));
  80         memcpy(inak.object_id, id->sha1, SHA1_DIGEST_LENGTH);
  81
  82         if (imsg_compose(ibuf, GOTD_IMSG_NAK, peerid, pid, -1,
  83             &inak, sizeof(inak)) == -1) {
  84                 err = got_error_from_errno("imsg_compose NAK");
  85                 goto done;
  86         }
  87
  88         err = gotd_imsg_flush(ibuf);
  89 done:
  90         if (err)
  91                 log_warnx("sending NAK: %s", err->msg);
  92 }