src/test/test-hash.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2016 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <stdio.h>
  21
  22 #include "alloc-util.h"
  23 #include "log.h"
  24 #include "string-util.h"
  25 #include "khash.h"
  26
  27 int main(int argc, char *argv[]) {
  28         _cleanup_(khash_unrefp) khash *h = NULL, *copy = NULL;
  29         _cleanup_free_ char *s = NULL;
  30         int r;
  31
  32         log_set_max_level(LOG_DEBUG);
  33
  34         assert_se(khash_new(&h, NULL) == -EINVAL);
  35         assert_se(khash_new(&h, "") == -EINVAL);
  36         r = khash_new(&h, "foobar");
  37         if (r == -EAFNOSUPPORT) {
  38                 puts("khash not supported on this kernel, skipping");
  39                 return EXIT_TEST_SKIP;
  40         }
  41         assert_se(r == -EOPNOTSUPP);
  42
  43         assert_se(khash_new(&h, "sha256") >= 0);
  44         assert_se(khash_get_size(h) == 32);
  45         assert_se(streq(khash_get_algorithm(h), "sha256"));
  46
  47         assert_se(khash_digest_string(h, &s) >= 0);
  48         assert_se(streq(s, "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"));
  49         s = mfree(s);
  50
  51         assert_se(khash_put(h, "foobar", 6) >= 0);
  52         assert_se(khash_digest_string(h, &s) >= 0);
  53         assert_se(streq(s, "c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2"));
  54         s = mfree(s);
  55
  56         assert_se(khash_put(h, "piep", 4) >= 0);
  57         assert_se(khash_digest_string(h, &s) >= 0);
  58         assert_se(streq(s, "f114d872b5ea075d3be9040d0b7a429514b3f9324a8e8e3dc3fb24c34ee56bea"));
  59         s = mfree(s);
  60
  61         assert_se(khash_put(h, "foo", 3) >= 0);
  62         assert_se(khash_dup(h, &copy) >= 0);
  63
  64         assert_se(khash_put(h, "bar", 3) >= 0);
  65         assert_se(khash_put(copy, "bar", 3) >= 0);
  66
  67         assert_se(khash_digest_string(h, &s) >= 0);
  68         assert_se(streq(s, "c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2"));
  69         s = mfree(s);
  70
  71         assert_se(khash_digest_string(copy, &s) >= 0);
  72         assert_se(streq(s, "c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2"));
  73         s = mfree(s);
  74
  75         h = khash_unref(h);
  76
  77         assert_se(khash_new_with_key(&h, "hmac(sha256)", "quux", 4) >= 0);
  78         assert_se(khash_get_size(h) == 32);
  79         assert_se(streq(khash_get_algorithm(h), "hmac(sha256)"));
  80
  81         assert_se(khash_digest_string(h, &s) >= 0);
  82         assert_se(streq(s, "abed9f8218ab473f77218a6a7d39abf1d21fa46d0700c4898e330ba88309d5ae"));
  83         s = mfree(s);
  84
  85         assert_se(khash_put(h, "foobar", 6) >= 0);
  86         assert_se(khash_digest_string(h, &s) >= 0);
  87         assert_se(streq(s, "33f6c70a60db66007d5325d5d1dea37c371354e5b83347a59ad339ce9f4ba3dc"));
  88
  89         return 0;
  90 }