mirror of https://github.com/MISP/MISP
432 lines
18 KiB
PHP
432 lines
18 KiB
PHP
<?php
|
|
App::uses('AppController', 'Controller');
|
|
App::uses('Folder', 'Utility');
|
|
App::uses('File', 'Utility');
|
|
|
|
/**
|
|
* Signatures Controller
|
|
*
|
|
* @property Signature $Signature
|
|
*/
|
|
class SignaturesController extends AppController {
|
|
|
|
public $components = array('Security');
|
|
|
|
function beforeFilter() {
|
|
// permit reuse of CSRF tokens on the search page.
|
|
if ('search' == $this->request->params['action']) {
|
|
$this->Security->csrfUseOnce = false;
|
|
}
|
|
|
|
// These variables are required for every view
|
|
$this->set('me', $this->Auth->user());
|
|
$this->set('isAdmin', $this->_isAdmin());
|
|
}
|
|
|
|
|
|
public function isAuthorized($user) {
|
|
// Admins can access everything
|
|
if (parent::isAuthorized($user)) {
|
|
return true;
|
|
}
|
|
// Only on own signatures for these actions
|
|
if (in_array($this->action, array('edit', 'delete'))) {
|
|
$signatureid = $this->request->params['pass'][0];
|
|
return $this->Signature->isOwnedByOrg($signatureid, $this->Auth->user('org'));
|
|
}
|
|
// Only on own events for these actions
|
|
if (in_array($this->action, array('add', 'add_attachment'))) {
|
|
$this->loadModel('Event');
|
|
$eventid = $this->request->params['pass'][0];
|
|
return $this->Event->isOwnedByOrg($eventid, $this->Auth->user('org'));
|
|
}
|
|
// the other pages are allowed by logged in users
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* index method
|
|
*
|
|
* @return void
|
|
*/
|
|
public function index() {
|
|
$this->Signature->recursive = 0;
|
|
$this->set('signatures', $this->paginate());
|
|
}
|
|
|
|
/**
|
|
* add method
|
|
*
|
|
* @return void
|
|
*/
|
|
public function add($event_id = null) {
|
|
if ($this->request->is('post')) {
|
|
$this->loadModel('Event');
|
|
// Replaced by isAuthorized
|
|
// // only own signatures
|
|
// $this->Event->recursive = 0;
|
|
// $event = $this->Event->findById($this->request->data['Signature']['event_id']);
|
|
// if (!$this->_isAdmin() && $this->Auth->user('org') != $event['Event']['org']) {
|
|
// throw new UnauthorizedException('You can only add signatures for your own organisation.');
|
|
// }
|
|
|
|
// Give error if someone tried to submit a signature with attachment or malware-sample type.
|
|
// FIXME this is bad ... it should rather by a messagebox or should be filtered out on the view level
|
|
if('attachment' == $this->request->data['Signature']['type'] ||
|
|
'malware-sample' == $this->request->data['Signature']['type']) {
|
|
$this->Session->setFlash(__('Attribute has not been added: attachments are added by "Add attachment" button', true), 'default', array(), 'error');
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
}
|
|
|
|
|
|
// remove the alerted flag from the event
|
|
$this->Event->id = $this->request->data['Signature']['event_id'];
|
|
$this->Event->saveField('alerted', 0);
|
|
|
|
//
|
|
// multiple signatures in batch import
|
|
//
|
|
if ($this->request->data['Signature']['batch_import'] == 1) {
|
|
// make array from value field
|
|
$signatures = explode("\n", $this->request->data['Signature']['value']);
|
|
|
|
$fails = ""; // will be used to keep a list of the lines that failed or succeeded
|
|
$successes = "";
|
|
foreach ($signatures as $key => $signature) {
|
|
$signature = trim($signature);
|
|
if (strlen($signature) == 0 )
|
|
continue; // don't do anything for empty lines
|
|
|
|
$this->Signature->create();
|
|
$this->request->data['Signature']['value'] = $signature; // set the value as the content of the single line
|
|
$this->request->data['Signature']['uuid'] = String::uuid();
|
|
if ($this->Signature->save($this->request->data)) {
|
|
$successes .= " ".($key+1);
|
|
} else {
|
|
$fails .= " ".($key+1);
|
|
}
|
|
|
|
}
|
|
// we added all the signatures,
|
|
if ($fails) {
|
|
// list the ones that failed
|
|
$this->Session->setFlash(__('The lines'.$fails.' could not be saved. Please, try again.', true), 'default', array(), 'error');
|
|
}
|
|
if ($successes) {
|
|
// list the ones that succeeded
|
|
$this->Session->setFlash(__('The lines'.$successes.' have been saved', true));
|
|
}
|
|
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
|
|
}
|
|
|
|
else {
|
|
//
|
|
// single signature
|
|
//
|
|
// create the signature
|
|
$this->Signature->create();
|
|
$this->request->data['Signature']['uuid'] = String::uuid();
|
|
|
|
if ($this->Signature->save($this->request->data)) {
|
|
// inform the user and redirect
|
|
$this->Session->setFlash(__('The attribute has been saved'));
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
} else {
|
|
$this->Session->setFlash(__('The attribute could not be saved. Please, try again.'));
|
|
}
|
|
}
|
|
} else {
|
|
// set the event_id in the form
|
|
$this->request->data['Signature']['event_id'] = $event_id;
|
|
}
|
|
|
|
// combobox for types
|
|
$types = $this->Signature->validate['type']['rule'][1];
|
|
$types = $this->_arrayToValuesIndexArray($types);
|
|
$this->set('types',compact('types'));
|
|
// combobos for categories
|
|
$categories = $this->Signature->validate['category']['rule'][1];
|
|
$categories = $this->_arrayToValuesIndexArray($categories);
|
|
$this->set('categories',compact('categories'));
|
|
}
|
|
|
|
|
|
public function download($id = null) {
|
|
$this->Signature->id = $id;
|
|
if (!$this->Signature->exists()) {
|
|
throw new NotFoundException(__('Invalid signature'));
|
|
}
|
|
|
|
$this->Signature->read();
|
|
$file = new File(APP.DS."files".DS.$this->Signature->data['Signature']['event_id'].DS.$this->Signature->data['Signature']['id']);
|
|
$filename = '';
|
|
if('attachment' == $this->Signature->data['Signature']['type']) {
|
|
$filename= $this->Signature->data['Signature']['value'];
|
|
} elseif ('malware-sample'== $this->Signature->data['Signature']['type']) {
|
|
$filename_hash = explode('|', $this->Signature->data['Signature']['value']);
|
|
$filename = $filename_hash[0].".zip";
|
|
} else {
|
|
throw new NotFoundException(__('Signature not an attachment or malware-sample'));
|
|
}
|
|
|
|
$file_ext = explode(".", $filename);
|
|
$this->viewClass = 'Media';
|
|
$params = array(
|
|
'id' => $file->path,
|
|
'name' => $filename,
|
|
'download' => true,
|
|
'path' => DS
|
|
);
|
|
$this->set($params);
|
|
}
|
|
|
|
/**
|
|
* add_attachment method
|
|
*
|
|
* @return void
|
|
*/
|
|
public function add_attachment($event_id = null) {
|
|
if ($this->request->is('post')) {
|
|
$this->loadModel('Event');
|
|
// // Replaced by isAuthorized
|
|
// // // only own signatures
|
|
// // $this->Event->recursive = 0;
|
|
// // $event = $this->Event->findById($this->request->data['Signature']['event_id']);
|
|
// // if (!$this->_isAdmin() && $this->Auth->user('org') != $event['Event']['org']) {
|
|
// // throw new UnauthorizedException('You can only add signatures for your own organisation.');
|
|
// // }
|
|
|
|
// Check if there were problems with the file upload
|
|
$filename = Sanitize::HTML($this->request->data['Signature']['value']['name']);
|
|
$tmpfile = new File($this->request->data['Signature']['value']['tmp_name']);
|
|
if ((isset($this->request->data['Signature']['value']['error']) && $this->request->data['Signature']['value']['error'] == 0) ||
|
|
(!empty( $this->request->data['Signature']['value']['tmp_name']) && $this->request->data['Signature']['value']['tmp_name'] != 'none')
|
|
) {
|
|
if(!is_uploaded_file($tmpfile->path))
|
|
throw new InternalErrorException('PHP says file was not uploaded. Are you attacking me?');
|
|
} else {
|
|
$this->Session->setFlash(__('There was a problem to upload the file.', true), 'default', array(), 'error');
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
}
|
|
|
|
// remove the alerted flag from the event
|
|
$this->Event->id = $this->request->data['Signature']['event_id'];
|
|
$this->Event->saveField('alerted', 0);
|
|
|
|
// save the file-info in the database
|
|
$this->Signature->create();
|
|
if($this->request->data['Signature']['malware']) {
|
|
$this->request->data['Signature']['type'] = "malware-sample";
|
|
$this->request->data['Signature']['value'] = $filename.'|'.$tmpfile->md5(); // TODO gives problems with bigger files
|
|
}
|
|
else {
|
|
$this->request->data['Signature']['type'] = "attachment";
|
|
$this->request->data['Signature']['value'] = $filename;
|
|
}
|
|
$this->request->data['Signature']['uuid'] = String::uuid();
|
|
$this->request->data['Signature']['to_ids'] = 0; // LATER permit user to send this to IDS
|
|
$this->request->data['Signature']['batch_import'] = 0;
|
|
|
|
if ($this->Signature->save($this->request->data)) {
|
|
// signature saved correctly in the db
|
|
} else {
|
|
$this->Session->setFlash(__('The attribute could not be saved. Did you already upload this file?'));
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
}
|
|
|
|
// no errors in file upload, entry already in db, now move the file where needed and zip it if required.
|
|
// no sanitization is required on the filename, path or type as we save
|
|
// create directory structure
|
|
$root_dir = APP.DS."files".DS.$this->request->data['Signature']['event_id'];
|
|
$dir = new Folder($root_dir, true);
|
|
// move the file to the correct location
|
|
$destpath = $root_dir.DS.$this->Signature->id; // id of the new signature in the database
|
|
$file = new File ($destpath);
|
|
$zipfile = new File ($destpath.'.zip');
|
|
$file_in_zip = new File($root_dir.DS.$filename); // FIXME do sanitization of the filename
|
|
|
|
if($file->exists() || $zipfile->exists() || $file_in_zip->exists()) {
|
|
// this should never happen as the signature id should be unique
|
|
$this->Session->setFlash(__('Attachment with this name already exist in this event.', true), 'default', array(), 'error');
|
|
// remove the entry from the database
|
|
$this->Signature->delete();
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
}
|
|
if(!move_uploaded_file($tmpfile->path, $file->path)) {
|
|
$this->Session->setFlash(__('Problem with uploading attachment. Cannot move it to its final location.', true), 'default', array(), 'error');
|
|
// remove the entry from the database
|
|
$this->Signature->delete();
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
}
|
|
|
|
// zip and password protect the malware files
|
|
if($this->request->data['Signature']['malware']) {
|
|
// TODO check if CakePHP has no easy/safe wrapper to execute commands
|
|
$exec_retval = ''; $exec_output = array();
|
|
rename($file->path, $file_in_zip->path); // FIXME prevent any attacks
|
|
exec("zip -j -P infected ".$zipfile->path.' "'.addslashes($file_in_zip->path).'"', $exec_output, $exec_retval);
|
|
if($exec_retval != 0) { // not EXIT_SUCCESS
|
|
$this->Session->setFlash(__('Problem with zipping the attachment. Please report to administrator. ', true), 'default', array(), 'error');
|
|
// remove the entry from the database
|
|
$this->Signature->delete();
|
|
$file_in_zip->delete();
|
|
$file->delete();
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
};
|
|
$file_in_zip->delete(); // delete the original not-zipped-file
|
|
rename($zipfile->path, $file->path); // rename the .zip to .nothing
|
|
}
|
|
|
|
// everything is done, now redirect to event view
|
|
$this->Session->setFlash(__('The attachment has been uploaded'));
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $this->request->data['Signature']['event_id']));
|
|
|
|
} else {
|
|
// set the event_id in the form
|
|
$this->request->data['Signature']['event_id'] = $event_id;
|
|
}
|
|
|
|
// combobos for categories
|
|
$categories = $this->Signature->validate['category']['rule'][1];
|
|
$categories = $this->_arrayToValuesIndexArray($categories);
|
|
$this->set('categories',compact('categories'));
|
|
}
|
|
|
|
/**
|
|
* edit method
|
|
*
|
|
* @param string $id
|
|
* @return void
|
|
*/
|
|
public function edit($id = null) {
|
|
$this->Signature->id = $id;
|
|
if (!$this->Signature->exists()) {
|
|
throw new NotFoundException(__('Invalid signature'));
|
|
}
|
|
// Replaced by isAuthorized
|
|
// // only own signatures
|
|
// $this->Signature->read();
|
|
// if (!$this->_isAdmin() && $this->Auth->user('org') != $this->Signature->data['Event']['org']) {
|
|
// throw new UnauthorizedException('You can only edit signatures from your own organisation.');
|
|
// }
|
|
|
|
$this->Signature->read();
|
|
$event_id = $this->Signature->data['Signature']['event_id'];
|
|
if('attachment' == $this->Signature->data['Signature']['type'] ||
|
|
'malware-sample'== $this->Signature->data['Signature']['type'] ) {
|
|
$this->set('attachment', true);
|
|
// FIXME we should ensure value cannot be changed here and not only on a view level (because of the associated file)
|
|
// $this->Session->setFlash(__('You cannot edit attachment attributes.', true), 'default', array(), 'error');
|
|
// $this->redirect(array('controller' => 'events', 'action' => 'view', $old_signature['Event']['id']));
|
|
} else {
|
|
$this->set('attachment', false);
|
|
}
|
|
|
|
debug($this->Signature->data);
|
|
if ($this->request->is('post') || $this->request->is('put')) {
|
|
|
|
if ($this->Signature->save($this->request->data)) {
|
|
// TODO should I check here if it's possible to change the event_id or org ?
|
|
$this->Session->setFlash(__('The attribute has been saved'));
|
|
|
|
debug($this->request->data);
|
|
$this->redirect(array('controller' => 'events', 'action' => 'view', $event_id));
|
|
} else {
|
|
$this->Session->setFlash(__('The attribute could not be saved. Please, try again.'));
|
|
}
|
|
} else {
|
|
$this->request->data = $this->Signature->read(null, $id);
|
|
}
|
|
|
|
// combobox for types
|
|
$types = $this->Signature->validate['type']['rule'][1];
|
|
$types = $this->_arrayToValuesIndexArray($types);
|
|
$this->set('types',compact('types'));
|
|
// combobox for categories
|
|
$categories = $this->Signature->validate['category']['rule'][1];
|
|
$categories = $this->_arrayToValuesIndexArray($categories);
|
|
$this->set('categories',compact('categories'));
|
|
}
|
|
|
|
|
|
/**
|
|
* delete method
|
|
*
|
|
* @param string $id
|
|
* @return void
|
|
*/
|
|
public function delete($id = null) {
|
|
if (!$this->request->is('post')) {
|
|
throw new MethodNotAllowedException();
|
|
}
|
|
$this->Signature->id = $id;
|
|
if (!$this->Signature->exists()) {
|
|
throw new NotFoundException(__('Invalid attribute'));
|
|
}
|
|
// Replaced by isAuthorized
|
|
// // only own signatures
|
|
// $this->Signature->read();
|
|
// if (!$this->_isAdmin() && $this->Auth->user('org') != $this->Signature->data['Event']['org']) {
|
|
// throw new UnauthorizedException('You can only delete signatures from your own organisation.');
|
|
// }
|
|
|
|
// attachment will be deleted with the beforeDelete() function in the Model
|
|
if ($this->Signature->delete()) {
|
|
$this->Session->setFlash(__('Attribute deleted'));
|
|
} else {
|
|
$this->Session->setFlash(__('Attribute was not deleted'));
|
|
}
|
|
|
|
$this->redirect($this->referer());
|
|
}
|
|
|
|
|
|
|
|
public function search() {
|
|
if ($this->request->is('post')) {
|
|
$keyword = $this->request->data['Signature']['keyword'];
|
|
$type = $this->request->data['Signature']['type'];
|
|
$category = $this->request->data['Signature']['category'];
|
|
|
|
// search the db
|
|
$conditions = array();
|
|
if($keyword) {
|
|
$conditions['Signature.value LIKE'] = '%'.$keyword.'%';
|
|
}
|
|
if($type != 'ALL') {
|
|
$conditions['Signature.type ='] = $type;
|
|
}
|
|
if($category != 'ALL') {
|
|
$conditions['Signature.category ='] = $category;
|
|
}
|
|
$this->Signature->recursive = 0;
|
|
$this->paginate = array(
|
|
'conditions' => $conditions
|
|
);
|
|
$this->set('signatures', $this->paginate());
|
|
|
|
// set the same view as the index page
|
|
$this->render('index');
|
|
} else {
|
|
// no search keyword is given, show the search form
|
|
|
|
// adding filtering by category and type
|
|
// combobox for types
|
|
$types = array('ALL');
|
|
$types = array_merge($types, $this->Signature->validate['type']['rule'][1]);
|
|
$types = $this->_arrayToValuesIndexArray($types);
|
|
$this->set('types',compact('types'));
|
|
|
|
// combobox for categories
|
|
$categories = array('ALL');
|
|
$categories = array_merge($categories, $this->Signature->validate['category']['rule'][1]);
|
|
$categories = $this->_arrayToValuesIndexArray($categories);
|
|
$this->set('categories',compact('categories'));
|
|
}
|
|
|
|
}
|
|
|
|
}
|