environment.c revision 1.1
1/* $NetBSD: environment.c,v 1.1 2011/04/13 18:15:40 elric Exp $ */ 2 3/* 4 * Copyright (c) 2000, 2005 Kungliga Tekniska H��gskolan 5 * (Royal Institute of Technology, Stockholm, Sweden). 6 * All rights reserved. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * 3. Neither the name of the Institute nor the names of its contributors 20 * may be used to endorse or promote products derived from this software 21 * without specific prior written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 26 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 */ 35 36 37#include <config.h> 38 39#include <stdio.h> 40#include <string.h> 41#include <ctype.h> 42#include <krb5/roken.h> 43 44/* find assignment in env list; len is length of variable including 45 * equal 46 */ 47 48static int 49find_var(char **env, char *assignment, size_t len) 50{ 51 int i; 52 for(i = 0; env != NULL && env[i] != NULL; i++) 53 if(strncmp(env[i], assignment, len) == 0) 54 return i; 55 return -1; 56} 57 58/* 59 * return count of environment assignments from open file F in 60 * assigned and list of malloced strings in env, return 0 or errno 61 * number 62 */ 63 64static int 65read_env_file(FILE *F, char ***env, int *assigned) 66{ 67 int idx = 0; 68 int i; 69 char **l; 70 char buf[BUFSIZ], *p, *r; 71 char **tmp; 72 int ret = 0; 73 74 *assigned = 0; 75 76 for(idx = 0; *env != NULL && (*env)[idx] != NULL; idx++); 77 l = *env; 78 79 /* This is somewhat more relaxed on what it accepts then 80 * Wietses sysv_environ from K4 was... 81 */ 82 while (fgets(buf, BUFSIZ, F) != NULL) { 83 buf[strcspn(buf, "#\n")] = '\0'; 84 85 for(p = buf; isspace((unsigned char)*p); p++); 86 if (*p == '\0') 87 continue; 88 89 /* Here one should check that it's a 'valid' env string... */ 90 r = strchr(p, '='); 91 if (r == NULL) 92 continue; 93 94 if((i = find_var(l, p, r - p + 1)) >= 0) { 95 char *val = strdup(p); 96 if(val == NULL) { 97 ret = ENOMEM; 98 break; 99 } 100 free(l[i]); 101 l[i] = val; 102 (*assigned)++; 103 continue; 104 } 105 106 tmp = realloc(l, (idx+2) * sizeof (char *)); 107 if(tmp == NULL) { 108 ret = ENOMEM; 109 break; 110 } 111 112 l = tmp; 113 l[idx] = strdup(p); 114 if(l[idx] == NULL) { 115 ret = ENOMEM; 116 break; 117 } 118 l[++idx] = NULL; 119 (*assigned)++; 120 } 121 if(ferror(F)) 122 ret = errno; 123 *env = l; 124 return ret; 125} 126 127/* 128 * return count of environment assignments from file and 129 * list of malloced strings in `env' 130 */ 131 132ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL 133read_environment(const char *file, char ***env) 134{ 135 int assigned; 136 FILE *F; 137 138 if ((F = fopen(file, "r")) == NULL) 139 return 0; 140 141 read_env_file(F, env, &assigned); 142 fclose(F); 143 return assigned; 144} 145 146ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL 147free_environment(char **env) 148{ 149 int i; 150 if (env == NULL) 151 return; 152 for (i = 0; env[i]; i++) 153 free(env[i]); 154 free(env); 155} 156