cadi/aaf/src/main/java/org/onap/aaf/cadi/configure/PlaceArtifactScripts.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.cadi.configure;
  23
  24 import java.io.File;
  25
  26 import org.onap.aaf.cadi.CadiException;
  27 import org.onap.aaf.cadi.util.Chmod;
  28 import org.onap.aaf.misc.env.Trans;
  29 import org.onap.aaf.misc.env.util.Chrono;
  30 import org.onap.aaf.misc.env.util.Split;
  31
  32 import certman.v1_0.Artifacts.Artifact;
  33 import certman.v1_0.CertInfo;
  34
  35 public class PlaceArtifactScripts extends ArtifactDir {
  36         @Override
  37         public boolean _place(Trans trans, CertInfo certInfo, Artifact arti) throws CadiException {
  38                 try {
  39                         // Setup check.sh script
  40                         String filename = arti.getNs()+".check.sh";
  41                         File f1 = new File(dir,filename);
  42                         String email = arti.getNotification() + '\n';
  43                         if(email.startsWith("mailto:")) {
  44                                 email=email.substring(7);
  45                         }  else {
  46                                 email=arti.getOsUser() + '\n';
  47                         }
  48
  49                         StringBuilder classpath = new StringBuilder();
  50                         boolean first = true;
  51                         for(String pth : Split.split(File.pathSeparatorChar, System.getProperty("java.class.path"))) {
  52                                 if(first) {
  53                                         first=false;
  54                                 } else {
  55                                         classpath.append(File.pathSeparatorChar);
  56                                 }
  57                                 File f = new File(pth);
  58                                 classpath.append(f.getCanonicalPath().replaceAll("[0-9]+\\.[0-9]+\\.[0-9]+","*"));
  59                         }
  60
  61                         write(f1,Chmod.to644,
  62                                         "#!/bin/bash " + f1.getCanonicalPath()+'\n',
  63                                         "# Certificate Manager Check Script\n",
  64                                         "# Check on Certificate, and renew if needed.\n",
  65                                         "# Generated by Certificate Manager " + Chrono.timeStamp()+'\n',
  66                                         "DIR="+arti.getDir()+'\n',
  67                                         "APP="+arti.getNs()+'\n',
  68                                         "EMAIL="+email,
  69                                         "CP=\""+classpath.toString()+"\"\n",
  70                                         checkScript
  71                                         );
  72
  73                         // Setup check.sh script
  74                         File f2 = new File(dir,arti.getNs()+".crontab.sh");
  75                         write(f2,Chmod.to644,
  76                                         "#!/bin/bash " + f2.getCanonicalPath()+'\n',
  77                                         "# Certificate Manager Crontab Loading Script\n",
  78                                         "# Add/Update a Crontab entry, that adds a check on Certificate Manager generated Certificate nightly.\n",
  79                                         "# Generated by Certificate Manager " + Chrono.timeStamp()+'\n',
  80                                         "TFILE=\"/tmp/cmcron$$.temp\"\n",
  81                                         "DIR=\""+arti.getDir()+"\"\n",
  82                                         "CF=\""+arti.getNs()+" Certificate Check Script\"\n",
  83                                         "SCRIPT=\""+f1.getCanonicalPath()+"\"\n",
  84                                         cronScript
  85                                         );
  86
  87                 } catch (Exception e) {
  88                         throw new CadiException(e);
  89                 }
  90                 return true;
  91         }
  92
  93         /**
  94          * Note: java.home gets Absolute Path of Java, where we probably want soft links from
  95          * JAVA_HOME
  96          * @return
  97          */
  98         private final static String javaHome() {
  99                 String rc = System.getenv("JAVA_HOME");
 100                 return rc==null?System.getProperty("java.home"):rc;
 101         }
 102         private final static String checkScript =
 103                         "> $DIR/$APP.msg\n\n" +
 104                         "function mailit {\n" +
 105                         "  if [ -e /bin/mail ]; then\n" +
 106                         "     MAILER=/bin/mail\n" +
 107                         "  elif [ -e /usr/bin/mail ]; then \n" +
 108                         "     MAILER=/usr/bin/mail\n" +
 109                         "  else \n" +
 110                         "     MAILER=\"\"\n" +
 111                         "  fi\n" +
 112                         " if [ \"$MAILER\" = \"\" ]; then\n" +
 113                         "    printf \"$*\"\n" +
 114                         " else \n" +
 115                         "    printf \"$*\" | $MAILER -s \"AAF Certman Notification for `uname -n`\" $EMAIL\n"+
 116                         " fi\n" +
 117                         "}\n\n" +
 118                         javaHome() + "/bin/" +"java -cp $CP " +
 119                                 Agent.class.getName() +
 120                                 " cadi_prop_files=$DIR/$APP.props check 2>  $DIR/$APP.STDERR > $DIR/$APP.STDOUT\n" +
 121                         "case \"$?\" in\n" +
 122                         "  0)\n" +
 123                         "    # Note: Validation will be mailed only the first day after any modification\n" +
 124                         "    if [ \"`find $DIR -mtime 0 -name $APP.check.sh`\" != \"\" ] ; then\n" +
 125                         "       mailit `echo \"Certficate Validated:\\n\\n\" | cat - $DIR/$APP.msg`\n" +
 126                         "    else\n" +
 127                         "       cat $DIR/$APP.msg\n" +
 128                         "    fi\n" +
 129                         "    ;;\n" +
 130                         "  1) mailit \"Error with Certificate Check:\\\\n\\\\nCheck logs $DIR/$APP.STDOUT and $DIR/$APP.STDERR on `uname -n`\"\n" +
 131                         "    ;;\n" +
 132                         "  2) mailit `echo \"Certificate Check Error\\\\n\\\\n\" | cat - $DIR/$APP.msg`\n" +
 133                         "    ;;\n" +
 134                         "  10) mailit `echo \"Certificate Replaced\\\\n\\\\n\" | cat - $DIR/$APP.msg`\n" +
 135                         "      if [ -e $DIR/$APP.restart.sh ]; then\n" +
 136                         "        # Note: it is THIS SCRIPT'S RESPONSIBILITY to notify upon success or failure as necessary!!\n" +
 137                         "        /bin/sh $DIR/$APP.restart.sh\n" +
 138                         "      fi\n" +
 139                         "    ;;\n" +
 140                         "  *) mailit `echo \"Unknown Error code for CM Agent\\\\n\\\\n\" | cat - $DIR/$APP.msg`\n" +
 141                         "    ;;\n" +
 142                         " esac\n\n" +
 143                         " # Note: make sure to cover this sripts' exit Code\n";
 144
 145         private final static String cronScript =
 146                         "crontab -l | sed -n \"/#### BEGIN $CF/,/END $CF ####/!p\" > $TFILE\n" +
 147                         "# Note: Randomize Minutes (0-60) and hours (1-4)\n" +
 148                         "echo \"#### BEGIN $CF ####\" >> $TFILE\n" +
 149                         "echo \"$(( $RANDOM % 60)) $(( $(( $RANDOM % 3 )) + 1 )) * * * /bin/bash $SCRIPT " +
 150                                 ">> $DIR/cronlog 2>&1 \" >> $TFILE\n" +
 151                         "echo \"#### END $CF ####\" >> $TFILE\n" +
 152                         "crontab $TFILE\n" +
 153                         "rm $TFILE\n";
 154 }
 155
 156
 157