1 /*- 2 * Copyright (c) 2007 Sean C. Farley <scf@FreeBSD.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer, 10 * without modification, immediately at the beginning of the file. 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 ``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 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 #include <stdio.h> 27 #include <stdlib.h> 28 #include <string.h> 29 30 31 #include <sys/cdefs.h> 32 __FBSDID("$FreeBSD$"); 33 34 35 extern char **environ; 36 const char *envName = "FOOBAR"; 37 const char *envValSmall = "Hi"; 38 const char *envValLarge = "Hi, again"; 39 const char *envValAny = "Any value"; 40 41 42 int 43 main(int argc, char **argv) 44 { 45 const char *env1 = NULL; 46 const char *env2 = NULL; 47 const char *env3 = NULL; 48 const char *env4 = NULL; 49 const char *env5 = NULL; 50 int testNdx; 51 52 /* Clean slate. */ 53 environ = NULL; 54 testNdx = 0; 55 56 /* Initial value of variable. */ 57 if (getenv(envName) != NULL) 58 printf("not "); 59 printf("ok %d - getenv(\"%s\")\n", ++testNdx, envName); 60 61 /* Set value of variable to smaller value and get value. */ 62 if ((setenv(envName, envValSmall, 1) != 0) || 63 ((env1 = getenv(envName)) == NULL) || 64 (strcmp(env1, envValSmall) != 0)) 65 printf("not "); 66 printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, 67 envValSmall); 68 69 /* Unset variable. */ 70 if ((unsetenv(envName) == -1) || ((env2 = getenv(envName)) != NULL)) 71 printf("not "); 72 printf("ok %d - unsetenv(\"%s\")\n", ++testNdx, envName); 73 74 /* Set variable to bigger value and get value. */ 75 if ((setenv(envName, envValLarge, 1) != 0) || 76 ((env3 = getenv(envName)) == NULL) || 77 (strcmp(env3, envValLarge) != 0)) 78 printf("not "); 79 printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, 80 envValLarge); 81 82 /* Set variable to smaller value and get value. */ 83 if ((setenv(envName, envValSmall, 1) != 0) || 84 ((env4 = getenv(envName)) == NULL) || 85 (strcmp(env4, envValSmall) != 0)) 86 printf("not "); 87 printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, 88 envValSmall); 89 90 /* Set variable to any value without overwrite and get value. */ 91 if ((setenv(envName, envValAny, 0) != 0) || 92 ((env5 = getenv(envName)) == NULL) || 93 (strcmp(env5, envValAny) == 0)) 94 printf("not "); 95 printf("ok %d - setenv(\"%s\", \"%s\", 0)\n", ++testNdx, envName, 96 envValAny); 97 98 /* 99 * Verify FreeBSD-ism about allowing a program to keep old pointers without 100 * risk of segfaulting. 101 */ 102 if ((strcmp(env1, envValSmall) != 0) || 103 (strcmp(env3, envValSmall) != 0) || 104 (strcmp(env4, envValSmall) != 0)) 105 printf("not "); 106 printf("ok %d - old variables point to valid memory\n", ++testNdx); 107 108 exit(EXIT_SUCCESS); 109 } 110