1#!/bin/sh
2#
3# Copyright (c) 2007 Pawel Jakub Dawidek <pjd@FreeBSD.org>
4# All rights reserved.
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"
37start_cmd="hostid_start"
38stop_cmd=":"
39reset_cmd="hostid_reset"
40extra_commands="reset"
41rcvar="hostid_enable"
42
43hostid_set()
44{
45	uuid=$1
46	# Generate hostid based on hostuuid - take first four bytes from md5(uuid).
47	id=`echo -n $uuid | /sbin/md5`
48	id="0x${id%????????????????????????}"
49
50	# Set both kern.hostuuid and kern.hostid.
51	#
52	check_startmsgs && echo "Setting hostuuid: ${uuid}."
53	${SYSCTL} kern.hostuuid="${uuid}" >/dev/null
54	check_startmsgs && echo "Setting hostid: ${id}."
55	${SYSCTL} kern.hostid=${id} >/dev/null
56}
57
58hostid_hardware()
59{
60	uuid=`kenv -q smbios.system.uuid`
61	x="[0-9a-f]"
62	y=$x$x$x$x
63	case "${uuid}" in
64	$y$y-$y-$y-$y-$y$y$y)
65		echo "${uuid}"
66		;;
67	esac
68}
69
70hostid_generate()
71{
72	# First look for UUID in hardware.
73	uuid=`hostid_hardware`
74	if [ -z ${uuid} ]; then
75		# If not found, fall back to software-generated UUID.
76		uuid=`uuidgen`
77	fi
78	hostid_set $uuid
79}
80
81hostid_reset()
82{
83	hostid_generate
84	# Store newly generated UUID in ${hostid_file}.
85	echo $uuid > ${hostid_file}
86	if [ $? -ne 0 ]; then
87		warn "could not store hostuuid in ${hostid_file}."
88	fi
89}
90
91hostid_start()
92{
93	# If ${hostid_file} already exists, we take UUID from there.
94	if [ -r ${hostid_file} ]; then
95		hostid_set `cat ${hostid_file}`
96	else
97		# No hostid file, generate UUID.
98		hostid_generate
99	fi
100}
101
102load_rc_config $name
103run_rc_command "$1"
104