5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews#ifndef lint
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsstatic char *rcsid = "$Id: setenv.c,v 1.1 2003/06/04 00:27:01 marka Exp $";
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews#endif
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews/*
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * Copyright (c) 2002 Japan Network Information Center.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * All rights reserved.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * By using this file, you agree to the terms and conditions set forth bellow.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * LICENSE TERMS AND CONDITIONS
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * The following License Terms and Conditions apply, unless a different
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * license is obtained from Japan Network Information Center ("JPNIC"),
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * a Japanese association, Kokusai-Kougyou-Kanda Bldg 6F, 2-3-4 Uchi-Kanda,
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * Chiyoda-ku, Tokyo 101-0047, Japan.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * 1. Use, Modification and Redistribution (including distribution of any
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * modified or derived work) in source and/or binary forms is permitted
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * under this License Terms and Conditions.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * 2. Redistribution of source code must retain the copyright notices as they
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * appear in each source code file, this License Terms and Conditions.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * 3. Redistribution in binary form must reproduce the Copyright Notice,
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * this License Terms and Conditions, in the documentation and/or other
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * materials provided with the distribution. For the purposes of binary
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * distribution the "Copyright Notice" refers to the following language:
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * "Copyright (c) 2000-2002 Japan Network Information Center. All rights reserved."
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * 4. The name of JPNIC may not be used to endorse or promote products
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * derived from this Software without specific prior written approval of
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * JPNIC.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * 5. Disclaimer/Limitation of Liability: THIS SOFTWARE IS PROVIDED BY JPNIC
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL JPNIC BE LIABLE
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews */
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews#include <stddef.h>
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews#include <string.h>
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews/*
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * We don't include <stdlib.h> here.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * Also <stdlib.h> may declare `environ' and its type might be different
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews * from ours.
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews */
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsextern char **environ;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewstypedef struct myenv myenv_t;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsstruct myenv {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews char *pointer;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenv_t *next;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenv_t *prev;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews};
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsstatic myenv_t *myenvs = NULL;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsvoid
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsmyunsetenv(const char *name) {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews char **e;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenv_t *mye;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews size_t namelen;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews extern void free(void *);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews namelen = strlen(name);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews for (e = environ; *e != NULL; e++) {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (strncmp(*e, name, namelen) == 0 && (*e)[namelen] == '=')
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews break;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews }
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (*e == NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews return;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews for (mye = myenvs; mye != NULL; mye = mye->next) {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (mye->pointer == *e) {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (mye->next != NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye->next->prev = mye->prev;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (mye->prev != NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye->prev->next = mye->next;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (mye->next == NULL && mye->prev == NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenvs = NULL;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews free(mye);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews free(*e);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews break;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews }
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews }
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews for ( ; *e != NULL; e++)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews *e = *(e + 1);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews}
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews#include <stdlib.h>
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsint
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrewsmysetenv(const char *name, const char *value, int overwrite) {
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenv_t *mye;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews char *buffer;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews int result;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (getenv(name) != NULL && !overwrite)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews return 0;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews buffer = (char *) malloc(strlen(name) + strlen(value) + 2);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (buffer == NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews return -1;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews strcpy(buffer, name);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews strcat(buffer, "=");
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews strcat(buffer, value);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myunsetenv(name);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye = (myenv_t *) malloc(sizeof(myenv_t));
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (mye == NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews return -1;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye->pointer = buffer;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye->next = myenvs;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews mye->prev = NULL;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews if (myenvs != NULL)
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenvs->prev = mye;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews myenvs = mye;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews result = putenv(buffer);
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews return result;
5c526acb82c882e41b655c31f5fa4425c87b671cMark Andrews}