/*- * Copyright (c) 2007 Sean C. Farley <scf@FreeBSD.org> * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer, * without modification, immediately at the beginning of the file. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/cdefs.h> extern char **environ; const char *envName = "FOOBAR"; const char *envValSmall = "Hi"; const char *envValLarge = "Hi, again"; const char *envValAny = "Any value"; int main(int argc, char **argv) { const char *env1 = NULL; const char *env2 = NULL; const char *env3 = NULL; const char *env4 = NULL; const char *env5 = NULL; int testNdx; /* Clean slate. */ environ = NULL; testNdx = 0; /* Initial value of variable. */ if (getenv(envName) != NULL) printf("not "); printf("ok %d - getenv(\"%s\")\n", ++testNdx, envName); /* Set value of variable to smaller value and get value. */ if ((setenv(envName, envValSmall, 1) != 0) || ((env1 = getenv(envName)) == NULL) || (strcmp(env1, envValSmall) != 0)) printf("not "); printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, envValSmall); /* Unset variable. */ if ((unsetenv(envName) == -1) || ((env2 = getenv(envName)) != NULL)) printf("not "); printf("ok %d - unsetenv(\"%s\")\n", ++testNdx, envName); /* Set variable to bigger value and get value. */ if ((setenv(envName, envValLarge, 1) != 0) || ((env3 = getenv(envName)) == NULL) || (strcmp(env3, envValLarge) != 0)) printf("not "); printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, envValLarge); /* Set variable to smaller value and get value. */ if ((setenv(envName, envValSmall, 1) != 0) || ((env4 = getenv(envName)) == NULL) || (strcmp(env4, envValSmall) != 0)) printf("not "); printf("ok %d - setenv(\"%s\", \"%s\", 1)\n", ++testNdx, envName, envValSmall); /* Set variable to any value without overwrite and get value. */ if ((setenv(envName, envValAny, 0) != 0) || ((env5 = getenv(envName)) == NULL) || (strcmp(env5, envValAny) == 0)) printf("not "); printf("ok %d - setenv(\"%s\", \"%s\", 0)\n", ++testNdx, envName, envValAny); /* * Verify FreeBSD-ism about allowing a program to keep old pointers without * risk of segfaulting. */ if ((strcmp(env1, envValSmall) != 0) || (strcmp(env3, envValSmall) != 0) || (strcmp(env4, envValSmall) != 0)) printf("not "); printf("ok %d - old variables point to valid memory\n", ++testNdx); exit(EXIT_SUCCESS); }