code.grnet.gr Git - snf-image/blob - snf-image-helper/tasks/50AssignHostname.in

   1 #! /bin/bash
   2
   3 # Copyright (C) 2011 GRNET S.A.
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18 # 02110-1301, USA.
  19
  20 ### BEGIN TASK INFO
  21 # Provides:             AssignHostname
  22 # RunBefore:            EnforcePersonality
  23 # RunAfter:             InstallUnattend
  24 # Short-Description:    Assign Hostname/Computer Name to the instance
  25 ### END TASK INFO
  26
  27 set -e
  28 . "@commondir@/common.sh"
  29
  30 trap task_cleanup EXIT
  31 report_task_start
  32
  33 # Check if the task should be prevented from running.
  34 check_if_excluded
  35
  36 windows_hostname() {
  37     local target="$1"
  38     local password="$2"
  39
  40     local tmp_unattend=$(mktemp) || exit 1
  41     add_cleanup rm "$tmp_unattend"
  42
  43     echo -n "Assigning new computer name..."
  44
  45     local namespace="urn:schemas-microsoft-com:unattend"
  46
  47     unattend=$(get_unattend "$target")
  48     if [ -z "$unattend" ]; then
  49         log_error "Unattend.xml is missing."
  50     fi
  51
  52     "$XMLSTARLET" ed -N x=$namespace -u "/x:unattend/x:settings/x:component/x:ComputerName" -v "$password" "$unattend" > "$tmp_unattend"
  53
  54     cat "$tmp_unattend" > "$unattend"
  55     echo done
  56 }
  57
  58 linux_hostname() {
  59     local target="$1"
  60     local hostname="$2"
  61
  62     local distro=$(get_base_distro "$target")
  63
  64     case "$distro" in
  65         debian)
  66             echo "$hostname" > "$target/etc/hostname";;
  67         redhat)
  68             sed -ie "s/HOSTNAME=.*$/HOSTNAME=$hostname/g" "$target/etc/sysconfig/network";;
  69         slackware|suse)
  70             #local domain=$(sed -e 's/^[^\.]*//g' < /etc/HOSTNAME)
  71
  72             # In slackware hostname and domain name are joined together. For now
  73             # I will not retain the domain name.
  74             echo "$hostname" > "${target}/etc/HOSTNAME";;
  75         gentoo)
  76             sed -ie "s/\(\(HOSTNAME\)\|\(hostname\)\)=.*$/\1=\"$hostname\"/" "$target/etc/conf.d/hostname";;
  77         arch)
  78             # In newer archlinux systems /etc/hostname is used to assign the hostname
  79             if [[ -f "$target/etc/hostname" ]]; then
  80                 echo "$hostname" > "$target/etc/hostname"
  81             fi
  82             # Older ones use the HOSTNAME variable in /etc/rc.conf
  83             sed -ie "s/^HOSTNAME=.*$/HOSTNAME=\"$hostname\"/" "$target/etc/rc.conf"
  84             if grep "^127\.0\.0\.1[ \t]*" "$target/etc/hosts" > /dev/null; then
  85                sed -ie "s/127\.0\.0\.1[ \t]*.*$/127.0.0.1\t$hostname/" "$target/etc/hosts"
  86             else
  87                echo -e "127.0.0.1\t$hostname" >> "$target/etc/hosts"
  88             fi;;
  89         *) log_error "Don't know how to assign hostname. Unknown linux distribution.";;
  90     esac
  91
  92     # Some Linux distributions assign the hostname to 127.0.1.1 in order to be
  93     # resolvable to an IP address. Lets replace this if found in /etc/hosts
  94     sed -ie "s/^[[:blank:]]*127\.0\.1\.1[[:blank:]].\+$/127.0.1.1\t$hostname/" "$target/etc/hosts"
  95 }
  96
  97 if [ ! -d "$SNF_IMAGE_TARGET" ]; then
  98     log_error "Target dir: \`$SNF_IMAGE_TARGET' is missing"
  99 fi
 100
 101 if [ -z "$SNF_IMAGE_HOSTNAME" ]; then
 102     log_error "Hostname is missing"
 103 fi
 104
 105 if [ "$SNF_IMAGE_PROPERTY_OSFAMILY" = "windows" ]; then
 106     windows_hostname "$SNF_IMAGE_TARGET" "$SNF_IMAGE_HOSTNAME"
 107 elif [ "$SNF_IMAGE_PROPERTY_OSFAMILY" = "linux" ]; then
 108     linux_hostname "$SNF_IMAGE_TARGET" "$SNF_IMAGE_HOSTNAME"
 109 fi
 110
 111 exit 0
 112
 113 # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :
 114