summaryrefslogtreecommitdiffstats
path: root/ffi/examples/encrypt-for.c
blob: f0c3444908232e0578c1d0cc1901b329e8f46b5d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/* This example demonstrates how to use the low-level interface to
   encrypt a file.  */

#define _GNU_SOURCE
#include <error.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <time.h>

#include <sequoia.h>

int
main (int argc, char **argv)
{
  struct stat st;
  int fd;
  uint8_t *b;
  sq_status_t rc;
  sq_error_t err;
  int use_armor = 1;
  sq_tpk_t tpk;
  sq_writer_t sink;
  sq_writer_stack_t writer = NULL;
  void *cipher = NULL;
  size_t cipher_bytes = 0;

  if (argc != 2)
    error (1, 0, "Usage: %s <keyfile> <plain >cipher", argv[0]);

  if (stat (argv[1], &st))
    error (1, errno, "%s", argv[1]);

  fd = open (argv[1], O_RDONLY);
  if (fd == -1)
    error (1, errno, "%s", argv[1]);

  b = mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
  close (fd);
  if (b == MAP_FAILED)
    error (1, errno, "mmap");

  tpk = sq_tpk_from_bytes (&err, b, st.st_size);
  if (tpk == NULL)
    error (1, 0, "sq_packet_parser_from_bytes: %s", sq_error_string (err));

  sink = sq_writer_alloc (&cipher, &cipher_bytes);

  if (use_armor)
    sink = sq_armor_writer_new (&err, sink, SQ_ARMOR_KIND_MESSAGE,
                                NULL, 0);

  writer = sq_writer_stack_message (sink);
  writer = sq_encryptor_new (&err,
			     writer,
			     NULL, 0, /* no passwords */
			     &tpk, 1,
			     SQ_ENCRYPTION_MODE_FOR_TRANSPORT);
  if (writer == NULL)
    error (1, 0, "sq_encryptor_new: %s", sq_error_string (err));

  writer = sq_literal_writer_new (&err, writer);
  if (writer == NULL)
    error (1, 0, "sq_literal_writer_new: %s", sq_error_string (err));

  size_t nread;
  uint8_t buf[4096];
  while ((nread = fread (buf, 1, sizeof buf, stdin)))
    {
      uint8_t *b = buf;
      while (nread)
	{
	  ssize_t written;
	  written = sq_writer_stack_write (&err, writer, b, nread);
	  if (written < 0)
            error (1, 0, "sq_writer_stack_write: %s", sq_error_string (err));

	  b += written;
	  nread -= written;
	}
    }

  rc = sq_writer_stack_finalize (&err, writer);
  writer = NULL;
  if (rc)
    error (1, 0, "sq_writer_stack_write: %s", sq_error_string (err));

  fwrite (cipher, 1, cipher_bytes, stdout);

  munmap (b, st.st_size);
  return 0;
}