contrib/libarchive/libarchive/test/test_archive_write_set_passphrase.c

   1 /*-
   2  * Copyright (c) 2011 Tim Kientzle
   3  * Copyright (c) 2014 Michihiro NAKAJIMA
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #include "test.h"
  28 __FBSDID("$FreeBSD$");
  29
  30 struct archive_write;
  31 extern const char * __archive_write_get_passphrase(struct archive_write *);
  32
  33 static void
  34 test(int pristine)
  35 {
  36         struct archive* a = archive_write_new();
  37         struct archive_write* aw = (struct archive_write *)a;
  38
  39         if (!pristine) {
  40                 archive_write_add_filter_gzip(a);
  41                 archive_write_set_format_iso9660(a);
  42         }
  43
  44         assertEqualInt(ARCHIVE_OK, archive_write_set_passphrase(a, "pass1"));
  45         /* An empty passphrase cannot be accepted. */
  46         assertEqualInt(ARCHIVE_FAILED, archive_write_set_passphrase(a, ""));
  47         /* NULL passphrases cannot be accepted. */
  48         assertEqualInt(ARCHIVE_FAILED, archive_write_set_passphrase(a, NULL));
  49         /* Check a passphrase. */
  50         assertEqualString("pass1", __archive_write_get_passphrase(aw));
  51         /* Change the passphrase. */
  52         assertEqualInt(ARCHIVE_OK, archive_write_set_passphrase(a, "pass2"));
  53         assertEqualString("pass2", __archive_write_get_passphrase(aw));
  54
  55         archive_write_free(a);
  56 }
  57
  58 DEFINE_TEST(test_archive_write_set_passphrase)
  59 {
  60         test(1);
  61         test(0);
  62 }
  63
  64
  65 static const char *
  66 callback1(struct archive *a, void *_client_data)
  67 {
  68         int *cnt;
  69
  70         (void)a; /* UNUSED */
  71
  72         cnt = (int *)_client_data;
  73         *cnt += 1;
  74         return ("passCallBack");
  75 }
  76
  77 DEFINE_TEST(test_archive_write_set_passphrase_callback)
  78 {
  79         struct archive* a = archive_write_new();
  80         struct archive_write* aw = (struct archive_write *)a;
  81         int cnt = 0;
  82
  83         archive_write_set_format_zip(a);
  84
  85         assertEqualInt(ARCHIVE_OK,
  86             archive_write_set_passphrase_callback(a, &cnt, callback1));
  87         /* Check a passphrase. */
  88         assertEqualString("passCallBack", __archive_write_get_passphrase(aw));
  89         assertEqualInt(1, cnt);
  90         /* Callback function should be called just once. */
  91         assertEqualString("passCallBack", __archive_write_get_passphrase(aw));
  92         assertEqualInt(1, cnt);
  93
  94         archive_write_free(a);
  95 }