switch-ztp/www/add_port_profile.php

<?php
//-
// Copyright (c) 2011-2017 Joe Clarke <jclarke@cisco.com>
// All rights reserved.

// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// 1. Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
// 2. Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.

// THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
// OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
// HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
// OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
// SUCH DAMAGE.

include_once 'db.inc.php';
include_once 'swreg/swreg.inc.php';
require_once 'Log.php';
require_once 'functions.php';

$dsn = "$db_driver:host=$db_host;dbname=$db_name";

$options = [
  PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
  PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
  PDO::ATTR_EMULATE_PREPARES => false,
];

try {
  $dbh = new PDO($dsn, $db_user, $db_pass, $options);
} catch (PDOException $e) {
  die($e->getMessage());
}

$logger = Log::singleton('file', LOGFILE, TOOL_NAME . ' : Add Port Profile');
if ($logger === false) {
  die("Failed to open logfile.\n");
}
$mask = Log::MAX(LOG_LEVEL);
$logger->setMask($mask);

$base = get_base($_SERVER['SCRIPT_NAME']);
$errors = array();

if (isset($_REQUEST['submit'])) {
  $pname = $_REQUEST['pname'];
  $pid = $_REQUEST['pid'];
  $vlans = $_REQUEST['vlan'];
  $start_ports = $_REQUEST['start_port'];
  $end_ports = $_REQUEST['end_port'];

  if (!isset($pname) || !isset($pid) || $pname == '') {
    array_push($errors, 'You must specify a profile name and PID');
  } else {
    $profile = '';
    $seen_sports = array();
    $seen_eports = array();
    $max_port = 0;

    for ($i = 0; $i < count($start_ports); ++$i) {
      $vlan = $vlans[$i];
      $start_port = $start_ports[$i];
      $end_port = $end_ports[$i];
      $row = $i + 1;

      if ($vlan == '__BOGUS__') {
        continue;
      }

      if ($start_port == '__BOGUS__' || $end_port == '__BOGUS__') {
        array_push($errors, "You must select a starting and ending port at row $row");
        continue;
      }

      if ($pid != '__ANY__') {
        preg_match("/\/(\d+)$/", $start_port, $sm);

        if ($sm[1] > $end_port) {
          array_push($errors, "Starting port must be less than equal to the ending port at row $row");
          continue;
        }
      } elseif ($vlan == 'DYNAMIC') {
        array_push($errors, "In order to make ports dynamic, you must choose an explicit PID at row $row");
        continue;
      }

      if (isset($seen_sports[$start_port])) {
        array_push($errors, "Starting port $start_port was already used at row {$seen_sports[$start_port]}");
        continue;
      }

      $seen_sports[$start_port] = $row;

      if (isset($seen_eports[$end_port])) {
        array_push($errors, "Ending port $end_port was already used at row {$seen_eports[$end_port]}");
        continue;
      }

      $seen_eports[$end_port] = $row;

      $profile .= generate_port_profile($start_port, $end_port, $vlan, $pid);
      $profile .= "!\n";
      if ($pid != '__ANY__' && $end_port > $max_port) {
        $max_port = $end_port;
      }
    }

    if ($profile == '') {
      array_push($errors, 'No profile ranges specified');
    }

    if (!preg_match("/^[\w\d\-_]+$/", $pname)) {
      array_push($errors, 'Profile name can only contain letters, numbers, hyphens, and underscores');
    }

    if ($pid != '__ANY__') {
      $ports = 8;
      foreach ($PID_PORTS as $pp => $np) {
        if ($pid == $pp) {
          $ports = $np;
          break;
        }
      }

      if ($max_port != $ports) {
        array_push($errors, 'All ports must be specified for a port profile');
      }
    }
  }

  if (count($errors) == 0) {
    $fd = fopen(PORT_PROFILE_DIR . "/{$pname}.tmpl", 'w');
    fwrite($fd, $profile);
    fclose($fd);

    $sql = 'INSERT INTO EXCEPTIONS VALUES (?, ?, ?)';
    try {
      $sth = $dbh->prepare($sql);
      /*if (isset($PID_MAP[$pid])) {
      $pid = $PID_MAP[$pid];
      }*/
      $sth->execute(array($pname, $pid, "{$pname}.tmpl"));
      $sth->closeCursor();
      $logger->info("User {$_SERVER['REMOTE_USER']} add port profile: name={$pname}, pid={$pid}, path={$pname}.tmpl");
    } catch (PDOException $e) {
      array_push($errors, "Failed to add new exception: {$e->getMessage()}");
    }

    cleanup();
    if (count($errors) == 0) {
      header("Location: $base/port_profiles.php");
      exit;
    }
  }
}

$nports = array_values($PID_PORTS);
rsort($nports, SORT_NUMERIC);
$mport = $nports[0];

print_header(TOOL_NAME . ': Add Port Profile');
?>

