Filebrowser dev

lib/agent/actions/fileretrieval/files_storage.js

@@ -0,0 +1,69 @@
+var join      = require('path').join,
+    base_path = join(__dirname, '..', '..'),
+    common    = require('./../../common'),
+    hooks     = require('./../../hooks'),
+    triggers  = require('./../../triggers'),
+    storage   = require('./../../utils/storage'),
+    fileretrieval = require(join(base_path, 'actions', 'fileretrieval'));
+
+var logger   = common.logger;
+    watching = false;
+
+var NAMESPACE = "fr/";
+
+var exist = function(id, cb) {
+  var key = NAMESPACE + id;
+  storage.all(function(err, files) {
+    if (err)
+      return logger.error(err.message);
+    if (files[key]) 
+      return cb(true);
+    return cb(false);
+  })
+}
+
+exports.store = function(id, path, size, user) {
+  exist(id, function(cb) {
+    if (cb == false) {
+      logger.debug('Storing file_id in DB: ' + id);
+      var opts = {
+        path: path,
+        size: size,
+        user: user
+      }
+      storage.set(NAMESPACE + id, opts);
+    }
+  });
+}
+
+exports.del = function(id) {
+  logger.debug('Removing file_id from DB: ' + id);
+  storage.del(NAMESPACE + id);
+}
+
+exports.run_stored = function(cb) {
+  storage.all(function(err, files) {
+    if (err)
+      return logger.error(err.message);
+
+    var count = Object.keys(files).length;
+    if (count <= 0)
+      return;
+
+    logger.warn('Re-uploading ' + count + ' pending files.');
+
+    for (key in files) {
+      if(key.indexOf(NAMESPACE) != 0) {
+        continue;
+      }
+      var opts = {
+        file_id: key.substring(3, key.length),
+        path: files[key].path,
+        user: files[key].user,
+        size: files[key].size,
+        resumable: true
+      }
+      fileretrieval.start(opts, cb);
+    }
+  })
+}

lib/agent/actions/fileretrieval/index.js

