/**
* Display Action - display/run the firewall
*
* @package las
* @version 1.0
*/
public function displayAction()
{
$params = $this->dispatcher->getParams();
if (isset($params[0]) && ($firewall = Firewalls::findFirst(intval($params[0]) ? $params[0] : ['name=:name:', 'bind' => ['name' => $params[0]]]))) {
echo Las::display($firewall->name);
}
}
/**
* Write method - add/edit the firewall
*
* @package las
* @version 1.0
*
* @param string $method type of method: create/update
* @return mixed
*/
public function write($method = 'create')
{
$validation = new Extension\Validation();
$validation->add('name', new Validator\PresenceOf());
$validation->add('name', new Validator\StringLength(['max' => 32]));
$validation->add('content', new Validator\PresenceOf());
$validation->add('description', new Validator\StringLength(['max' => 1024]));
$validation->add('status', new Validator\InclusionIn(['domain' => Firewalls::status()]));
$validation->setLabels(['name' => __('Name'), 'content' => __('Content'), 'description' => __('Description'), 'status' => __('Status')]);
$messages = $validation->validate($_POST);
// Return messages if validation not pass
if (count($messages)) {
return $validation->getMessages();
} else {
$this->name = $this->request->getPost('name', 'string');
$this->content = $this->request->getPost('content');
$this->description = $this->request->getPost('description', 'string');
$this->status = $this->request->getPost('status', 'int');
$this->date = date('Y-m-d H:i:s');
// Try to write the record
if ($this->{$method}() === true) {
return $this;
} else {
\Las\Bootstrap::log($this->getMessages());
return $this->getMessages();
}
}
}
/**
* Main Action - las crontab
*
* @package las
* @version 1.0
*/
public function mainAction()
{
// Check if there is some firewall to compile
if ($firewall = Firewalls::findFirst('status=' . Firewalls::COMPILE)) {
$firewall->status = Firewalls::COMPILED;
$firewall->update();
$this->dispatcher->forward(['task' => 'firewall', 'action' => 'compile', 'params' => [$firewall->id]]);
}
// Check if there is some firewall to reload
if ($firewall = Firewalls::findFirst('status=' . Firewalls::RELOAD)) {
$firewall->status = Firewalls::COMPILED;
$firewall->update();
$this->dispatcher->forward(['task' => 'firewall', 'action' => 'display', 'params' => [$firewall->id]]);
}
// Check if there is some task to run
if ($task = Tasks::findFirst('status=' . Tasks::RELOAD . ' OR status=' . Tasks::ACTIVE . ' AND when!="@reboot" AND when!="@tmp" AND next< ' . time() . ' ORDER BY next ASC')) {
$task->next = \Las\Library\Crontab::parse($task->when);
$task->status = Tasks::ACTIVE;
$task->update();
switch ($task->type) {
case Tasks::FIREWALL:
$this->dispatcher->forward(['task' => 'firewall', 'action' => 'display', 'params' => [$task->firewall_id]]);
break;
case Tasks::CUTOFF:
$this->dispatcher->forward(['task' => 'cron', 'action' => 'cutoff']);
break;
case Tasks::PAYMENT:
$this->dispatcher->forward(['task' => 'cron', 'action' => 'payment']);
break;
case Tasks::PING:
$this->dispatcher->forward(['task' => 'cron', 'action' => 'ping']);
break;
}
}
}
/**
* Reload action - run the firewall
*
* @package las
* @version 1.0
*/
public function reloadAction()
{
// Get id from url params and check if record exist
$params = $this->router->getParams();
if (isset($params[0]) && ($firewall = Firewalls::findFirst(intval($params[0]) ? $params[0] : ['name=:name:', 'bind' => ['name' => $params[0]]]))) {
$this->tag->setTitle(__('Firewalls') . ' / ' . __('Run'));
$this->view->pick('msg');
// Reload at real time or trigger reload
if ($this->las['general']['realTime'] && $this->las['general']['rootPassword']) {
try {
// Try to run command as root
$output = Las::cmd('php ' . ROOT_PATH . '/private/index.php firewall display ' . $firewall->id . ' | sh', true);
$this->tag->setTitle(__('Success'));
$this->flashSession->success($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Success') . '!</strong> ' . __("The :name firewall was reloaded.", [':name' => '<i>' . $firewall->name . '</i>']));
$this->view->setVar('title', __('Success'));
// Display debug if setting is enabled
if ($this->las['general']['debugCmd']) {
$this->view->setVars(['content' => $output, 'redirect' => false]);
} else {
$this->view->setVars(['redirect' => 'admin/firewalls/display/' . $firewall->id]);
}
} catch (Exception $e) {
$errors = ['error' => get_class($e) . '[' . $e->getCode() . ']: ' . $e->getMessage(), 'info' => $e->getFile() . '[' . $e->getLine() . ']', 'debug' => "Trace: \n" . $e->getTraceAsString() . "\n"];
// Display warning flash and log
$this->tag->setTitle(__('Warning'));
$this->view->setVars(['title' => __('Warning'), 'content' => \Las\Bootstrap::log($errors)]);
$this->flashSession->warning($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Warning') . '!</strong> ' . __("Something is wrong!"));
}
} else {
// Trigger reload
$this->tag->setTitle(__('Notice'));
$this->view->setVars(['title' => __('Notice'), 'redirect' => 'admin/firewalls']);
$this->flashSession->notice($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Notice') . '!</strong> ' . __("The :name firewall was scheduled to reload.", [':name' => '<i>' . $firewall->name . '</i>']));
// Update firewall's status
$firewall->status = Firewalls::RELOAD;
$firewall->update();
}
}
}
/**
* Edit action - edit the task
*
* @package las
* @version 1.0
*/
public function editAction()
{
// Get id from url params and check if record exist
$params = $this->router->getParams();
if (isset($params[0]) && ($task = Tasks::findFirst($params[0]))) {
$firewalls = Firewalls::find();
if (!count($firewalls)) {
$this->flashSession->notice($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Notice') . '!</strong> ' . __("Please add the firewall first") . ': ' . $this->tag->linkTo('admin/firewalls/add', __('Add')));
}
// Set title, pick view and send variables
$this->tag->setTitle(__('Tasks') . ' / ' . __('Edit'));
$this->view->pick('tasks/write');
$this->view->setVars(['type' => Tasks::type(true), 'status' => Tasks::status(true), 'firewalls' => $firewalls]);
// Check if the form has been sent
if ($this->request->isPost() === true && $this->request->hasPost('submit')) {
$task->__set('firewalls', $firewalls);
$valid = $task->write('update');
// Check if data are valid
if ($valid instanceof Tasks) {
$this->flashSession->success($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Success') . '!</strong> ' . __("The data has been saved."));
} else {
$this->view->setVar('errors', $valid);
$this->flashSession->warning($this->tag->linkTo(['#', 'class' => 'close', 'title' => __("Close"), '×']) . '<strong>' . __('Warning') . '!</strong> ' . __("Please correct the errors."));
}
} else {
$diff = ['firewall' => $task->firewall_id, 'next' => $task->next ? date('Y-m-d H:i', $task->next) : null];
// Values to fill out the form
$this->tag->setDefaults(array_merge(get_object_vars($task), $diff));
}
} else {
parent::notFoundAction();
}
}