1 /* 	$OpenBSD: common.c,v 1.1 2014/06/24 01:14:18 djm Exp $ */
2 /*
3  * Helpers for key API tests
4  *
5  * Placed in the public domain
6  */
7 
8 #include "includes.h"
9 
10 #include <sys/types.h>
11 #include <sys/param.h>
12 #include <sys/stat.h>
13 #include <fcntl.h>
14 #include <stdio.h>
15 #ifdef HAVE_STDINT_H
16 #include <stdint.h>
17 #endif
18 #include <stdlib.h>
19 #include <string.h>
20 #include <unistd.h>
21 
22 #include <openssl/bn.h>
23 #include <openssl/rsa.h>
24 #include <openssl/dsa.h>
25 #include <openssl/objects.h>
26 #ifdef OPENSSL_HAS_NISTP256
27 # include <openssl/ec.h>
28 #endif
29 
30 #include "../test_helper/test_helper.h"
31 
32 #include "ssherr.h"
33 #include "authfile.h"
34 #include "sshkey.h"
35 #include "sshbuf.h"
36 
37 #include "common.h"
38 
39 struct sshbuf *
40 load_file(const char *name)
41 {
42 	int fd;
43 	struct sshbuf *ret;
44 
45 	ASSERT_PTR_NE(ret = sshbuf_new(), NULL);
46 	ASSERT_INT_NE(fd = open(test_data_file(name), O_RDONLY), -1);
47 	ASSERT_INT_EQ(sshkey_load_file(fd, name, ret), 0);
48 	close(fd);
49 	return ret;
50 }
51 
52 struct sshbuf *
53 load_text_file(const char *name)
54 {
55 	struct sshbuf *ret = load_file(name);
56 	const u_char *p;
57 
58 	/* Trim whitespace at EOL */
59 	for (p = sshbuf_ptr(ret); sshbuf_len(ret) > 0;) {
60 		if (p[sshbuf_len(ret) - 1] == '\r' ||
61 		    p[sshbuf_len(ret) - 1] == '\t' ||
62 		    p[sshbuf_len(ret) - 1] == ' ' ||
63 		    p[sshbuf_len(ret) - 1] == '\n')
64 			ASSERT_INT_EQ(sshbuf_consume_end(ret, 1), 0);
65 		else
66 			break;
67 	}
68 	/* \0 terminate */
69 	ASSERT_INT_EQ(sshbuf_put_u8(ret, 0), 0);
70 	return ret;
71 }
72 
73 BIGNUM *
74 load_bignum(const char *name)
75 {
76 	BIGNUM *ret = NULL;
77 	struct sshbuf *buf;
78 
79 	buf = load_text_file(name);
80 	ASSERT_INT_NE(BN_hex2bn(&ret, (const char *)sshbuf_ptr(buf)), 0);
81 	sshbuf_free(buf);
82 	return ret;
83 }
84 
85