scripts/system/add_domain.sh

   1 #!/bin/bash
   2
   3 # this set of functions serve the main purpose of adding new domains or subdomains to the bind9 DNS server on the current host.
   4 # it is currently highly specific to running a bunch of domains on a linux VM, where the VM has one IP address.
   5 # note that bind 'named' must already be configured.
   6 # also, it is assumed that if a subdomain is being added, then the containing domain has already been configured and is
   7 # configured in a file similar to "blah.com.conf" in /etc/bind.
   8 #
   9 # Author: Chris Koeritz
  10
  11 # some defaults that are convenient for current purposes.
  12 # hmmm: these would need to be parameterized somehow for this script to become really general.
  13
  14 # in our scheme, the single IP address that all our domains map to.
  15 IP_ADDRESS="10.28.42.20"
  16 # the email address (where first dot is replaced by @) for the administrator of the domain.
  17 SERVER_ADMIN="fred.cakelampvm.com"
  18 # the name of the name server for the new domains (should already be configured).
  19 MAIN_NAME_SERVER="ns.cakelampvm.com"
  20 # the name of the mail server for a new domain (should already be configured).
  21 MAIL_SERVER="mail.cakelampvm.com"
  22 # the distribution name to be listed in info for the new domain or subdomain.
  23 DISTRO="ubuntu"
  24
  25 # creates a totally new domain config file for DNS.
  26 function write_new_domain_file()
  27 {
  28   local domain_name="$1"; shift
  29
  30   local domain_file="/etc/bind/${domain_name}.conf"
  31
  32   echo "adding a totally new domain called $domain_name"
  33   echo "using the config file: $domain_file"
  34
  35   if [ -f $domain_file ]; then
  36     echo "The domain configuration file already exists at:"
  37     echo "  $domain_file"
  38     echo "Please remove this file before proceeding, if it is junk.  For example:"
  39     echo "  sudo rm $domain_file"
  40     exit 1
  41   fi
  42
  43   echo "
  44 \$TTL 1W
  45 @       IN SOA  @       ${SERVER_ADMIN}. (
  46                 2017100801 ; serial
  47                 2H ; refresh
  48                 8M ; retry
  49                 14D ; expiry
  50                 6H ) ; minimum
  51
  52         IN NS           ${MAIN_NAME_SERVER}.
  53         IN MX   10      ${MAIL_SERVER}.
  54
  55 ${domain_name}. IN A    ${IP_ADDRESS}
  56         IN HINFO        \"linux server\" \"${DISTRO}\"
  57 " >"$domain_file"
  58 }
  59
  60 # hooks up a new config file into bind's list of zones.
  61 function add_zone_for_new_domain()
  62 {
  63   local domain_name="$1"; shift
  64
  65   local domain_file="/etc/bind/${domain_name}.conf"
  66
  67   echo "adding a new domain configured by ${domain_file} into"
  68   echo "the named.conf.local configuration file."
  69
  70   # append the reference to the new conf file in the zone list.
  71   echo "
  72 zone \"${domain_name}\" in {
  73         file \"${domain_file}\";
  74         type master;
  75         allow-query { any; };
  76 };
  77
  78 ////////////////////////////////////////////////////////////////////////////
  79
  80 " >> /etc/bind/named.conf.local
  81 }
  82
  83 # adds a new subdomain under a containing domain.
  84 function add_new_subdomain()
  85 {
  86   local new_domain="$1"; shift
  87
  88   # split up the full domain name into subdomain portion and containing domain.
  89   local subdomain="${new_domain%.*.*}"
  90   local containing_domain="${new_domain#*.}"
  91
  92   echo "adding a subdomain $subdomain to containing domain $containing_domain"
  93
  94   local domain_file="/etc/bind/${containing_domain}.conf"
  95   # see if config file already exists; if not, complain.
  96   if [ ! -f "$domain_file" ]; then
  97     echo "The domain configuration file for $new_domain is missing."
  98     echo "It should already be present in: $domain_file"
  99     echo "Please add the containing domain before trying to add a subdomain."
 100     exit 1
 101   fi
 102
 103   # see if subdomain already present in config.
 104   if [ $(grep -q "$new_domain" "$domain_file") ]; then
 105     echo "The subdomain $subdomain already seems to exist in the domain"
 106     echo "configuration file: $domain_file"
 107     echo "Please edit the config file to remove the subdomain before trying"
 108     echo "to re-add the subdomain."
 109     exit 1
 110   fi
 111
 112   # append the new subdomain into the config file.
 113   echo "
 114 ${subdomain}.${containing_domain}.    IN A    ${IP_ADDRESS}
 115         IN HINFO \"linux server\" \"${DISTRO}\"
 116 " >> /etc/bind/${containing_domain}.conf
 117
 118 }
 119
 120 function restart_bind()
 121 {
 122   echo restarting DNS server.
 123   service bind9 restart
 124   if [ $? -ne 0 ]; then
 125     echo "The bind service did not restart properly.  Please check the error logs."
 126     exit 1
 127   fi
 128   echo DNS server restarted.
 129 }
 130
 131 # main body of script.
 132
 133 if (( $EUID != 0 )); then
 134   echo "This script must be run as root or sudo."
 135   exit 1
 136 fi
 137
 138 new_domain="$1"; shift
 139
 140 if [ -z "$new_domain" ]; then
 141   echo "This script needs a domain name to add to DNS."
 142   exit 1
 143 fi
 144
 145 # if domain name has three or more components, then add a subdomain.
 146 # otherwise, add a full new domain.
 147 if [[ $new_domain == *"."*"."* ]]; then
 148   # add a subdomain to the containing domain.
 149   add_new_subdomain "$new_domain"
 150   restart_bind
 151 else
 152   # create a totally new domain in DNS.
 153   write_new_domain_file "$new_domain"
 154   add_zone_for_new_domain "$new_domain"
 155   restart_bind
 156 fi
 157
 158