tests/common/config.c

   1 /*
   2  * Copyright (c) 2010 Your File System Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR `AS IS'' AND ANY EXPRESS OR
  14  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  15  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  16  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  17  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  18  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  19  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  20  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  22  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  23  */
  24
  25 /*!
  26  * Common functions for building a configuration directory
  27  */
  28
  29 #include <afsconfig.h>
  30 #include <afs/param.h>
  31 #include <roken.h>
  32
  33 #include <afs/cellconfig.h>
  34 #include <afs/afsutil.h>
  35
  36 #include <hcrypto/des.h>
  37
  38 #include <tests/tap/basic.h>
  39 #include "common.h"
  40
  41 static FILE *
  42 openConfigFile(char *dirname, char *filename) {
  43     char *path = NULL;
  44     FILE *file;
  45
  46     if (asprintf(&path, "%s/%s", dirname, filename) == -1)
  47         return NULL;
  48
  49     file = fopen(path, "w");
  50     free(path);
  51     return file;
  52 }
  53
  54 static void
  55 unlinkConfigFile(char *dirname, char *filename) {
  56     char *path;
  57
  58     if (asprintf(&path, "%s/%s", dirname, filename) != -1) {
  59         unlink(path);
  60         free(path);
  61     }
  62 }
  63
  64 /*!
  65  * Build a test configuration directory, containing a CellServDB and ThisCell
  66  * file for the "example.org" cell
  67  *
  68  * @return
  69  * The path to the configuration directory. This should be freed by the caller
  70  * using free()
  71  *
  72  */
  73
  74 char *
  75 afstest_BuildTestConfig(void) {
  76     char *dir = NULL;
  77     FILE *file;
  78     struct hostent *host;
  79     char hostname[255];
  80     struct in_addr iaddr;
  81
  82     if (asprintf(&dir, "%s/afs_XXXXXX", gettmpdir()) == -1)
  83         goto fail;
  84
  85     if (mkdtemp(dir) == NULL)
  86         goto fail;
  87
  88     /* Work out which IP address to use in our CellServDB. We figure this out
  89      * according to the IP address which ubik is most likely to pick for one of
  90      * our db servers */
  91
  92     gethostname(hostname, sizeof(hostname));
  93     host = gethostbyname(hostname);
  94     if (!host)
  95         return NULL;
  96
  97     memcpy(&iaddr, host->h_addr, 4);
  98
  99     file = openConfigFile(dir, "CellServDB");
 100     fprintf(file, ">example.org # An example cell\n");
 101     fprintf(file, "%s #test.example.org\n", inet_ntoa(iaddr));
 102     fclose(file);
 103
 104     /* Create a ThisCell file */
 105     file = openConfigFile(dir, "ThisCell");
 106     fprintf(file, "example.org");
 107     fclose(file);
 108
 109     return dir;
 110
 111 fail:
 112     if (dir)
 113         free(dir);
 114     return NULL;
 115 }
 116
 117 /*!
 118  * Delete at test configuration directory
 119  */
 120
 121 void
 122 afstest_UnlinkTestConfig(char *dir)
 123 {
 124     DIR *dirp;
 125     struct dirent *de;
 126
 127     /* Sanity check, only zap directories that look like ours */
 128     if (!strstr(dir, "afs_"))
 129         return;
 130     if (getenv("MAKECHECK") != NULL) {
 131         dirp = opendir(dir);
 132         if (!dirp)
 133             return;
 134         while ((de = readdir(dirp)))
 135             unlinkConfigFile(dir, de->d_name);
 136         rmdir(dir);
 137     }
 138 }
 139
 140 int
 141 afstest_AddDESKeyFile(struct afsconf_dir *dir)
 142 {
 143     char keymaterial[]="\x19\x17\xff\xe6\xbb\x77\x2e\xfc";
 144
 145     /* Make sure that it is actually a valid key */
 146     DES_set_odd_parity((DES_cblock *)keymaterial);
 147
 148     return afsconf_AddKey(dir, 1, keymaterial, 1);
 149 }