MISP/app/Console/Command/AdminShell.php

<?php
App::uses('AppShell', 'Console/Command');
class AdminShell extends AppShell
{
	public $uses = array('Event', 'Post', 'Attribute', 'Job', 'User', 'Task', 'Whitelist', 'Server', 'Organisation', 'AdminSetting', 'Galaxy', 'Taxonomy', 'Warninglist', 'Noticelist', 'ObjectTemplate', 'Bruteforce');

	public function jobGenerateCorrelation() {
		$jobId = $this->args[0];
		$this->loadModel('Job');
		$this->Job->id = $jobId;
		$this->loadModel('Attribute');
		$this->Attribute->generateCorrelation($jobId, 0);
		$this->Job->saveField('progress', 100);
		$this->Job->saveField('message', 'Job done.');
		$this->Job->saveField('status', 4);
	}

	public function jobPurgeCorrelation() {
		$jobId = $this->args[0];
		$this->loadModel('Job');
		$this->Job->id = $jobId;
		$this->loadModel('Attribute');
		$this->Attribute->purgeCorrelations();
		$this->Job->saveField('progress', 100);
		$this->Job->saveField('message', 'Job done.');
		$this->Job->saveField('status', 4);
	}

	public function jobGenerateShadowAttributeCorrelation() {
		$jobId = $this->args[0];
		$this->loadModel('Job');
		$this->Job->id = $jobId;
		$this->loadModel('ShadowAttribute');
		$this->ShadowAttribute->generateCorrelation($jobId);
	}

	public function updateGalaxies() {
		// The following is 7.x upwards only
		//$value = $this->args[0] ?? $this->args[0] ?? 0;
		$value = empty($this->args[0])  ? null : $this->args[0];
		if ($value === 'false') $value = 0;
		if ($value === 'true') $value = 1;
		if ($value === 'force') $value = 1;
		$force = $value;
		$result = $this->Galaxy->update($force);
		if ($result) {
			echo 'Galaxies updated';
		} else {
			echo 'Could not update Galaxies';
		}
	}

	# FIXME: Make Taxonomy->update() return a status string on API if successful
	public function updateTaxonomies() {
		$result = $this->Taxonomy->update();
		if ($result) {
			echo 'Taxonomies updated';
		} else {
			echo 'Could not update Taxonomies';
		}
	}

	public function updateWarningLists() {
		$result = $this->Galaxy->update();
		if ($result) {
			echo 'Warning lists updated';
		} else {
			echo 'Could not update warning lists';
		}
	}

	public function updateNoticeLists() {
		$result = $this->Noticelist->update();
		if ($result) {
			echo 'Notice lists updated';
		} else {
			echo 'Could not update notice lists';
		}
	}

	# FIXME: Debug and make it work, fails to pass userId/orgId properly
	public function updateObjectTemplates() {
		if (empty($this->args[0])) {
			echo 'Usage: ' . APP . '/cake ' . 'Admin updateNoticeLists [user_id]';
		} else {
			$userId = $this->args[0];
			$user = $this->User->find('first', array(
				'recursive' => -1,
				'conditions' => array(
					'User.id' => $userId,
				),
				'fields' => array('User.id', 'User.org_id')
			));
			if (empty($user)) {
				echo 'User not found';
			} else {
				$result = $this->ObjectTemplate->update($user, false,false);
				if ($result) {
					echo 'Object templates updated';
				} else {
					echo 'Could not update object templates';
				}
			}
		}
	}

	public function jobUpgrade24() {
		$jobId = $this->args[0];
		$user_id = $this->args[1];
		$this->loadModel('Job');
		$this->Job->id = $jobId;
		$this->loadModel('Server');
		$this->Server->upgrade2324($user_id, $jobId);
		$this->Job->saveField('progress', 100);
		$this->Job->saveField('message', 'Job done.');
		$this->Job->saveField('status', 4);
	}

	public function prune_update_logs() {
		$jobId = $this->args[0];
		$user_id = $this->args[1];
		$user = $this->User->getAuthUser($user_id);
		$this->loadModel('Job');
		$this->Job->id = $jobId;
		$this->loadModel('Log');
		$this->Log->pruneUpdateLogs($jobId, $user);
		$this->Job->saveField('progress', 100);
		$this->Job->saveField('message', 'Job done.');
		$this->Job->saveField('status', 4);
	}

	public function getSetting() {
		$param = empty($this->args[0]) ? 'all' : $this->args[0];
		$settings = $this->Server->serverSettingsRead();
		$result = $settings;
		if (!empty($param)) {
			$result = 'No valid setting found for ' . $param;
			foreach ($settings as $setting) {
				if ($setting['setting'] == $param) {
					$result = $setting;
					break;
				}
			}
		}
		echo json_encode($result, JSON_PRETTY_PRINT) . PHP_EOL;
	}

	public function setSetting() {
		$setting_name = !isset($this->args[0]) ? null : $this->args[0];
		$value = !isset($this->args[1]) ? null : $this->args[1];
		if ($value === 'false') $value = 0;
		if ($value === 'true') $value = 1;
        $cli_user = array('id' => 0, 'email' => 'SYSTEM', 'Organisation' => array('name' => 'SYSTEM'));
		if (empty($setting_name) || $value === null) {
			echo 'Invalid parameters. Usage: ' . APP . 'Console/cake Admin setSetting [setting_name] [setting_value]';
		} else {
            $setting = $this->Server->getSettingData($setting_name);
            if (empty($setting)) {
                echo 'Invalid setting. Please make sure that the setting that you are attempting to change exists.';
            }
            $result = $this->Server->serverSettingsEditValue($cli_user, $setting, $value);
            if ($result === true) {
                echo 'Setting changed.';
            } else {
                echo $result;
            }
		}
        echo PHP_EOL;
	}

	public function setDatabaseVersion() {
		if (empty($this->args[0])) echo 'Invalid parameters. Usage: ' . APP . 'Console/cake Admin setDatabaseVersion [db_version]' . PHP_EOL;
		else {
			$db_version = $this->AdminSetting->find('first', array(
				'conditions' => array('setting' => 'db_version')
			));
			if (!empty($db_version)) {
				$db_version['value'] = trim($this->args[0]);
				$this->AdminSetting->save($db_version);
				echo 'Database version set. MISP will replay all of the upgrade scripts since the selected version on the next user login.' . PHP_EOL;
			} else {
				echo 'Something went wrong. Could not find the existing db version.' . PHP_EOL;
			}
		}
	}

    public function getAuthkey() {
        if (empty($this->args[0])) {
            echo 'Invalid parameters. Usage: ' . APP . 'Console/cake Admin getAuthkey [user_email]' . PHP_EOL;
        } else {
            $user = $this->User->find('first', array(
                'recursive' => -1,
                'conditions' => array('User.email' => strtolower($this->args[0])),
                'fields' => array('User.authkey')
            ));
            if (empty($user)) {
                echo 'Invalid user.' . PHP_EOL;
            } else {
                echo $user['User']['authkey'] . PHP_EOL;
            }
        }
    }

	public function clearBruteforce()
	{
		$conditions = array('Bruteforce.username !=' => '');
		if (!empty($this->args[0])) {
            $conditions = array('Bruteforce.username' => $this->args[0]);
        }
		$result = $this->Bruteforce->deleteAll($conditions, false, false);
		$target = empty($this->args[0]) ? 'all users' : $this->args[0];
		if ($result) {
			echo 'Brutefoce entries for ' . $target . ' deleted.' . PHP_EOL;
		} else {
			echo 'Something went wrong, could not delete bruteforce entries for ' . $target . '.' . PHP_EOL;
		}
	}
}