plugins/OStatus/actions/pushhub.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2010, StatusNet, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 /**
  21  * Integrated PuSH hub; lets us only ping them what need it.
  22  * @package Hub
  23  * @maintainer Brion Vibber <brion@status.net>
  24  */
  25
  26 if (!defined('STATUSNET')) {
  27     exit(1);
  28 }
  29
  30 /**
  31  * Things to consider...
  32  * should we purge incomplete subscriptions that never get a verification pingback?
  33  * when can we send subscription renewal checks?
  34  *    - at next send time probably ok
  35  * when can we handle trimming of subscriptions?
  36  *    - at next send time probably ok
  37  * should we keep a fail count?
  38  */
  39 class PushHubAction extends Action
  40 {
  41     function arg($arg, $def=null)
  42     {
  43         // PHP converts '.'s in incoming var names to '_'s.
  44         // It also merges multiple values, which'll break hub.verify and hub.topic for publishing
  45         // @fixme handle multiple args
  46         $arg = str_replace('hub.', 'hub_', $arg);
  47         return parent::arg($arg, $def);
  48     }
  49
  50     protected function prepare(array $args=array())
  51     {
  52         GNUsocial::setApi(true); // reduce exception reports to aid in debugging
  53         return parent::prepare($args);
  54     }
  55
  56     protected function handle()
  57     {
  58         $mode = $this->trimmed('hub.mode');
  59         switch ($mode) {
  60         case "subscribe":
  61         case "unsubscribe":
  62             $this->subunsub($mode);
  63             break;
  64         case "publish":
  65             // TRANS: Client exception.
  66             throw new ClientException(_m('Publishing outside feeds not supported.'), 400);
  67         default:
  68             // TRANS: Client exception. %s is a mode.
  69             throw new ClientException(sprintf(_m('Unrecognized mode "%s".'),$mode), 400);
  70         }
  71     }
  72
  73     /**
  74      * Process a request for a new or modified PuSH feed subscription.
  75      * If asynchronous verification is requested, updates won't be saved immediately.
  76      *
  77      * HTTP return codes:
  78      *   202 Accepted - request saved and awaiting verification
  79      *   204 No Content - already subscribed
  80      *   400 Bad Request - rejecting this (not specifically spec'd)
  81      */
  82     function subunsub($mode)
  83     {
  84         $callback = $this->argUrl('hub.callback');
  85
  86         $topic = $this->argUrl('hub.topic');
  87         if (!$this->recognizedFeed($topic)) {
  88             // TRANS: Client exception. %s is a topic.
  89             throw new ClientException(sprintf(_m('Unsupported hub.topic %s this hub only serves local user and group Atom feeds.'),$topic));
  90         }
  91
  92         $lease = $this->arg('hub.lease_seconds', null);
  93         if ($mode == 'subscribe' && $lease != '' && !preg_match('/^\d+$/', $lease)) {
  94             // TRANS: Client exception. %s is the invalid lease value.
  95             throw new ClientException(sprintf(_m('Invalid hub.lease "%s". It must be empty or positive integer.'),$lease));
  96         }
  97
  98         $secret = $this->arg('hub.secret', null);
  99         if ($secret != '' && strlen($secret) >= 200) {
 100             // TRANS: Client exception. %s is the invalid hub secret.
 101             throw new ClientException(sprintf(_m('Invalid hub.secret "%s". It must be under 200 bytes.'),$secret));
 102         }
 103
 104         $sub = HubSub::getByHashkey($topic, $callback);
 105         if (!$sub instanceof HubSub) {
 106             // Creating a new one!
 107             $sub = new HubSub();
 108             $sub->topic = $topic;
 109             $sub->callback = $callback;
 110         }
 111         if ($mode == 'subscribe') {
 112             if ($secret) {
 113                 $sub->secret = $secret;
 114             }
 115             if ($lease) {
 116                 $sub->setLease(intval($lease));
 117             }
 118         }
 119
 120         $verify = $this->arg('hub.verify'); // TODO: deprecated
 121         $token = $this->arg('hub.verify_token', null);  // TODO: deprecated
 122         if ($verify == 'sync') {    // pre-0.4 PuSH
 123             $sub->verify($mode, $token);
 124             header('HTTP/1.1 204 No Content');
 125         } else {    // If $verify is not "sync", we might be using PuSH 0.4
 126             $sub->scheduleVerify($mode, $token);    // If we were certain it's PuSH 0.4, token could be removed
 127             header('HTTP/1.1 202 Accepted');
 128         }
 129     }
 130
 131     /**
 132      * Check whether the given URL represents one of our canonical
 133      * user or group Atom feeds.
 134      *
 135      * @param string $feed URL
 136      * @return boolean true if it matches, false if not a recognized local feed
 137      * @throws exception if local entity does not exist
 138      */
 139     protected function recognizedFeed($feed)
 140     {
 141         $matches = array();
 142         // Simple mapping to local ID for user or group
 143         if (preg_match('!/(\d+)\.atom$!', $feed, $matches)) {
 144             $id = $matches[1];
 145             $params = array('id' => $id, 'format' => 'atom');
 146
 147             // Double-check against locally generated URLs
 148             switch ($feed) {
 149             case common_local_url('ApiTimelineUser', $params):
 150                 $user = User::getKV('id', $id);
 151                 if (!$user instanceof User) {
 152                     // TRANS: Client exception. %s is a feed URL.
 153                     throw new ClientException(sprintf(_m('Invalid hub.topic "%s". User does not exist.'),$feed));
 154                 }
 155                 return true;
 156
 157             case common_local_url('ApiTimelineGroup', $params):
 158                 $group = Local_group::getKV('group_id', $id);
 159                 if (!$group instanceof Local_group) {
 160                     // TRANS: Client exception. %s is a feed URL.
 161                     throw new ClientException(sprintf(_m('Invalid hub.topic "%s". Local_group does not exist.'),$feed));
 162                 }
 163                 return true;
 164             }
 165             common_debug("Feed was not recognized by any local User or Group Atom feed URLs: {$feed}");
 166             return false;
 167         }
 168
 169         // Profile lists are unique per user, so we need both IDs
 170         if (preg_match('!/(\d+)/lists/(\d+)/statuses\.atom$!', $feed, $matches)) {
 171             $user = $matches[1];
 172             $id = $matches[2];
 173             $params = array('user' => $user, 'id' => $id, 'format' => 'atom');
 174
 175             // Double-check against locally generated URLs
 176             switch ($feed) {
 177             case common_local_url('ApiTimelineList', $params):
 178                 $list = Profile_list::getKV('id', $id);
 179                 $user = User::getKV('id', $user);
 180                 if (!$list instanceof Profile_list || !$user instanceof User || $list->tagger != $user->id) {
 181                     // TRANS: Client exception. %s is a feed URL.
 182                     throw new ClientException(sprintf(_m('Invalid hub.topic %s; list does not exist.'),$feed));
 183                 }
 184                 return true;
 185             }
 186             common_debug("Feed was not recognized by any local Profile_list Atom feed URL: {$feed}");
 187             return false;
 188         }
 189
 190         common_debug("Unknown feed URL structure, can't match against local user, group or profile_list: {$feed}");
 191         return false;
 192     }
 193
 194     /**
 195      * Grab and validate a URL from POST parameters.
 196      * @throws ClientException for malformed or non-http/https URLs
 197      */
 198     protected function argUrl($arg)
 199     {
 200         $url = $this->arg($arg);
 201         $params = array('domain_check' => false, // otherwise breaks my local tests :P
 202                         'allowed_schemes' => array('http', 'https'));
 203         $validate = new Validate();
 204         if ($validate->uri($url, $params)) {
 205             return $url;
 206         } else {
 207             // TRANS: Client exception.
 208             // TRANS: %1$s is this argument to the method this exception occurs in, %2$s is a URL.
 209             throw new ClientException(sprintf(_m('Invalid URL passed for %1$s: "%2$s"'),$arg,$url));
 210         }
 211     }
 212
 213     /**
 214      * Get HubSub subscription record for a given feed & subscriber.
 215      *
 216      * @param string $feed
 217      * @param string $callback
 218      * @return mixed HubSub or false
 219      */
 220     protected function getSub($feed, $callback)
 221     {
 222         return HubSub::getByHashkey($feed, $callback);
 223     }
 224 }