1178825Sdfr/* 2233294Sstas * Copyright (c) 2005 Kungliga Tekniska H��gskolan 3233294Sstas * (Royal Institute of Technology, Stockholm, Sweden). 4233294Sstas * All rights reserved. 5178825Sdfr * 6233294Sstas * Redistribution and use in source and binary forms, with or without 7233294Sstas * modification, are permitted provided that the following conditions 8233294Sstas * are met: 9178825Sdfr * 10233294Sstas * 1. Redistributions of source code must retain the above copyright 11233294Sstas * notice, this list of conditions and the following disclaimer. 12178825Sdfr * 13233294Sstas * 2. Redistributions in binary form must reproduce the above copyright 14233294Sstas * notice, this list of conditions and the following disclaimer in the 15233294Sstas * documentation and/or other materials provided with the distribution. 16178825Sdfr * 17233294Sstas * 3. Neither the name of the Institute nor the names of its contributors 18233294Sstas * may be used to endorse or promote products derived from this software 19233294Sstas * without specific prior written permission. 20178825Sdfr * 21233294Sstas * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 22233294Sstas * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23233294Sstas * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24233294Sstas * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 25233294Sstas * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26233294Sstas * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27233294Sstas * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28233294Sstas * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29233294Sstas * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30233294Sstas * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31233294Sstas * SUCH DAMAGE. 32178825Sdfr */ 33178825Sdfr 34178825Sdfr#include <config.h> 35178825Sdfr 36178825Sdfr#include "roken.h" 37178825Sdfr#include "test-mem.h" 38178825Sdfr 39178825Sdfrchar *s1 = "VAR1=VAL1#comment\n\ 40178825SdfrVAR2=VAL2 VAL2 #comment\n\ 41178825Sdfr#this another comment\n\ 42178825Sdfr\n\ 43178825SdfrVAR3=FOO"; 44178825Sdfr 45178825Sdfrchar *s2 = "VAR1=ENV2\n\ 46178825Sdfr"; 47178825Sdfr 48178825Sdfrstatic void 49178825Sdfrmake_file(char *tmpl, size_t l) 50178825Sdfr{ 51178825Sdfr int fd; 52178825Sdfr strlcpy(tmpl, "env.XXXXXX", l); 53178825Sdfr fd = mkstemp(tmpl); 54178825Sdfr if(fd < 0) 55178825Sdfr err(1, "mkstemp"); 56178825Sdfr close(fd); 57178825Sdfr} 58178825Sdfr 59178825Sdfrstatic void 60178825Sdfrwrite_file(const char *fn, const char *s) 61178825Sdfr{ 62178825Sdfr FILE *f; 63178825Sdfr f = fopen(fn, "w"); 64178825Sdfr if(f == NULL) { 65178825Sdfr unlink(fn); 66178825Sdfr err(1, "fopen"); 67178825Sdfr } 68178825Sdfr if(fwrite(s, 1, strlen(s), f) != strlen(s)) 69178825Sdfr err(1, "short write"); 70178825Sdfr if(fclose(f) != 0) { 71178825Sdfr unlink(fn); 72178825Sdfr err(1, "fclose"); 73178825Sdfr } 74178825Sdfr} 75178825Sdfr 76178825Sdfrint 77178825Sdfrmain(int argc, char **argv) 78178825Sdfr{ 79178825Sdfr char **env = NULL; 80178825Sdfr int count = 0; 81178825Sdfr char fn[MAXPATHLEN]; 82178825Sdfr int error = 0; 83178825Sdfr 84178825Sdfr make_file(fn, sizeof(fn)); 85178825Sdfr 86178825Sdfr write_file(fn, s1); 87178825Sdfr count = read_environment(fn, &env); 88178825Sdfr if(count != 3) { 89178825Sdfr warnx("test 1: variable count %d != 3", count); 90178825Sdfr error++; 91178825Sdfr } 92178825Sdfr 93178825Sdfr write_file(fn, s2); 94178825Sdfr count = read_environment(fn, &env); 95178825Sdfr if(count != 1) { 96178825Sdfr warnx("test 2: variable count %d != 1", count); 97178825Sdfr error++; 98178825Sdfr } 99178825Sdfr 100178825Sdfr unlink(fn); 101178825Sdfr count = read_environment(fn, &env); 102178825Sdfr if(count != 0) { 103178825Sdfr warnx("test 3: variable count %d != 0", count); 104178825Sdfr error++; 105178825Sdfr } 106178825Sdfr for(count = 0; env && env[count]; count++); 107178825Sdfr if(count != 3) { 108178825Sdfr warnx("total variable count %d != 3", count); 109178825Sdfr error++; 110178825Sdfr } 111178825Sdfr free_environment(env); 112233294Sstas 113233294Sstas 114178825Sdfr return error; 115178825Sdfr} 116