@@ -0,0 +1,124 @@
+"use strict";
+
+//////////////////////////////////////////
+// Prey JS FileRetrieval
+// (C) 2016 Prey, Inc.
+// by Mauricio Schneider and Javier Acuña - http://preyproject.com
+// GPLv3 Licensed
+//////////////////////////////////////////
+
+var fs = require('fs'),
+    path = require('path'),
+    mime = require('mime'),
+    needle = require('needle'),
+    join = require('path').join,
+    api = require('./../../plugins/control-panel/api'),
+    files = require('./files_storage'),
+    EventEmitter = require('events').EventEmitter;
+
+var common   = require('./../../common'),
+    system   = common.system,
+    config   = common.config,
+    protocol = config._values['control-panel'].protocol,
+    host     = config._values['control-panel'].host,
+    url      = protocol + '://' + host,
+    node_bin = path.join(system.paths.current, 'bin', 'node'),
+    logger   = common.logger,
+    os_name  = process.platform.replace('darwin', 'mac').replace('win32', 'windows');
+
+var em,
+    cp;
+
+var UPLOAD_SERVER = url + '/upload/upload',
+    RESUMABLE_HEADER = 'X-Prey-Upload-Resumable',
+    OPEN_TIMEOUT  = 180000,
+    READ_TIMEOUT  = 5000;
+
+exports.check_pending_files = function() {
+  files.run_stored();
+}
+
+exports.start = function(options, cb) {
+  var file_path = options.path,
+      file_id = options.file_id,
+      file_size = parseInt(options.size),
+      user = options.user;
+
+  em = em || new EventEmitter();
+  if (cb) cb(null, em);
+
+  if (!options.resumable)
+    files.store(file_id, file_path, file_size, user);
+
+  var file = {
+    total: 0,
+    path: file_path,
+    user: user,
+    id: file_id,
+    size: file_size,
+    resumable: options.resumable
+  }
+
+  retrieve_file(file, function() {
+    files.del(file.id);
+  });
+}
+
+function retrieve_file(file, cb) {
+  if (file.resumable) {
+    var url = UPLOAD_SERVER + '?uploadID=' + file.id;
+    // Make a call to get the last byte processed by the upload server
+    // before the disconnection in order to resume the upload from that position
+    needle.request('get', url, null, function(err, res) {
+      if (err) return em.emit(err);
+      var data = JSON.parse(res.body);
+      file.total = data.Total;
+      get_file(file, cb);
+    })
+    return;
+  }
+  get_file(file, cb);
+}
+
+function get_file(file, cb) {
+  var buffsize = (file.size == 0) ? 1 : (file.size - file.total);
+  var buf = new Buffer(buffsize); 
+  var fd = fs.openSync(file.path, "r");
+
+  fs.read(fd, buf, 0, file.size - file.total, file.total, function(err, read, buf) {
+    if (err) return em.emit(err);
+    upload_file(file, buf, cb);
+  })
+}
+
+function upload_file(file, buf, cb) {
+  var options = {
+    open_timeout: OPEN_TIMEOUT,
+    read_timeout: READ_TIMEOUT
+  };
+
+  if (file.total > 0) {
+    RESUMABLE_HEADER = file.total;
+  }
+  var url = UPLOAD_SERVER + '?uploadID='+file.id;
+  needle.post(url, buf, options, function(err, res) {
+    if (err) {
+      logger.error(err)
+      return em.emit(err);
+    }
+    var out = res.statusCode;
+    if (out !== 200 && out !== 201) {
+      var err = new Error('There was an error communicating with the server');
+      logger.error(err)
+      em.emit(err);
+    } 
+    cb();
+  })
+  em.emit('end');
+}
+
+exports.stop = function() {
+  if (cp && !cp.exitCode) {
+    cp.kill();
+  }
+}

lib/agent/actions/wipe/index.js

@@ -78,7 +78,7 @@ exports.start = function(opts, cb) {
       })
 
       child.on('exit', function(code) {
-        if (code != 0) 
+        if (code !== 0)
           last_err = new Error('Wipe command failed.');
 
         finished(last_err);

lib/agent/actions/wipe/runner.js

@@ -28,5 +28,4 @@ process.on('SIGTERM', function() {
 
 process.on('exit', function(code) {
   console.log('Wipe finished. Last error: ' + (last_err || 'none.'));
-  // process.exit(last_err ? 1 : 0);
 })

lib/agent/plugins/control-panel/api/push.js

@@ -2,8 +2,6 @@ var keys    = require('./keys'),
     errors  = require('./errors'),
     request = require('./request');
 
-var format  = '.json';
-
 var check_keys = function() {
   if (!keys.present())
     throw errors.get('MISSING_KEY');
@@ -22,24 +20,29 @@ var post = function(url, data, opts, cb) {
 
 exports.response = function(data, opts, cb) {
   check_keys();
-  var url = '/devices/' + keys.get().device + '/response' + format;
+  var url = format_url('response');
   post(url, data, opts, cb);
 }
 
 exports.event = function(data, opts, cb) {
   check_keys();
-  var url = '/devices/' + keys.get().device + '/events' + format;
+  var url = format_url('events');
   post(url, data, opts, cb);
 }
 
 exports.data = function(data, opts, cb) {
   check_keys();
-  var url = '/devices/' + keys.get().device + '/data' + format;
+  var url = format_url('data');
   post(url, data, opts, cb);
 }
 
 exports.report = function(data, opts, cb) {
   check_keys();
-  var url = '/devices/' + keys.get().device + '/reports' + format;
+  var url = format_url(reports);
   post(url, data, opts, cb);
 }
+
+function format_url(endpoint) {
+  var format  = '.json';
+  return '/devices/' + keys.get().device + '/' + endpoint + format;
+}