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 "Since we don't want to tear that down if it has specialized configuration"
  39     echo "data in it, we will just leave it in place and consider our job done.
  40     exit 0
  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