shithub: riscv

ref: 717237303d368378283ad719ee4f2c580b99fd2b
dir: /sys/src/ape/lib/ap/plan9/_envsetup.c/

#include "lib.h"
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <dirent.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include "sys9.h"
#include "dir.h"

/*
 * Called before main to initialize environ.
 * Some plan9 environment variables
 * have 0 bytes in them (notably $path);
 * we change them to 1's (and execve changes back)
 *
 * Also, register the note handler.
 */

char **environ;
int *_errnoloc;
unsigned long _clock;
static void fdsetup(char *, char *);
static void sigsetup(char *, char *);

enum {
	Envhunk=7000,
};

void
_envsetup(void)
{
	int dfd;
	int n, nd, m, i, j, f;
	int psize, cnt;
	int nohandle;
	int fdinited;
	char *ps, *p;
	char **pp;
	Dir *d9, *d9a;

	ps = 0;
	psize = 0;
	nohandle = 0;
	fdinited = 0;
	cnt = 0;
	dfd = _OPEN("/env", OREAD|OCEXEC);
	if(dfd < 0)
		goto done;
	psize = Envhunk;
	ps = p = malloc(psize);
	nd = _dirreadall(dfd, &d9a);
	_CLOSE(dfd);
	for(j=0; j<nd; j++){
		d9 = &d9a[j];
		n = strlen(d9->name);
		m = d9->length;
		i = p - ps;
		if(i+n+5+m+1 > psize) {
			psize += (n+m+6 < Envhunk)? Envhunk : n+m+6;
			ps = realloc(ps, psize);
			p = ps + i;
		}
		memcpy(p, "/env/", 5);
		memcpy(p+5, d9->name, n+1);
		f = _OPEN(p, OREAD|OCEXEC);
		if(f < 0)
			continue;
		if(n > 3 && memcmp(d9->name, "fn#", 3)==0){
			if(m > 3
			&& _READ(f, p+3, m) == m
			&& memcmp(p+3, "fn ", 3) == 0
			&& memcmp(p+3+3, d9->name+3, n-3) == 0
			&& memchr(p+3+n, '{', m-n) != 0){
				memcpy(p, "#()", 3);
				p[3+m] = '\0';
				p += m+4;
				cnt++;
			}
			_CLOSE(f);
			continue;
		}
		memcpy(p, d9->name, n);
		p[n] = '=';
		if(_READ(f, p+n+1, m) != m)
			m = 0;
		_CLOSE(f);
		if(p[n+m]=='\0')
			m--;
		for(i=0; i<m; i++)
			if(p[n+1+i]=='\0')
				p[n+1+i] = '\1';
		p[n+1+m] = '\0';
		if(strcmp(d9->name, "_fdinfo") == 0) {
			_fdinit(p+n+1, p+n+1+m);
			fdinited = 1;
		} else if(strcmp(d9->name, "_sighdlr") == 0)
			sigsetup(p+n+1, p+n+1+m);
		else if(strcmp(d9->name, "nohandle") == 0)
			nohandle = 1;
		p += n+m+2;
		cnt++;
	}
	free(d9a);
	if(!fdinited)
		_fdinit(0, 0);
done:
	environ = pp = malloc((1+cnt)*sizeof(char *));
	p = ps;
	for(i = 0; i < cnt; i++) {
		*pp++ = p;
		p = memchr(p, 0, ps+psize-p);
		if (!p)
			break;
		p++;
	}
	*pp = 0;
	if(!nohandle)
		_NOTIFY(_notehandler);
}

static void
sigsetup(char *s, char *se)
{
	int sig;
	char *e;

	while(s < se){
		sig = strtoul(s, &e, 10);
		if(s == e)
			break;
		s = e;
		if(sig <= MAXSIG)
			_sighdlr[sig] = SIG_IGN;
	}
}