<body class="flex" onLoad='purgePorts(); updatePorts("__ANY__", 0, 0, 1, 0); return true;'>
  <script>
    var lastStart = 1;

    var PID_PORTS = [];
    var PORT_PATTERNS = [];
    <?php
    foreach ($PID_PORTS as $pp => $np) {
      ?>
      PID_PORTS["<?= $pp ?>"] = <?= $np ?>;
      <?php

    }

    foreach ($PORT_PATTERNS as $p => $pa) {
      ?>
      PORT_PATTERNS["<?= $p ?>"] = "<?= $pa ?>";
      <?php

    }
    ?>

    function setLastStart(port, increment) {
      var regex = /([0-9]+)$/;
      try {
        var matches = regex.exec(port);
        lastStart = matches[1];
        if (increment) {
          lastStart++;
        }
      } catch (err) { }
    }

    function purgePorts() {
      for (var i = 0; i < <?= $mport ?>; i++) {
        var sSelect = document.getElementById("start_port[" + i + "]");
        var eSelect = document.getElementById("end_port[" + i + "]");
        deleteOptions(sSelect);
        deleteOptions(eSelect);

        createOption(sSelect, '--Please Select--', '__BOGUS__');
        createOption(eSelect, '--Please Select--', '__BOGUS__');
      }
    }

    function updatePorts(pid, sidx, eidx, send, eend) {
      if (typeof send === undefined) {
        send = <?= $mport ?>;
      }
      if (typeof eend === undefined) {
        eend = <?= $mport ?>;
      }
      for (var i = sidx; i < send; i++) {
        var sSelect = document.getElementById("start_port[" + i + "]");
        if (pid == "__ANY__") {
          createOption(sSelect, 'First Switch Port', '%%START_PORT%%');
        } else {
          endp = PID_PORTS[pid] + 1;
          ppat = PORT_PATTERNS[pid];
          for (var j = lastStart; j < endp; j++) {
            createOption(sSelect, ppat + j, ppat + j);
          }
        }

      }

      for (var i = eidx; i < eend; i++) {
        var eSelect = document.getElementById("end_port[" + i + "]");
        if (pid == "__ANY__") {
          createOption(eSelect, 'Last Switch Port', '%%MAX_PORT%%');
        } else {
          endp = PID_PORTS[pid] + 1;
          for (var j = lastStart; j < endp; j++) {
            createOption(eSelect, j, j);
          }
        }
      }
    }

  </script>
  <div id="headswreg">
    <div id="header">
      <h1>
        <?= TOOL_NAME ?>: Add Port Profile
      </h1>
    </div>
    <br />
  </div>
  <form name="port_profile_add_form" method="POST" action="<?= $_SERVER['PHP_SELF'] ?>">
    <div class="apage">
      <table class="noborder" summary="Profile Options">
        <tr>
          <td class="left_act">Profile Name:
            <input type="text" name="pname" size="16">
          </td>
          <td>Select PID:
            <select name="pid" id="pid"
              onChange='purgePorts(); updatePorts(this.options[this.selectedIndex].value, 0, 0, 1, 0); return true;'>
              <option value="__ANY__" selected>Any</option>
              <?php
              foreach ($PID_PORTS as $p => $n) {
                ?>
                <option value="<?= $p ?>">
                  <?= $p ?>
                </option>
                <?php

              }
              ?>
            </select>
          </td>
        </tr>
      </table>
    </div>
    <div id="page">
      <div id="itemhead">
        <h2 class="tabletitle">Port Profile</h2>
      </div>
      <div align="center">
        <?php
        foreach ($errors as $error) {
          ?>
          <p class="error">
            <?= $error ?>
          </p>
          <?php

        }
        ?>
      </div>
      <table class="full" summary="New Port Profile Table">
        <thead>
          <tr>
            <th>Row No.</th>
            <th>Starting Port</th>
            <th>Ending Port</th>
            <th>VLAN</th>
          </tr>
        </thead>
        <tbody>
          <?php
          for ($i = 0; $i < $mport; ++$i) {
            ?>
            <tr <?= get_row_color($i) ?>>
              <td>
                <?= $i + 1 ?>.
              </td>
              <td><select name="start_port[<?= $i ?>]" id="start_port[<?= $i ?>]"
                  onChange='setLastStart(this.options[this.selectedIndex].value, false); pSelect = document.getElementById("pid"); updatePorts(pSelect.options[pSelect.selectedIndex].value, <?= $i + 1 ?>, <?= $i ?>, <?= $i + 1 ?>, <?= $i + 1 ?>); return true;'>
                </select></td>
              <td><select name="end_port[<?= $i ?>]" id="end_port[<?= $i ?>]"
                  onChange='setLastStart(this.options[this.selectedIndex].value, true); pSelect = document.getElementById("pid"); updatePorts(pSelect.options[pSelect.selectedIndex].value, <?= $i + 1 ?>, <?= $i + 1 ?>, <?= $i + 2 ?>, <?= $i + 1 ?>); return true;'>
                </select></td>
              <td><select name="vlan[<?= $i ?>]">
                  <option value="__BOGUS__" selected>--Please Select--</option>
                  <option value="DYNAMIC">DYNAMIC</option>
                  <option value="TRUNK">TRUNK</option>
                  <option value="UPLINK TRUNK">UPLINK TRUNK</option>
                  <option disabled>
                    &#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;&#9472;</option>
                  <?php
                  ksort($VLANS);
                  foreach ($VLANS as $vn => $vi) {
                    ?>
                    <option value="<?= $vn ?>">
                      <?= $vn ?>
                    </option>
                    <?php

                  } ?>
                </select></td>
            </tr>
            <?php

          }
          ?>
        </tbody>
      </table>
      <br />
      <div align="center">
        <input type="reset" name="reset" value="Reset">
        <input type="submit" name="preview" value="Preview"
          onClick='this.form.action="<?= $base ?>/profile_preview.php"; this.form.target="_blank";'>
        <input type="submit" name="submit" value="Add"
          onClick='this.form.action="<?= $_SERVER['PHP_SELF'] ?>"; this.form.target="_self";'>
      </div>
      <table class="noborder" summary="Return Table">
        <tr>
          <td class="left_act">
            <a href="<?= $base ?>/port_profiles.php">&lt;&lt;&lt;Return to Main Page</a>
          </td>
        </tr>
      </table>
    </div>
  </form>
</body>

</html>
<?php
cleanup();
?>