0ca483d5d0
Add a convenience function for writing a single buffer to a file. Signed-off-by: Jeremy Kerr <jeremy.kerr@canonical.com>
157 lines
3.4 KiB
C
157 lines
3.4 KiB
C
/*
|
|
* Copyright (C) 2012 Jeremy Kerr <jeremy.kerr@canonical.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 3
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
|
|
* USA.
|
|
*
|
|
* In addition, as a special exception, the copyright holders give
|
|
* permission to link the code of portions of this program with the OpenSSL
|
|
* library under certain conditions as described in each individual source file,
|
|
* and distribute linked combinations including the two.
|
|
*
|
|
* You must obey the GNU General Public License in all respects for all
|
|
* of the code used other than OpenSSL. If you modify file(s) with this
|
|
* exception, you may extend this exception to your version of the
|
|
* file(s), but you are not obligated to do so. If you do not wish to do
|
|
* so, delete this exception statement from your version. If you delete
|
|
* this exception statement from all source files in the program, then
|
|
* also delete it here.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include <openssl/bio.h>
|
|
#include <openssl/pem.h>
|
|
#include <openssl/err.h>
|
|
|
|
#include <ccan/talloc/talloc.h>
|
|
#include <ccan/read_write_all/read_write_all.h>
|
|
|
|
#include "fileio.h"
|
|
|
|
EVP_PKEY *fileio_read_pkey(const char *filename)
|
|
{
|
|
EVP_PKEY *key = NULL;
|
|
BIO *bio;
|
|
|
|
bio = BIO_new_file(filename, "r");
|
|
if (!bio)
|
|
goto out;
|
|
|
|
key = PEM_read_bio_PrivateKey(bio, NULL, NULL, NULL);
|
|
|
|
out:
|
|
BIO_free_all(bio);
|
|
if (!key) {
|
|
fprintf(stderr, "Can't load key from file '%s'\n", filename);
|
|
ERR_print_errors_fp(stderr);
|
|
}
|
|
return key;
|
|
}
|
|
|
|
X509 *fileio_read_cert(const char *filename)
|
|
{
|
|
X509 *cert = NULL;
|
|
BIO *bio;
|
|
|
|
bio = BIO_new_file(filename, "r");
|
|
if (!bio)
|
|
goto out;
|
|
|
|
cert = PEM_read_bio_X509(bio, NULL, NULL, NULL);
|
|
|
|
out:
|
|
BIO_free_all(bio);
|
|
if (!cert) {
|
|
fprintf(stderr, "Can't load certificate from file '%s'\n",
|
|
filename);
|
|
ERR_print_errors_fp(stderr);
|
|
}
|
|
return cert;
|
|
}
|
|
|
|
int fileio_read_file(void *ctx, const char *filename,
|
|
uint8_t **out_buf, size_t *out_len)
|
|
{
|
|
struct stat statbuf;
|
|
uint8_t *buf;
|
|
size_t len;
|
|
int fd, rc;
|
|
|
|
rc = -1;
|
|
|
|
fd = open(filename, O_RDONLY);
|
|
if (fd < 0) {
|
|
perror("open");
|
|
goto out;
|
|
}
|
|
|
|
rc = fstat(fd, &statbuf);
|
|
if (rc) {
|
|
perror("fstat");
|
|
goto out;
|
|
}
|
|
|
|
len = statbuf.st_size;
|
|
|
|
buf = talloc_array(ctx, uint8_t, len);
|
|
if (!buf) {
|
|
perror("talloc");
|
|
goto out;
|
|
}
|
|
|
|
if (!read_all(fd, buf, len)) {
|
|
perror("read_all");
|
|
goto out;
|
|
}
|
|
|
|
rc = 0;
|
|
|
|
out:
|
|
if (fd >= 0)
|
|
close(fd);
|
|
if (rc) {
|
|
fprintf(stderr, "Error reading file %s\n", filename);
|
|
} else {
|
|
*out_buf = buf;
|
|
*out_len = len;
|
|
}
|
|
return rc;
|
|
|
|
}
|
|
|
|
int fileio_write_file(const char *filename, uint8_t *buf, size_t len)
|
|
{
|
|
int fd;
|
|
|
|
fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
|
|
if (fd < 0) {
|
|
perror("open");
|
|
return -1;
|
|
}
|
|
|
|
if (!write_all(fd, buf, len)) {
|
|
perror("write_all");
|
|
close(fd);
|
|
return -1;
|
|
}
|
|
|
|
close(fd);
|
|
return 0;
|
|
}
|