summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDan McGee <dan@archlinux.org>2011-08-11 15:46:18 -0500
committerDan McGee <dan@archlinux.org>2011-08-15 07:07:13 -0500
commit31f2e0cba3281660a2a3ffc6f902a7019cb4699b (patch)
tree285ace09e573a6b214045fdccf4c4bb91ac0a606
parentff88228abde9c288b3e08f5090a1ff3941fdc018 (diff)
Add ALPM sha256sum routines
These mirror ones we already have for md5sums. Signed-off-by: Dan McGee <dan@archlinux.org>
-rw-r--r--lib/libalpm/alpm.h1
-rw-r--r--lib/libalpm/util.c80
2 files changed, 81 insertions, 0 deletions
diff --git a/lib/libalpm/alpm.h b/lib/libalpm/alpm.h
index 3bbef0ab..6d696ec8 100644
--- a/lib/libalpm/alpm.h
+++ b/lib/libalpm/alpm.h
@@ -1058,6 +1058,7 @@ char *alpm_dep_compute_string(const alpm_depend_t *dep);
/* checksums */
char *alpm_compute_md5sum(const char *name);
+char *alpm_compute_sha256sum(const char *filename);
/** @addtogroup alpm_api_errors Error Codes
* @{
diff --git a/lib/libalpm/util.c b/lib/libalpm/util.c
index adb1d192..d0661289 100644
--- a/lib/libalpm/util.c
+++ b/lib/libalpm/util.c
@@ -46,8 +46,10 @@
#ifdef HAVE_LIBSSL
#include <openssl/md5.h>
+#include <openssl/sha.h>
#else
#include "md5.h"
+#include "sha2.h"
#endif
/* libalpm */
@@ -747,6 +749,53 @@ static int md5_file(const char *path, unsigned char output[16])
fclose(f);
return 0;
}
+
+/* third param is so we match the PolarSSL definition */
+static int sha2_file(const char *path, unsigned char output[32], int is224)
+{
+ FILE *f;
+ size_t n;
+ SHA256_CTX ctx;
+ unsigned char *buf;
+
+ CALLOC(buf, 8192, sizeof(unsigned char), return 1);
+
+ if((f = fopen(path, "rb")) == NULL) {
+ free(buf);
+ return 1;
+ }
+
+ if(is224) {
+ SHA224_Init(&ctx);
+ } else {
+ SHA256_Init(&ctx);
+ }
+
+ while((n = fread(buf, 1, sizeof(buf), f)) > 0) {
+ if(is224) {
+ SHA224_Update(&ctx, buf, n);
+ } else {
+ SHA256_Update(&ctx, buf, n);
+ }
+ }
+
+ if(is224) {
+ SHA224_Final(output, &ctx);
+ } else {
+ SHA256_Final(output, &ctx);
+ }
+
+ memset(&ctx, 0, sizeof(SHA256_CTX));
+ free(buf);
+
+ if(ferror(f) != 0) {
+ fclose(f);
+ return 2;
+ }
+
+ fclose(f);
+ return 0;
+}
#endif
/** Get the md5 sum of file.
@@ -799,6 +848,37 @@ int _alpm_test_md5sum(const char *filepath, const char *md5sum)
return ret;
}
+/** Get the sha256 sum of file.
+ * @param filename name of the file
+ * @return the checksum on success, NULL on error
+ * @addtogroup alpm_misc
+ */
+char SYMEXPORT *alpm_compute_sha256sum(const char *filename)
+{
+ unsigned char output[32];
+ char *sha256sum;
+ int ret, i;
+
+ ASSERT(filename != NULL, return NULL);
+
+ /* allocate 64 chars plus 1 for null */
+ CALLOC(sha256sum, 65, sizeof(char), return NULL);
+ /* defined above for OpenSSL, otherwise defined in sha2.h */
+ ret = sha2_file(filename, output, 0);
+
+ if(ret > 0) {
+ return NULL;
+ }
+
+ /* Convert the result to something readable */
+ for (i = 0; i < 32; i++) {
+ /* sprintf is acceptable here because we know our output */
+ sprintf(sha256sum +(i * 2), "%02x", output[i]);
+ }
+
+ return sha256sum;
+}
+
/* Note: does NOT handle sparse files on purpose for speed. */
int _alpm_archive_fgets(struct archive *a, struct archive_read_buffer *b)
{