123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973 |
- <?php
- /**
- * @file
- * Functions to handle the NodeSquirrel backup destination.
- */
- define('NODESQUIRREL_SECRET_KEY_PATTERN', '/^[0-9a-f]{32}\:?[0-9a-f]{32}$/');
- /**
- * Check that a nodesquirel key is valid.
- */
- function backup_migrate_nodesquirrel_check_secret_key($secret_key) {
- if ($destination = backup_migrate_nodesquirrel_get_destination($secret_key)) {
- if ($destination->confirm_destination()) {
- return $destination;
- }
- }
- return FALSE;
- }
- /**
- * Get the NS destination for the given key.
- */
- function backup_migrate_nodesquirrel_get_destination($secret_key) {
- if ($secret_key) {
- backup_migrate_include('destinations');
- $destination = backup_migrate_create_destination('nodesquirrel', array('machine_name' => 'nodesquirrel'));
- $destination->settings['secret_key'] = $secret_key;
- return $destination;
- }
- return NULL;
- }
- /**
- * Get a helper link to activate a site and create a tree.
- */
- function backup_migrate_nodesquirrel_get_activate_help_text() {
- $activate_link = backup_migrate_nodesquirrel_get_activate_link();
- return array(
- '#type' => 'item',
- '#title' => t('Need a Secret Key?'),
- '#markup' => t('Visit !nodesquirrel.', array('!nodesquirrel' => $activate_link)),
- '#description' => t('Don\'t worry if you don\'t have an account yet. You can create one when you get there.'),
- );
- }
- /**
- * Get a helper link to activate a site and create a tree.
- */
- function backup_migrate_nodesquirrel_get_activate_link() {
- $activate_link = l('nodesquirrel.com/activate', variable_get('nodesquirrel_activate_url', 'http://manage.nodesquirrel.com/activate'), array('query' => array('url' => url('', array('absolute' => TRUE)), 'email' => variable_get('site_mail', ''), 'configure' => url($_GET['q'], array('absolute' => TRUE)))));
- return $activate_link;
- }
- /**
- * Get a helper link to activate a site and create a tree.
- */
- function backup_migrate_nodesquirrel_get_manage_link($destination) {
- $url = variable_get('nodesquirrel_manage_url', 'https://manage.nodesquirrel.com') . '/backups/' . $destination->_get_destination();
- return l($url, $url);
- }
- /**
- * Get a helper link to activate a site and create a tree.
- */
- function backup_migrate_nodesquirrel_get_plan_link() {
- $url = variable_get('nodesquirrel_manage_url', 'https://manage.nodesquirrel.com') . '/plan';
- return l($url, $url);
- }
- /**
- * Get the path to the nodesquirrel settings tab.
- */
- function backup_migrate_nodesquirrel_settings_path() {
- $path = BACKUP_MIGRATE_MENU_PATH . '/settings/destinations/nodesquirrel/edit';
- return $path;
- }
- /**
- * NodeSquirrel settings page callback.
- */
- function backup_migrate_nodesquirrel_settings() {
- return drupal_get_form('backup_migrate_nodesquirrel_settings_form');
- }
- /**
- * NodeSquirrel settings form.
- */
- function backup_migrate_nodesquirrel_credentials_settings_form($key = '', $status) {
- $collapse = !empty($status);
- $form['nodesquirrel_credentials'] = array(
- '#type' => 'fieldset',
- '#title' => t('NodeSquirrel Credentials'),
- '#collapsible' => $collapse,
- '#collapsed' => $collapse,
- );
- $form['nodesquirrel_credentials']['nodesquirrel_secret_key'] = array(
- '#type' => 'textfield',
- '#title' => t('Secret Key'),
- '#size' => 80,
- '#default_value' => $key,
- );
- if (empty($key)) {
- $form['nodesquirrel_credentials']['secret_key_help'] = backup_migrate_nodesquirrel_get_activate_help_text();
- }
- return $form;
- }
- /**
- * Return a form element with some help text describing NodeSquirrel.
- */
- function backup_migrate_nodesquirrel_info_form() {
- $form = array();
- $form['nodesquirrel_info'] = array(
- '#type' => 'fieldset',
- '#title' => t('What is NodeSquirrel?'),
- );
- $form['nodesquirrel_info']['intro'] = array(
- '#type' => 'markup',
- '#markup' => t('<p>NodeSquirrel is the cloud backup service for Drupal built by the maintainers of Backup and Migrate. It is secure, reliable and affordable.</p><p>NodeSquirrel is a paid service and profits help support Backup and Migrate.</p><p>Find out more at !nodesquirrel</p>', array('!nodesquirrel' => l('nodesquirrel.com', 'http://www.nodesquirrel.com'), '!add' => l(t('add other offsite destinations'), BACKUP_MIGRATE_MENU_PATH . '/destination/list/add'), '!bam' => l(t('Backup and Migrate project page'), 'http://drupal.org/project/backup_migrate'))),
- );
- return $form;
- }
- /**
- * Display the NodeSquirrel status on the configuration form.
- */
- function backup_migrate_nodesquirrel_status_form($key, $destination, $status) {
- $form = array();
- $form['nodesquirrel_status'] = array(
- '#type' => 'fieldset',
- '#title' => t('NodeSquirrel Status'),
- );
- $form['nodesquirrel_status']['status'] = array(
- '#type' => 'item',
- '#title' => t('NodeSquirrel Status'),
- '#markup' => t('Not Configured. Enter your Secret Key below to get started.'),
- );
- // Warn the user if the key they entered is invalid.
- if ($key && empty($status)) {
- $form['nodesquirrel_status']['status']['#markup'] = t('Your secret key does not seem to be valid. Please check that you entered it correctly or visit !ns to generate a new key.', array('!ns' => backup_migrate_nodesquirrel_get_activate_link()));
- }
- else if (!empty($destination) && is_array($status)) {
- if (!empty($status['lifetime_backups_used']) && !empty($status['lifetime_backups']) && $status['lifetime_backups_used'] >= $status['lifetime_backups']) {
- $form['nodesquirrel_status']['status']['#markup'] = t('Your !num backup trial has expired. Visit !link to continue backing up.', array('!num' => $status['lifetime_backups'], '!link' => backup_migrate_nodesquirrel_get_plan_link()));
- }
- else {
- $form['nodesquirrel_status']['status']['#markup'] = t('Ready to Backup');
- if (user_access('perform backup')) {
- $form['nodesquirrel_status']['status']['#markup'] .= ' ' . l('(' . t('backup now') . ')', BACKUP_MIGRATE_MENU_PATH);
- }
- }
- if (!empty($status['plan_name'])) {
- $form['nodesquirrel_status']['plan_name'] = array(
- '#type' => 'item',
- '#title' => t('Current Plan'),
- '#markup' => check_plain($status['plan_name'])
- );
- if (isset($status['plan_id']) && strpos($status['plan_id'], 'trial') !== FALSE) {
- if (isset($status['lifetime_backups']) && isset($status['lifetime_backups_used'])) {
- $remains = $status['lifetime_backups'] - $status['lifetime_backups_used'];
- $remains = $remains > 0 ? $remains : t('none');
- $form['nodesquirrel_status']['plan_name']['#markup'] .= ' ' . t('(@remains remaining of @backups backup trial)', array('@backups' => $status['lifetime_backups'], '@remains' => $remains));
- }
- if (isset($status['lifespan']) && isset($status['age']) && $status['lifespan'] > 0) {
- $remains = ceil(($status['lifespan'] - $status['age']) / 86400);
- if ($remains <= 0) {
- $form['nodesquirrel_status']['plan_name']['#markup'] .= ' ' . t('(Your !span day trial has expired.)', array('!span' => ceil($status['lifespan'] / 86400)));
- }
- else {
- $form['nodesquirrel_status']['plan_name']['#markup'] .= ' ' . format_plural($remains, '(1 day remaining)', '(!span days remaining)', array('!span' => ceil($remains)));
- }
- }
- }
- }
- if (isset($status['backups_used'])) {
- $form['nodesquirrel_status']['backups_used'] = array(
- '#type' => 'item',
- '#title' => t('Number of Stored Backups'),
- '#markup' => $status['backups_used'] == 0 ? t('None') : number_format($status['backups_used'])
- );
- }
- if (isset($status['last_backup'])) {
- $form['nodesquirrel_status']['last_backup'] = array(
- '#type' => 'item',
- '#title' => t('Last Backup'),
- '#markup' => empty($status['last_backup']) ? t('Never') : t('!date (!ago ago)', array('!date' => format_date($status['last_backup'], 'small'), '!ago' => format_interval(time() - $status['last_backup'], 1)))
- );
- }
- if ($status['bytes_per_locker']) {
- if (isset($status['bytes_used'])) {
- $form['nodesquirrel_status']['space'] = array(
- '#type' => 'item',
- '#title' => t('Storage Space'),
- '#markup' => t('!used used of !total (!remaining remaining)', array('!used' => backup_migrate_format_size($status['bytes_used']), '!total' => backup_migrate_format_size($status['bytes_per_locker']), '!remaining' => backup_migrate_format_size(max(0, $status['bytes_per_locker'] - $status['bytes_used']))))
- );
- }
- else {
- $form['nodesquirrel_status']['space'] = array(
- '#type' => 'item',
- '#title' => t('Total Storage Space'),
- '#markup' => t('!total', array('!total' => backup_migrate_format_size($status['bytes_per_locker'])))
- );
- }
- }
- $form['nodesquirrel_status']['manage'] = array(
- '#type' => 'item',
- '#title' => t('Management Console'),
- '#markup' => backup_migrate_nodesquirrel_get_manage_link($destination),
- '#description' => t('You can use the NodeSquirrel management console to add and edit your sites, reset your secret key, download and delete backups, and modify your NodeSquirrel account.'),
- );
- }
- return $form;
- }
- function backup_migrate_nodesquirrel_schedule_settings_form($destination, $status) {
- backup_migrate_include('sources', 'schedules', 'profiles');
- // If the schedule has been overriden it must be edited in the schedule tab.
- $schedule = backup_migrate_crud_get_item('schedule', 'nodesquirrel');
- $default = 60*60*24;
- $form = array();
- $form['nodesquirrel_schedule'] = array(
- '#type' => 'fieldset',
- '#title' => t('Backup Schedule'),
- '#description' => t('Set up a schedule to back up your site to NodeSquirrel. You can customize this schedule or add additional schedules in the !schedule.', array('!schedule' => l(t('Schedules tab'), BACKUP_MIGRATE_MENU_PATH . '/schedule'), '!cron' => l(t('cron'), 'http://drupal.org/cron'))),
- );
- $key = 'nodesquirrel_schedule';
- $form['nodesquirrel_schedule'][$key] = array();
- $defaults = array(
- 'period' => empty($schedule) ? variable_get('nodesquirrel_schedule', 60*60*24) : $schedule->get('period'),
- 'enabled' => empty($schedule) ? variable_get('nodesquirrel_schedule_enabled', TRUE) : $schedule->get('enabled'),
- 'source_id' => empty($schedule) ? variable_get('nodesquirrel_schedule_source_id', 'db') : $schedule->get('source_id'),
- );
- $form['nodesquirrel_schedule'][$key]['nodesquirrel_schedule_enabled'] = array(
- '#type' => 'checkbox',
- '#title' => t('Automatically backup to NodeSquirrel'),
- '#default_value' => $defaults['enabled'],
- );
- $form['nodesquirrel_schedule'][$key]['settings'] = array(
- '#type' => 'backup_migrate_dependent',
- '#dependencies' => array(
- 'nodesquirrel_schedule_enabled' => TRUE,
- ),
- );
- $form['nodesquirrel_schedule'][$key]['settings']['nodesquirrel_schedule_source_id'] = _backup_migrate_get_source_pulldown($defaults['source_id']);
- $options = array(
- (60*60) => t('Once an hour'),
- (60*60*24) => t('Once a day'),
- (60*60*24*7) => t('Once a week'),
- );
- $period = $defaults['period'];
- if (!isset($options[$period])) {
- $options[$period] = empty($schedule) ? t('Custom') : $schedule->get('frequency_description');
- }
- $form['nodesquirrel_schedule'][$key]['settings']['nodesquirrel_schedule'] = array(
- '#type' => 'select',
- '#title' => t('Schedule Frequency'),
- '#options' => $options,
- '#default_value' => $period,
- );
- return $form;
- }
- /**
- * NodeSquirrel settings form.
- */
- function backup_migrate_nodesquirrel_settings_form($form_state) {
- _backup_migrate_message_callback('_backup_migrate_message_browser');
- $form = array();
- $key = variable_get('nodesquirrel_secret_key', '');
- $status = array();
- if ($destination = backup_migrate_nodesquirrel_get_destination($key)) {
- $status = $destination->check_limits();
- }
- $form += backup_migrate_nodesquirrel_info_form();
- $form += backup_migrate_nodesquirrel_status_form($key, $destination, $status);
- $form += backup_migrate_nodesquirrel_credentials_settings_form($key, $status);
- $form += backup_migrate_nodesquirrel_schedule_settings_form($destination, $status);
- return system_settings_form($form);
- }
- /**
- * A destination for sending database backups to the NodeSquirel backup service.
- *
- * @ingroup backup_migrate_destinations
- */
- class backup_migrate_destination_nodesquirrel extends backup_migrate_destination {
- var $supported_ops = array('scheduled backup', 'manual backup', 'remote backup', 'restore', 'list files', 'configure', 'delete');
- var $cache_files = TRUE;
- // Don't generate a metadata file as NodeSquirrel can save metadata natively.
- var $save_metadata = FALSE;
- /**
- * Get the destination name. Provide a default.
- */
- function get_name() {
- if (empty($this->name)) {
- return t('NodeSquirrel');
- }
- return $this->name;
- }
- /**
- * Get the menu items for manipulating this type.
- */
- function get_menu_items() {
- $items = array();
- $path = BACKUP_MIGRATE_MENU_PATH . '/nodesquirrel';
- $items[$path] = array(
- 'title' => t('NodeSquirrel'),
- 'page callback' => 'backup_migrate_menu_callback',
- 'page arguments' => array('destinations.nodesquirrel', 'backup_migrate_nodesquirrel_settings'),
- 'access arguments' => array('administer backup and migrate'),
- 'weight' => 10,
- 'type' => MENU_LOCAL_TASK,
- );
- return $items;
- }
- /**
- * Declare any mysql databases defined in the settings.php file as a possible destination.
- */
- function destinations() {
- $out = array();
- $out['nodesquirrel'] = backup_migrate_create_destination('nodesquirrel', array('machine_name' => 'nodesquirrel'));
- if ($secret_key = variable_get('nodesquirrel_secret_key', '')) {
- $out['nodesquirrel']->settings['secret_key'] = $secret_key;
- }
- else {
- $out['nodesquirrel']->supported_ops = array('manual backup', 'remote backup');
- }
- return $out;
- }
- /**
- * Returns a form to be completed before saving to the destination can be complete.
- */
- function before_backup_form($settings) {
- $form = array();
- $key = $this->settings('secret_key');
- if (empty($key)) {
- $form += backup_migrate_nodesquirrel_info_form();
- $form += backup_migrate_nodesquirrel_credentials_settings_form();
- }
- return $form;
- }
- /**
- * Validate the before_backup_form form.
- */
- function before_backup_form_validate($settings, $form, $form_state) {
- if (isset($form_state['values']['nodesquirrel_secret_key'])) {
- $key = trim($form_state['values']['nodesquirrel_secret_key']);
- if ($error = $this->vaidate_key($key)) {
- form_set_error('secret_key', $error);
- }
- }
- }
- /**
- * Submit the before_backup_form form.
- */
- function before_backup_form_submit($settings, $form, $form_state) {
- if (isset($form_state['values']['nodesquirrel_secret_key'])) {
- $this->settings['secret_key'] = $form_state['values']['nodesquirrel_secret_key'];
- variable_set('nodesquirrel_secret_key', $this->settings['secret_key']);
- }
- }
- /**
- * Save to the NodeSquirrel destination.
- */
- function save_file($file, $settings) {
- if ($destination = $this->_get_destination()) {
- srand((double)microtime()*1000000);
- $filename = $file->filename();
- $filesize = filesize($file->filepath());
- $ticket = $this->_xmlrpc('backups.getUploadTicket', array($destination, $filename, $filesize, $file->file_info));
- if ($ticket) {
-
- $url = $ticket['url'];
-
- // If the ticket requires authentication add our username/password to the url.
- if (!empty($ticket['auth']) && $ticket['auth'] = 'basic') {
- $parts = parse_url($ticket['url']);
- list($parts['user'], $parts['pass']) = $this->get_user_pass();
- $url = $this->glue_url($parts, FALSE);
- }
- $out = $this->_post_file($url, 'POST', $ticket['params'], $file);
- if ($out->code == 200) {
- // Confirm the upload.
- $confirm = $this->_xmlrpc('backups.confirmUpload', array($destination, $filename, $filesize));
- if ($confirm['success']) {
- // Set a message with a link to the manage console.
- $url = variable_get('nodesquirrel_manage_url', 'http://manage.nodesquirrel.com') . '/backups/' . $this->_get_destination();
- _backup_migrate_message('Your backup has been saved to your NodeSquirrel account. View it at !account', array('!account' => l($url, $url)));
- return $file;
- }
- else {
- _backup_migrate_message('The backup file never made it to the NodeSquirrel backup server. There may have been a network problem. Please try again later');
- }
- }
- else {
- $error = !empty($out->headers['x-bams-error']) ? $out->headers['x-bams-error'] : $out->error;
- _backup_migrate_message('The NodeSquirrel server returned the following error: %err', array('%err' => $error), 'error');
- }
- }
- else if ($err = xmlrpc_error()) {
- // XMLRPC errors are already handled by the server function below.
- }
- else {
- _backup_migrate_message('The NodeSquirrel server refused the backup but did not specify why. Maybe the server is down.');
- }
- }
- return NULL;
- }
- /**
- * Load from the NodeSquirrel destination.
- */
- function load_file($file_id) {
- if ($destination = $this->_get_destination()) {
- backup_migrate_include('files');
- $file = new backup_file(array('filename' => $file_id));
- $ticket = $this->_xmlrpc('backups.getDownloadTicket', array($destination, $file_id));
- if ($ticket && $url = $ticket['url']) {
- // If the ticket requires authentication add our username/password to the url.
- if (!empty($ticket['auth']) && $ticket['auth'] = 'basic') {
- $parts = parse_url($ticket['url']);
- $parts['user'] = @$this->dest_url['user'];
- $parts['pass'] = @$this->dest_url['pass'];
- $url = $this->glue_url($parts, FALSE);
- }
-
- $out = drupal_http_request($url);
- if ($out->code == 200) {
- file_put_contents($file->filepath(), $out->data);
- return $file;
- }
- else {
- $error = !empty($out->headers['x-bams-error']) ? $out->headers['x-bams-error'] : $out->error;
- _backup_migrate_message('The server returned the following error: %err', array('%err' => $error), 'error');
- }
- }
- }
- return NULL;
- }
- /**
- * Delete from the NodeSquirrel destination.
- */
- function delete_file($file_id) {
- if ($destination = $this->_get_destination()) {
- $result = $this->_xmlrpc('backups.deleteFile', array($destination, $file_id));
- }
- }
- /**
- * List the files in the remote destination.
- */
- function _list_files() {
- $files = array();
- backup_migrate_include('files');
- if ($destination = $this->_get_destination()) {
- $file_list = $this->_xmlrpc('backups.listFiles', array($destination));
- foreach ((array)$file_list as $file) {
- $files[$file['filename']] = new backup_file($file);
- }
- }
- return $files;
- }
- /**
- * List the files in the remote destination.
- */
- function check_limits() {
- if (empty($this->limits)) {
- $this->limits = $this->_xmlrpc('backups.getLimits', array($this->_get_destination()));
- }
- return $this->limits;
- }
- /**
- * Check that a destination is valid.
- */
- function confirm_destination() {
- return $this->check_limits();
- }
- /**
- * Get the form for the settings for this destination.
- */
- function edit_form() {
- $form = parent::edit_form();
- // If this is a new destination but the default NS destination has not been created yet,
- // redirect to the NS config screen.
- if (!$this->get_id() && !variable_get('nodesquirrel_secret_key', '')) {
- drupal_goto(BACKUP_MIGRATE_MENU_PATH . '/nodesquirrel');
- }
- $form['settings'] = array('#tree' => TRUE);
- $activate_link = backup_migrate_nodesquirrel_get_activate_link();
- // Retrieve the key from the settings or get it from the get string if this is an auto-config action.
- $key = $this->settings('secret_key');
- if (!empty($_GET['key']) && preg_match(NODESQUIRREL_SECRET_KEY_PATTERN, $_GET['key'])) {
- $key = $_GET['key'];
- }
- $form['settings']['secret_key'] = array(
- '#type' => 'textfield',
- '#title' => t('Secret Key'),
- '#default_value' => $key,
- );
- $form['settings']['location'] = array('#type' => 'value', '#value' => '');
- $form['settings']['secret_key_help'] = array(
- '#type' => 'item',
- '#title' => t('Need a Secret Key?'),
- '#markup' => t('Visit !nodesquirrel.', array('!nodesquirrel' => $activate_link)),
- );
- return $form;
- }
- /**
- * Submit the configuration form. Glue the url together and add the old password back if a new one was not specified.
- */
- function edit_form_validate($form, &$form_state) {
- $key = trim($form_state['values']['settings']['secret_key']);
- if ($error = $this->vaidate_key($key)) {
- form_set_error('secret_key', $error);
- }
- }
- /**
- * Validate a secret key. Returns error text if the key is invalid.
- */
- function vaidate_key($key) {
- $error = FALSE;
- if ($key) {
- if (!preg_match(NODESQUIRREL_SECRET_KEY_PATTERN, $key)) {
- return 'The secret key you entered is not the right format. Please make sure you copy it exactly.';
- }
- $this->settings['secret_key'] = check_plain($key);
- $limits = $this->check_limits();
- if (!$limits) {
- $err = xmlrpc_error();
- if (!empty($err->code) && $err->code == '401') {
- return 'Could not log in to the NodeSquirrel server. Please check that your secret key is correct.';
- }
- else {
- return 'Your site could not be found on NodeSquirrel. Please check that your secret key is correct.';
- }
- }
- }
- else {
- return 'You must enter a NodeSquirrel secret key.';
- }
- return FALSE;
- }
- /**
- * Submit the configuration form. Glue the url together and add the old password back if a new one was not specified.
- */
- function edit_form_submit($form, &$form_state) {
- $form_state['values']['secret_key'] = check_plain($form_state['values']['settings']['secret_key']);
- parent::edit_form_submit($form, $form_state);
- }
- /**
- * Get the destination id or warn the user that it has not been set.
- */
- function _get_destination($warn = TRUE) {
- list($id, $key) = $this->get_user_pass();
- return $id;
- }
- /**
- * Get the destination id or warn the user that it has not been set.
- */
- function _get_private_key($warn = TRUE) {
- list($id, $key) = $this->get_user_pass();
- return $key;
- }
- /**
- * Break the secret key into the public/private key (user/pass).
- */
- function get_user_pass() {
- $key = $this->settings('secret_key');
- // The username is the first 32 chars.
- $user = substr($key, 0, 32);
- // The pass is the last 32 chars. There may be a separating character.
- $pass = substr($key, strlen($key) - 32);
- return array($user, $pass);
- }
- function get_display_location() {
- return t('NodeSquirrel.com');
- }
- function add_scheme($url) {
- return 'http://' . $url;
- }
- /**
- * Get the form for the settings for this destination.
- */
- function _xmlrpc($method, $args = array()) {
- // Retrieve the severs or read them from a stored variable.
- $servers = $this->_get_endpoints();
- // Do the actual call.
- return $this->__xmlrpc($method, $args, $servers);
- }
- /**
- * Get the form for the settings for this destination.
- */
- function __xmlrpc($method, $args, $servers, $retry = 3) {
- if ($servers && --$retry > 0) {
- // Add the key authentication arguments if we can.
- if ($this->_sign_request($args)) {
- $url = reset($servers);
- // Try each available server in order.
- while ($url) {
-
- $url = $this->add_scheme($url);
- $out = xmlrpc($url, array($method => $args));
-
- // Check for errors.
- $err = xmlrpc_error();
- if ($err && $err->is_error) {
- switch ($err->code) {
- case '500':
- case '503':
- case '404':
- // Some sort of server error. Try the next one.
- $url = next($servers);
- // If we're at the end of the line then try refetching the urls
- if (!$url) {
- $servers = $this->_get_endpoints(TRUE, $retry);
- return $this->__xmlrpc($method, $args, $servers, $retry);
- }
- break;
- case '300':
- // 'Multiple Choices' means that the existing server list needs to be refreshed.
- $servers = $this->_get_endpoints(TRUE, $retry);
- return $this->__xmlrpc($method, $args, $servers, $retry);
- break;
- case '401':
- case '403':
- // Authentication failed.
- _backup_migrate_message('Couldn\'t log in to NodeSquirrel. The server error was: %err', array('%err' => $err->message), 'error');
- return FALSE;
- break;
- default:
- // Some sort of client error. Don't try the next server because it'll probably say the same thing.
- _backup_migrate_message('The NodeSquirrel server returned the following error: %err', array('%err' => $err->message), 'error');
- return FALSE;
- break;
- }
- }
- // No error, return the result.
- else {
- return $out;
- }
- }
- }
- }
- }
- /**
- * Genrate a hash with a given secret key, timestamp and random value.
- */
- function _get_hash($time, $nonce) {
- if ($private_key = $this->_get_private_key()) {
- $message = $time . ':' . $nonce . ':' . $private_key;
- // Use HMAC-SHA1 to authenticate the call.
- $hash = base64_encode(
- pack('H*', sha1((str_pad($private_key, 64, chr(0x00)) ^ (str_repeat(chr(0x5c), 64))) .
- pack('H*', sha1((str_pad($private_key, 64, chr(0x00)) ^ (str_repeat(chr(0x36), 64))) .
- $message))))
- );
- return $hash;
- }
- _backup_migrate_message('You must enter a valid secret key to use NodeSquirrel.', array(), 'error');
- return FALSE;
- }
- /**
- * Genrate a hash with a given secret key, timestamp and random value.
- */
- function _sign_request(&$args) {
- $nonce = md5(mt_rand());
- $time = time();
- $hash = $this->_get_hash($time, $nonce);
- if ($hash) {
- array_unshift($args, $nonce);
- array_unshift($args, $time);
- array_unshift($args, $hash);
- return TRUE;
- }
- else {
- return FALSE;
- }
- }
- /**
- * Retrieve the list of servers.
- */
- function _get_endpoints($refresh = FALSE, $retry = 3) {
- $servers = (array)variable_get('nodesquirrel_endpoint_urls', array());
- // No servers saved or a force refreshr required.
- if ($refresh || empty($servers)) {
- $servers = array_unique(array_merge($servers, variable_get('nodesquirrel_default_endpoint_urls', array('api.nodesquirrel.com/services/xmlrpc'))));
- // Call the get endpoints method but use the default or previous servers to avoid infinite loops.
- $new_servers = $this->__xmlrpc('backups.getEndpoints', array($this->_get_destination(), 'xmlrpc'), $servers, $retry);
- if ($new_servers) {
- variable_set('nodesquirrel_endpoint_urls', $new_servers);
- $servers = $new_servers;
- }
- }
- return $servers;
- }
- /**
- * Post a file via http.
- *
- * This looks a lot like a clone of drupal_http_request but it can post a large
- * file without reading the whole file into memory.
- */
- function _post_file($url, $method = 'GET', $params = array(), $file = NULL, $retry = 3) {
- global $db_prefix;
-
- $result = new stdClass();
-
- // Parse the URL and make sure we can handle the schema.
- $uri = parse_url($url);
-
- if ($uri == FALSE) {
- $result->error = 'unable to parse URL';
- $result->code = -1001;
- return $result;
- }
-
- if (!isset($uri['scheme'])) {
- $result->error = 'missing schema';
- $result->code = -1002;
- return $result;
- }
-
- switch ($uri['scheme']) {
- case 'http':
- case 'feed':
- $port = isset($uri['port']) ? $uri['port'] : 80;
- $host = $uri['host'] . ($port != 80 ? ':'. $port : '');
- $fp = @fsockopen($uri['host'], $port, $errno, $errstr, 15);
- break;
- case 'https':
- // Note: Only works for PHP 4.3 compiled with OpenSSL.
- $port = isset($uri['port']) ? $uri['port'] : 443;
- $host = $uri['host'] . ($port != 443 ? ':'. $port : '');
- $fp = @fsockopen('ssl://'. $uri['host'], $port, $errno, $errstr, 20);
- break;
- default:
- $result->error = 'invalid schema '. $uri['scheme'];
- $result->code = -1003;
- return $result;
- }
-
- // Make sure the socket opened properly.
- if (!$fp) {
- // When a network error occurs, we use a negative number so it does not
- // clash with the HTTP status codes.
- $result->code = -$errno;
- $result->error = trim($errstr);
-
- // Mark that this request failed. This will trigger a check of the web
- // server's ability to make outgoing HTTP requests the next time that
- // requirements checking is performed.
- // @see system_requirements()
- variable_set('drupal_http_request_fails', TRUE);
-
- return $result;
- }
-
- // Construct the path to act on.
- $path = isset($uri['path']) ? $uri['path'] : '/';
- if (isset($uri['query'])) {
- $path .= '?'. $uri['query'];
- }
- // Prepare the data payload.
- $boundary = '---------------------------'. substr(md5(rand(0,32000)),0,10);
- $data_footer = "\r\n--$boundary--\r\n";
- $data_header = '';
- foreach ($params as $key => $value) {
- $data_header .="--$boundary\r\n";
- $data_header .= "Content-Disposition: form-data; name=\"".$key."\"\r\n";
- $data_header .= "\r\n".$value."\r\n";
- $data_header .="--$boundary\r\n";
- }
-
- // Add the file header to the post payload.
- $data_header .="--$boundary\r\n";
- $data_header .= "Content-Disposition: form-data; name=\"file\"; filename=\"". $file->filename() ."\"\r\n";
- $data_header .= "Content-Type: application/octet-stream;\r\n";
- $data_header .= "\r\n";
- // Calculate the content length.
- $content_length = strlen($data_header . $data_footer) + filesize($file->filepath());
- //file_get_contents($file->filepath()));
-
- // Create HTTP request.
- $defaults = array(
- // RFC 2616: "non-standard ports MUST, default ports MAY be included".
- // We don't add the port to prevent from breaking rewrite rules checking the
- // host that do not take into account the port number.
- 'Host' => "Host: $host",
- 'Content-type' => "Content-type: multipart/form-data, boundary=$boundary",
- 'User-Agent' => 'User-Agent: NodeSquirrel Client/1.x (+http://www.nodesquirrel.com) (Drupal '. VERSION .'; Backup and Migrate 2.x)',
- 'Content-Length' => 'Content-Length: '. $content_length
- );
- // If the server url has a user then attempt to use basic authentication
- if (isset($uri['user'])) {
- $defaults['Authorization'] = 'Authorization: Basic '. base64_encode($uri['user'] . (!empty($uri['pass']) ? ":". $uri['pass'] : ''));
- }
- $request = $method .' '. $path ." HTTP/1.0\r\n";
- $request .= implode("\r\n", $defaults);
- $request .= "\r\n\r\n";
- $result->request = $request;
- // Write the headers and start of the headers
- fwrite($fp, $request);
- fwrite($fp, $data_header);
- // Copy the file 512k at a time to prevent memory issues.
- if ($fp_in = fopen($file->filepath(), 'rb')) {
- while (!feof($fp_in)) {
- fwrite($fp, fread($fp_in, 1024 * 512));
- }
- $success = TRUE;
- }
- @fclose($fp_in);
- // Finish the write.
- fwrite($fp, $data_footer);
- // Fetch response.
- $response = '';
- while (!feof($fp) && $chunk = fread($fp, 1024)) {
- $response .= $chunk;
- }
- fclose($fp);
-
- if (variable_get('debug_http_request', FALSE)) {
- drupal_debug(date('r'));
- drupal_debug($request);
- drupal_debug($response);
- }
-
- // Parse response.
- list($split, $result->data) = explode("\r\n\r\n", $response, 2);
- $split = preg_split("/\r\n|\n|\r/", $split);
-
- list($protocol, $code, $status_message) = explode(' ', trim(array_shift($split)), 3);
- $result->protocol = $protocol;
- $result->status_message = $status_message;
-
- $result->headers = array();
-
- // Parse headers.
- while ($line = trim(array_shift($split))) {
- list($header, $value) = explode(':', $line, 2);
- if (isset($result->headers[$header]) && $header == 'Set-Cookie') {
- // RFC 2109: the Set-Cookie response header comprises the token Set-
- // Cookie:, followed by a comma-separated list of one or more cookies.
- $result->headers[$header] .= ','. trim($value);
- }
- else {
- $result->headers[$header] = trim($value);
- }
- }
-
- $responses = array(
- 100 => 'Continue', 101 => 'Switching Protocols',
- 200 => 'OK', 201 => 'Created', 202 => 'Accepted', 203 => 'Non-Authoritative Information', 204 => 'No Content', 205 => 'Reset Content', 206 => 'Partial Content',
- 300 => 'Multiple Choices', 301 => 'Moved Permanently', 302 => 'Found', 303 => 'See Other', 304 => 'Not Modified', 305 => 'Use Proxy', 307 => 'Temporary Redirect',
- 400 => 'Bad Request', 401 => 'Unauthorized', 402 => 'Payment Required', 403 => 'Forbidden', 404 => 'Not Found', 405 => 'Method Not Allowed', 406 => 'Not Acceptable', 407 => 'Proxy Authentication Required', 408 => 'Request Time-out', 409 => 'Conflict', 410 => 'Gone', 411 => 'Length Required', 412 => 'Precondition Failed', 413 => 'Request Entity Too Large', 414 => 'Request-URI Too Large', 415 => 'Unsupported Media Type', 416 => 'Requested range not satisfiable', 417 => 'Expectation Failed',
- 500 => 'Internal Server Error', 501 => 'Not Implemented', 502 => 'Bad Gateway', 503 => 'Service Unavailable', 504 => 'Gateway Time-out', 505 => 'HTTP Version not supported'
- );
- // RFC 2616 states that all unknown HTTP codes must be treated the same as the
- // base code in their class.
- if (!isset($responses[$code])) {
- $code = floor($code / 100) * 100;
- }
-
- switch ($code) {
- case 200: // OK
- case 304: // Not modified
- break;
- case 301: // Moved permanently
- case 302: // Moved temporarily
- case 307: // Moved temporarily
- $location = $result->headers['Location'];
-
- if ($retry) {
- $result = drupal_http_request($result->headers['Location'], $headers, $method, $data, --$retry);
- $result->redirect_code = $result->code;
- }
- $result->redirect_url = $location;
-
- break;
- default:
- $result->error = $status_message;
- }
-
- $result->code = $code;
- return $result;
- }
- }
|