2 files changed, 30 insertions, 0 deletions

diff --git a/classes/meta-virt-hosts.bbclass b/classes/meta-virt-hosts.bbclass
new file mode 100644
index 00000000..80aefb76
--- /dev/null
+++ b/classes/meta-virt-hosts.bbclass
@@ -0,0 +1,29 @@
+# This doesn't work, since it seems to be too late for sanity checking.
+# IMAGE_FEATURES[validitems] += ' ${@bb.utils.contains("DISTRO_FEATURES", "virtualization", "virt-unique-hostname; ", "",d)}'
+
+ROOTFS_POSTPROCESS_COMMAND += '${@bb.utils.contains("IMAGE_FEATURES", "virt-unique-hostname", "virt_gen_hostname; ", "",d)}'
+ROOTFS_POSTPROCESS_COMMAND += '${@bb.utils.contains("IMAGE_FEATURES", "virt-unique-hostname", "virt_set_hostname; ", "",d)}'
+
+python virt_gen_hostname() {
+    import uuid
+
+    targetname = d.getVar("VIRT_TARGETNAME")
+    if targetname != None:
+        return
+
+    status, date = oe.utils.getstatusoutput("date +%d-%m-%y")
+    if status:
+        bb.warn("Can't get the date string for target hostname")
+
+    uuid = ':'.join(['{:02x}'.format((uuid.getnode() >> ele) & 0xff) for ele in range(0,8*6,8)][::-1])
+    if uuid:
+        targetname = "%s-%s" %  (d.getVar("MACHINE"), uuid.split(":")[0])
+    else:
+        targetname = "%s-%s" %  (d.getVar("MACHINE"), date)
+
+    d.setVar("VIRT_TARGETNAME", targetname)
+}
+
+virt_set_hostname() {
+    echo "${VIRT_TARGETNAME}" > ${IMAGE_ROOTFS}/etc/hostname
+}
diff --git a/conf/layer.conf b/conf/layer.conf
index 149b42ee..4a1448a4 100644
--- a/conf/layer.conf
+++ b/conf/layer.conf
@@ -57,5 +57,6 @@ K8S_CONFIG_PATH = "${LAYERDIR}/conf/distro/include/k8s-versions.inc"
 USER_CLASSES:append = " meta-virt-cfg"
 USER_CLASSES:append = " meta-virt-k8s-cfg"
 USER_CLASSES:append = " meta-virt-xen-cfg"
+USER_CLASSES:append = " meta-virt-hosts"
 
 HOSTTOOLS_NONFATAL += "getent"