xref: /freebsd/libexec/rc/rc.d/hostid (revision 4d3fc8b0)
1#!/bin/sh
2#
3# Copyright (c) 2007 Pawel Jakub Dawidek <pjd@FreeBSD.org>
4# Copyright (c) 2015 Xin LI <delphij@FreeBSD.org>
5#
6# Redistribution and use in source and binary forms, with or without
7# modification, are permitted provided that the following conditions
8# are met:
9# 1. Redistributions of source code must retain the above copyright
10#    notice, this list of conditions and the following disclaimer.
11# 2. Redistributions in binary form must reproduce the above copyright
12#    notice, this list of conditions and the following disclaimer in the
13#    documentation and/or other materials provided with the distribution.
14#
15# THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25# SUCH DAMAGE.
26#
27# $FreeBSD$
28#
29
30# PROVIDE: hostid
31# REQUIRE: sysctl
32# KEYWORD: nojail
33
34. /etc/rc.subr
35
36name="hostid"
37desc="Generate a unique host ID"
38start_cmd="hostid_start"
39stop_cmd=":"
40reset_cmd="hostid_reset"
41extra_commands="reset"
42rcvar="hostid_enable"
43
44hostid_set()
45{
46	uuid=$1
47	# Generate hostid based on hostuuid - take first four bytes from md5(uuid).
48	id=`echo -n $uuid | /sbin/md5`
49	id="0x${id%????????????????????????}"
50
51	# Set both kern.hostuuid and kern.hostid.
52	#
53	startmsg "Setting hostuuid: ${uuid}."
54	${SYSCTL} kern.hostuuid="${uuid}" >/dev/null
55	startmsg "Setting hostid: ${id}."
56	${SYSCTL} kern.hostid=${id} >/dev/null
57}
58
59valid_hostid()
60{
61	uuid=$1
62
63	x="[0-9a-f]"
64	y=$x$x$x$x
65
66	# Check against a blacklist before
67	# accepting the UUID.
68	case "${uuid}" in
69	00000000-0000-0000-0000-000000000000)
70		;;
71	00020003-0004-0005-0006-000700080009)
72		;;
73	03000200-0400-0500-0006-000700080009)
74		;;
75	07090201-0103-0301-0807-060504030201)
76		;;
77	11111111-1111-1111-1111-111111111111)
78		;;
79	11111111-2222-3333-4444-555555555555)
80		;;
81	4c4c4544-0000-2010-8020-80c04f202020)
82		;;
83	58585858-5858-5858-5858-585858585858)
84		;;
85	890e2d14-cacd-45d1-ae66-bc80e8bfeb0f)
86		;;
87	8e275844-178f-44a8-aceb-a7d7e5178c63)
88		;;
89	dc698397-fa54-4cf2-82c8-b1b5307a6a7f)
90		;;
91	fefefefe-fefe-fefe-fefe-fefefefefefe)
92		;;
93	*-ffff-ffff-ffff-ffffffffffff)
94		;;
95	$y$y-$y-$y-$y-$y$y$y)
96		return 0
97		;;
98	esac
99
100	return 1
101}
102
103hostid_hardware()
104{
105	uuid=`kenv -q smbios.system.uuid`
106
107	if valid_hostid $uuid; then
108		echo "${uuid}"
109	elif [ "$uuid" ]; then
110		echo "INVALID"
111	fi
112}
113
114hostid_generate()
115{
116	# First look for UUID in hardware.
117	uuid=`hostid_hardware`
118
119	# Warn about invalid UUIDs
120	if [ "${uuid}" = "INVALID" ]; then
121		warn "hostid: unable to figure out a UUID from DMI data, generating a new one"
122		sleep 2
123		uuid=""
124	fi
125
126	# Generate a random UUID if invalid or not found
127	if [ -z "${uuid}" ]; then
128		# If not found, fall back to software-generated UUID.
129		uuid=`uuidgen ${hostid_uuidgen_flags}`
130	fi
131	hostid_set $uuid
132}
133
134hostid_reset()
135{
136	hostid_generate
137	# Store newly generated UUID in ${hostid_file}.
138	echo $uuid > ${hostid_file}
139	if [ $? -ne 0 ]; then
140		warn "could not store hostuuid in ${hostid_file}."
141	fi
142}
143
144hostid_start()
145{
146	# If ${hostid_file} already exists, we take UUID from there.
147	if [ -r ${hostid_file} ]; then
148		read saved_hostid < ${hostid_file}
149		if valid_hostid ${saved_hostid}; then
150			hostid_set ${saved_hostid}
151			exit 0
152		fi
153	fi
154
155	# No hostid file, generate UUID.
156	hostid_generate
157}
158
159load_rc_config $name
160run_rc_command "$1"
161