Code cleanup
[snf-image] / snf-image-helper / tasks / 50ChangePassword.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:             ChangePassword
22 # RunBefore:            EnforcePersonality
23 # RunAfter:             InstallUnattend
24 # Short-Description:    Changes Password for specified users
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_password() {
37     local target password
38     target="$1"
39     password="$2"
40
41     echo "@echo off" > "$target/Windows/SnfScripts/ChangeAdminPassword.cmd"
42
43     if [ -z "$SNF_IMAGE_PROPERTY_USERS" ]; then
44         warn "Image property \`USERS' is missing or empty. " \
45             "Changing the password for default user: \`Administrator'."
46
47         SNF_IMAGE_PROPERTY_USERS="Administrator"
48     fi
49
50     for usr in $SNF_IMAGE_PROPERTY_USERS; do
51         echo -n "Installing new password for user \`$usr'..."
52         echo "net user $usr $password" >> \
53             "$target/Windows/SnfScripts/ChangeAdminPassword.cmd"
54         echo done
55     done
56 }
57
58 linux_password() {
59     local target password hash users tmp_shadow
60     target="$1"
61     password="$2"
62
63     hash=$("@scriptsdir@/snf-passtohash.py" "$password")
64     if [ ! -e "$target/etc/shadow" ]; then
65        log_error "No /etc/shadow found!" 
66     fi
67     
68     users=()
69     
70     if [ -n "$SNF_IMAGE_PROPERTY_USERS" ]; then
71         for usr in $SNF_IMAGE_PROPERTY_USERS; do
72             users+=("$usr")
73         done
74     else
75         warn "Image property \`USERS' is missing or empty. " \
76             "Changing the password for default user: \`root'."
77         users+=("root")
78     fi
79
80     for i in $(seq 0 1 $((${#users[@]}-1))); do
81         tmp_shadow="$(mktemp)"
82         add_cleanup rm "$tmp_shadow"
83
84         echo -n "Setting ${users[$i]} password..."
85         if ! grep "^${users[$i]}:" "$target/etc/shadow" > /dev/null; then
86             log_error "User: \`${users[$i]}' does not exist."
87         fi
88     
89         echo "${users[$i]}:$hash:15103:0:99999:7:::" > "$tmp_shadow"
90         grep -v "${users[$i]}" "$target/etc/shadow" >> "$tmp_shadow"
91         cat "$tmp_shadow" > "$target/etc/shadow"
92         echo "done"
93     done
94 }
95
96 if [ ! -d "$SNF_IMAGE_TARGET" ]; then
97     log_error "Target dir: \`$SNF_IMAGE_TARGET' is missing"
98 fi
99
100 if [ -z "$SNF_IMAGE_PASSWORD" ]; then
101     log_error "Password is missing"
102 fi
103
104 #trim users var
105 SNF_IMAGE_PROPERTY_USERS=$(echo $SNF_IMAGE_PROPERTY_USERS)
106
107 if [ "$SNF_IMAGE_PROPERTY_OSFAMILY" = "windows" ]; then
108     windows_password "$SNF_IMAGE_TARGET" "$SNF_IMAGE_PASSWORD"
109 elif [ "$SNF_IMAGE_PROPERTY_OSFAMILY" = "linux" ]; then
110     linux_password "$SNF_IMAGE_TARGET" "$SNF_IMAGE_PASSWORD"
111 fi
112
113 exit 0
114
115 # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :
116