1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
22/*	  All Rights Reserved  	*/
23
24
25/*
26 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
27 * Use is subject to license terms.
28 */
29
30#pragma ident	"%Z%%M%	%I%	%E% SMI"       /* SVr4.0 1.3 */
31
32/*
33 *  setuname [-t] [-s name] [-n node]
34 */
35
36/*
37 *  Header files referenced:
38 *	<stdio.h>	Standard I/O
39 *	<unistd.h>	Standard UNIX definitions
40 *	<string.h>	String handling
41 *	<fmtmsg.h>	Standard message generation
42 *	<ctype.h>	Character types
43 *	<errno.h>	Error handling
44 *	<signal.h>	Signal handling
45 *	<sys/types.h>	Data types
46 *	<sys/fcntl.h>	File control
47 *	<sys/utsname.h>	System Name
48 *	<sys/sys3b.h>	sys3b() definitions
49 *	<nlist.h>	Definitions for Sun symbol table entries
50 */
51
52#include	<stdio.h>
53#include	<unistd.h>
54#include	<string.h>
55#include	<fmtmsg.h>
56#include	<ctype.h>
57#include	<errno.h>
58#include	<signal.h>
59#include	<sys/types.h>
60#include	<sys/uio.h>
61#include	<sys/fcntl.h>
62#include	<sys/psw.h>
63#include	<sys/utsname.h>
64
65#if u3b || u3b15 || u3b2
66#include	<sys/sys3b.h>
67#endif
68
69#if sun
70#include	<nlist.h>
71#include	<kvm.h>
72#endif
73
74/*
75 * Externals referenced (and not defined in a header)
76 *	optind		index to the next arg for getopt()
77 *	opterr		FLAG, TRUE tells getopt() to write messages
78 *	optarg		Ptr to an option's argument
79 *	getopt()	Gets an option from the command line
80 *	putenv()	Writes values into the environment
81 *	exit()		Exit the process
82 *	access()	Check accessibility of a file
83 *	malloc()	Allocate a block of main memory
84 *	free()		Free allocated space
85 *	lseek()		Seek within a file
86 *	open()		Open a file
87 *	close()		Close an open file
88 */
89
90extern	int		optind;		/* argv[] index of next arg */
91extern	int		opterr;		/* TRUE if getopt() is to print msgs */
92extern	char	       *optarg;		/* Argument to parsed option */
93extern	int		getopt();	/* Get an option from the command line */
94extern	int	       	putenv();	/* Put a value into the environment */
95extern	void		exit();		/* Exit the process */
96extern	int		access();	/* Check the accessibility of a file */
97extern	void	       *malloc();	/* Get a chunk of main memory */
98extern	void		free();		/* Free alloc'd space */
99extern	long		lseek();	/* Seek within a file */
100extern	int		open();		/* Open a file */
101extern	int		close();	/* Close an open a file */
102
103/*
104 *  L O C A L   D E F I N I T I O N S
105 */
106
107/*
108 * Constants
109 */
110
111#ifndef	TRUE
112#define	TRUE		(1)
113#endif
114
115#ifndef	FALSE
116#define	FALSE		(0)
117#endif
118
119#ifndef	NULL
120#define	NULL		(0)
121#endif
122
123#define	OPTSTRING	"tn:s:"
124
125#define	EX_OK		0
126#define	EX_ERROR	1
127
128#define	RC_FILENAME	"/etc/rc2.d/S18setuname"
129#define RC_DIRNAME	"/etc/rc2.d"
130
131
132/*
133 *  Messages
134 */
135
136#define	E_USAGE		"usage: setuname [-t] [-s name] [-n node]"
137#define	E_MISSING	"Either -s name or -n node must be specified"
138#define	E_UNAME		"Unable to get existing uname values"
139#define E_INVNAME	"System-name invalid: %s"
140#define E_LONGNAME	"System-name too long: %s"
141#define E_INVNODE	"Network node-name invalid: %s"
142#define E_LONGNODE	"Network node-name too long: %s"
143#define E_NOPERMS	"No permissions, request denied"
144#define E_NOSUCHDIR	"Directory doesn't exist: %s"
145#define	E_INTERNAL	"Internal error: %d"
146
147/*
148 * Macros:
149 *	stdmsg(r,l,s,t)	    Write a standard message.
150 *				'r' is the recoverability flag
151 *				'l' is the label
152 *				's' is the severity
153 *				't' is the text.
154 *	strend(p)	    Return the address of the end of a string
155 *			    (This is supposed to be defined in <sys/inline.h>
156 *			    but that file has string-handing def'ns that
157 *			    conflict with <string.h>, so we can't use it!
158 *			    MR dn89-04701 requests this fix.
159 */
160
161#define	stdmsg(r,l,s,t)	(void) fmtmsg(MM_PRINT|MM_UTIL|r,l,s,t,MM_NULLACT,MM_NULLTAG)
162#define strend(p)       strrchr(p,'\0')
163
164/*
165 * Local functions:
166 *	setuname	Changes the system name and the network node name
167 */
168
169static int	setuname();		/* This does the "real" work */
170
171
172/*
173 * Local data
174 *	lbl		Buffer for the standard message label
175 *	txt		Buffer for the standard message text
176 */
177
178static	char		lbl[MM_MXLABELLN+1];	/* Space for std msg label */
179static	char		msg[MM_MXTXTLN+1];	/* Space for std msg text  */
180
181/*
182 *  int main(argc, argv)
183 *	int	argc
184 *	char   *argv;
185 */
186
187int
188main(argc, argv)
189	int	argc;			/* Argument count  */
190	char   *argv[];			/* Argument vector */
191{
192	/* Automatic data */
193	char	       *n_arg;			/* Ptr to arg for -n */
194	char	       *s_arg;			/* Ptr to arg for -s */
195	int		t_seen;			/* FLAG, -t option seen */
196	char	       *cmdname;		/* Ptr to the command's name */
197	char	       *p;			/* Temp pointer */
198	int		usageerr;		/* FLAG, TRUE if usage error */
199	int		exitcode;		/* Value to exit with */
200	int		c;			/* Temp character */
201	int		ok;			/* Flag, everything okay? */
202
203	/* Build the standard-message label */
204	if (p = strrchr(argv[0], '/')) cmdname = p+1;
205	else cmdname = argv[0];
206	(void) strcat(strcpy(lbl, "UX:"), cmdname);
207
208	/* Make only the text in standard messages appear (SVR4.0 only) */
209	(void) putenv("MSGVERB=text");
210
211
212	/* Initializations */
213	n_arg = s_arg = (char *) NULL;
214	t_seen = FALSE;
215
216
217	/*
218	 * Parse command
219	 */
220
221	usageerr = FALSE;
222	opterr = FALSE;
223	while (!usageerr && (c = getopt(argc, argv, OPTSTRING)) != EOF) switch(c) {
224
225	case 'n':			/* -n node */
226	    if (n_arg) usageerr = TRUE;
227	    else n_arg = optarg;
228	    break;
229
230	case 's':			/* -s name */
231	    if (s_arg) usageerr = TRUE;
232	    else s_arg = optarg;
233	    break;
234
235	case 't':			/* -t */
236	    if (t_seen) usageerr = TRUE;
237	    else t_seen = TRUE;
238	    break;
239
240	default:			/* Something that doesn't exist */
241	    usageerr = TRUE;
242	}   /* switch() */
243
244	/* If there was a usage error, report the error and exit */
245	if ((argc >= (optind+1)) || usageerr) {
246	    stdmsg(MM_NRECOV, lbl, MM_ERROR, E_USAGE);
247	    exit(EX_ERROR);
248	}
249
250	/* Either -n <node> or -s <name> has to be specified */
251	if (!(n_arg || s_arg)) {
252	    stdmsg(MM_NRECOV, lbl, MM_ERROR, E_MISSING);
253	    exit(EX_ERROR);
254	}
255
256
257	/*
258	 * Validate arguments:
259	 *  - The length of the system name must be less than SYS_NMLN-1
260	 *    characters,
261	 *  - The length of the network node-name must be less than
262	 *    SYS_NMLN-1 characters,
263	 *  - The system name must equal [a-zA-Z0-9-_]+,
264	 *  - The network node-name must equal [a-zA-Z0-9-_]+.
265	 */
266
267	/* Check the length and the character-set of the system name */
268	if (s_arg) {
269
270	    /* Check length of the system name */
271	    if (strlen(s_arg) > (size_t)(SYS_NMLN-1)) {
272		(void) sprintf(msg, E_LONGNAME, s_arg);
273		stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
274		exit(EX_ERROR);
275	    }
276
277	    /* Check the character-set */
278	    ok = TRUE;
279	    for (p = s_arg ; ok && *p ; p++) {
280		if (!isalnum(*p) && (*p != '-') && (*p != '_')) ok = FALSE;
281	    }
282	    if (!ok || (p == s_arg)) {
283		(void) sprintf(msg, E_INVNAME, s_arg);
284		stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
285		exit(EX_ERROR);
286	    }
287	}
288
289	/* Check the length and the character-set of the network node-name */
290
291	if (n_arg) {
292
293	    /* Check length of the network node-name */
294	    if (strlen(n_arg) > (size_t)(SYS_NMLN-1)) {
295		(void) sprintf(msg, E_LONGNODE, n_arg);
296		stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
297		exit(EX_ERROR);
298	    }
299
300	    /* Check the character-set */
301	    ok = TRUE;
302	    for (p = n_arg ; ok && *p ; p++) {
303		if (!isalnum(*p) && (*p != '-') && (*p != '_')) ok = FALSE;
304	    }
305	    if (!ok || (p == n_arg)) {
306		(void) sprintf(msg, E_INVNODE, n_arg);
307		stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
308		exit(EX_ERROR);
309	    }
310	}
311
312
313	/*
314	 * Make sure we have access to needed resources:
315	 *   -  Read/write access to kernel memory (/dev/kmem)
316	 *   -  If -t is not specified, read/write access to /etc/rc2.d
317	 *   -  If -t is not specified, read access to /etc/rc2.d/S18setuname
318	 */
319
320	if (access("/dev/kmem", R_OK|W_OK) == 0) {
321	    if (access(RC_DIRNAME, R_OK|W_OK) == 0) {
322		if ((access(RC_FILENAME, R_OK) != 0) &&
323		    (access(RC_FILENAME, F_OK) == 0)) {
324		    stdmsg(MM_NRECOV, lbl, MM_ERROR, E_NOPERMS);
325		    exit(EX_ERROR);
326		}
327	    }
328	    else {
329		if (access(RC_DIRNAME, F_OK) == 0) {
330		    stdmsg(MM_NRECOV, lbl, MM_ERROR, E_NOPERMS);
331		    exit(EX_ERROR);
332		}
333		else {
334		    (void) sprintf(msg, E_NOSUCHDIR, RC_DIRNAME);
335		    stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
336		    exit(EX_ERROR);
337		}
338	    }
339	}
340	else {
341	    stdmsg(MM_NRECOV, lbl, MM_ERROR, E_NOPERMS);
342	    exit(EX_ERROR);
343	}
344
345
346	/* Attempt the setuname */
347	if (setuname(t_seen, s_arg, n_arg) == 0) exitcode = EX_OK;
348	else {
349	    (void) sprintf(msg, E_INTERNAL, errno);
350	    stdmsg(MM_NRECOV, lbl, MM_ERROR, msg);
351	    exitcode = EX_ERROR;
352	}
353
354	/* Finished */
355	return (exitcode);
356}  /* main() */
357
358/*
359 * int setuname(temp, name, node)
360 *	int	temp
361 *	char   *name
362 *	char   *node
363 *
364 *	Set any or all of the following machine parameters, either
365 *	temporarily or permanently, depending on <temp>.
366 *	    - System name
367 *	    - Network Node-name
368 */
369
370static int
371setuname(temp, sysname, nodename)
372	int	temp;		/* Set in kernel only flag */
373	char   *sysname;	/* System name */
374	char   *nodename;	/* Network node-name */
375{
376	/* Automatic Data */
377	struct utsname	utsname;	/* Space for the kernel's utsname information */
378#if u3b || u3b15 || u3b2
379	struct s3bsym  *symbtbl;	/* The kernel's symbol table */
380#endif
381#if sun
382	struct nlist nl[] = {
383		{"utsname", 0, 0, 0, 0, 0},
384		{NULL}
385	};
386	kvm_t *kd;
387#endif
388	uintptr_t	utsname_addr;	/* Addr of "utsname" in the kernel */
389	char	       *sysnm = (char *)NULL;		/* System name to set (from file or arg) */
390	char	       *nodenm = (char *)NULL;		/* Network node-name to set (from file or arg) */
391	FILE	       *fd;		/* Std I/O File Descriptor for /etc/rc2.d/S18setuname */
392	char	       *p;		/* Temp pointer */
393	void	      (*oldsighup)();	/* Function to call for SIGHUP */
394	void	      (*oldsigint)();	/* Function to call for SIGINT */
395	int		rtncd;		/* Value to return to the caller */
396	unsigned long	symbtblsz;	/* The size of the kernel's symbol table, in bytes */
397	int		memfd;		/* File descriptor:  open kernel memory */
398	int		i;		/* Temp counter */
399
400
401	/* Nothing's gone wrong yet (but we've only just begun!) */
402	rtncd = 0;
403
404
405	/*
406	 * Get the virtual address of the symbol "utsname" in the kernel
407	 * so we can get set the system name and/or the network node-name
408	 * directly in the kernel's memory space.
409	 */
410
411#if u3b || u3b15 || u3b2
412	if ((sys3b(S3BSYM, (struct s3bsym *) &symbtblsz, sizeof(symbtblsz)) == 0) &&
413	    (symbtbl = (struct s3bsym *) malloc(symbtblsz))) {
414
415	    (void) sys3b(S3BSYM, symbtbl, symbtblsz);
416	    p = (char *) symbtbl;
417	    for (i = symbtbl->count; i-- && (strcmp(p, "utsname") != 0) ; p = S3BNXTSYM(p)) ;
418	    if (i >= 0) utsname_addr = S3BSVAL(p);
419	    else rtncd = -1;
420	    free((void *) symbtbl);
421
422	} else rtncd = -1;
423
424#elif sun
425        /* Check out namelist and memory files. */
426	if ((kd = kvm_open(NULL, NULL, NULL, O_RDWR, NULL)) == NULL)
427		rtncd = -1;
428	else if (kvm_nlist(kd, nl) != 0)
429		rtncd = -1;
430	else if (nl[0].n_value == 0)
431		rtncd = -1;
432	else
433		utsname_addr = (uintptr_t)nl[0].n_value;
434#else
435	if (nlist("/unix", nl) != 0)
436		rtncd = -1;
437#endif
438	if (rtncd != 0) return(rtncd);
439
440	/*
441	 * Open the kernel's memory, get the existing "utsname" structure,
442	 * change the system name and/or the network node-name in that struct,
443	 * write it back out to kernel memory, then close kernel memory.
444	 */
445#ifdef sun
446	if (kvm_kread(kd, utsname_addr, &utsname, sizeof (utsname)) ==
447	    sizeof (utsname)) {
448		if (sysname)
449			(void) strncpy(utsname.sysname, sysname,
450			    sizeof (utsname.sysname));
451		if (nodename)
452			(void) strncpy(utsname.nodename, nodename,
453			    sizeof (utsname.nodename));
454		(void) kvm_kwrite(kd, utsname_addr, &utsname, sizeof (utsname));
455		kvm_close(kd);
456	} else
457		return (-1);
458#else /* sun */
459	if ((memfd = open("/dev/kmem", O_RDWR, 0)) > 0) {
460	    if ((lseek(memfd, (long) utsname_addr, SEEK_SET) != -1) &&
461		(read(memfd, &utsname, sizeof(utsname)) == sizeof(utsname))) {
462		if (sysname) (void) strncpy(utsname.sysname, sysname, sizeof(utsname.sysname));
463		if (nodename) (void) strncpy(utsname.nodename, nodename, sizeof(utsname.nodename));
464		(void) lseek(memfd, (long) utsname_addr, SEEK_SET);
465		(void) write(memfd, &utsname, sizeof(utsname));
466		(void) close(memfd);
467	    } else rtncd = -1;
468	} else rtncd = -1;
469	if (rtncd != 0) return(rtncd);
470#endif /* sun */
471
472
473	/*
474	 * If the "temp" flag is FALSE, we need to permanently set the
475	 * system name in the file  /etc/rc2.d/S18setuname
476	 */
477
478	if (!temp) {
479	    /*
480	     * If a name was specified by the caller, use that, otherwise, use
481	     * whatever was in the "rc" file.
482	     */
483
484		if (sysname) sysnm = sysname;
485		if (nodename) nodenm = nodename;
486
487
488	    /*
489	     * Write the file /etc/rc2.d/S18setuname so that the system name is
490	     * set on boots and state changes.
491	     *
492	     * DISABLED SIGNALS: SIGHUP, SIGINT
493	     */
494
495	    /* Give us a reasonable chance to complete without interruptions */
496		oldsighup = signal(SIGHUP, SIG_IGN);
497		oldsigint = signal(SIGINT, SIG_IGN);
498
499	    /* Write the new setuname "rc" file */
500		if (sysname != NULL) {
501			if ((fd = fopen(RC_FILENAME, "w")) != (FILE *) NULL) {
502				(void) fprintf(fd, "# %s\n", sysnm);
503				(void) fprintf(fd, "#\n");
504				(void) fprintf(fd, "# This script, generated by the setuname command,\n");
505				(void) fprintf(fd, "# sets the system's system-name\n");
506				(void) fprintf(fd, "#\n");
507			if (sysnm && *sysnm)
508				(void) fprintf(fd, "setuname -t -s %s\n", sysnm);
509			(void) fclose(fd);
510			} else return(rtncd = -1);
511		}
512
513		if(nodename != NULL) {
514			char curname[SYS_NMLN];
515			int curlen;
516			FILE *file;
517
518			if ((file = fopen("/etc/nodename", "r")) != NULL) {
519				curlen = fread(curname, sizeof(char), SYS_NMLN, file);
520				for (i = 0; i < curlen; i++) {
521					if (curname[i] == '\n') {
522						curname[i] = '\0';
523						break;
524					}
525				}
526				if (i == curlen) {
527					curname[curlen] = '\0';
528				}
529				(void)fclose(file);
530			} else {
531				curname[0] = '\0';
532			}
533			if (strcmp(curname, nodenm) != 0) {
534				if ((file = fopen("/etc/nodename", "w")) == NULL) {
535					(void) fprintf(stderr, "setuname: error in writing name\n");
536					exit(1);
537				}
538				if (fprintf(file, "%s\n", nodenm) < 0) {
539					(void) fprintf(stderr, "setuname: error in writing name\n");
540					exit(1);
541				}
542				(void)fclose(file);
543			}
544		}
545	    /* Restore signal handling */
546		(void) signal(SIGHUP, oldsighup);
547		(void) signal(SIGINT, oldsigint);
548	}	/* if (!temp) */
549
550	/* Fini */
551	return(rtncd);
552}
553