plugins/OStatus/lib/magicenvelope.php

   1 <?php
   2 /**
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2010, StatusNet, Inc.
   5  *
   6  * A sample module to show best practices for StatusNet plugins
   7  *
   8  * PHP version 5
   9  *
  10  * This program is free software: you can redistribute it and/or modify
  11  * it under the terms of the GNU Affero General Public License as published by
  12  * the Free Software Foundation, either version 3 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU Affero General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Affero General Public License
  21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  * @package   StatusNet
  24  * @author    James Walker <james@status.net>
  25  * @copyright 2010 StatusNet, Inc.
  26  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
  27  * @link      http://status.net/
  28  */
  29 class MagicEnvelope
  30 {
  31     const ENCODING = 'base64url';
  32
  33     const NS = 'http://salmon-protocol.org/ns/magic-env';
  34
  35     protected $data      = null;    // When stored here it is _always_ base64url encoded
  36     protected $data_type = null;
  37     protected $encoding  = null;
  38     protected $alg       = null;
  39     protected $sig       = null;
  40
  41     /**
  42      * Extract envelope data from an XML document containing an <me:env> or <me:provenance> element.
  43      *
  44      * @param string XML source
  45      * @return mixed associative array of envelope data, or false on unrecognized input
  46      *
  47      * @fixme will spew errors to logs or output in case of XML parse errors
  48      * @fixme may give fatal errors if some elements are missing or invalid XML
  49      * @fixme calling DOMDocument::loadXML statically triggers warnings in strict mode
  50      */
  51     public function __construct($xml=null) {
  52         if (!empty($xml)) {
  53             $dom = DOMDocument::loadXML($xml);
  54             if (!$dom instanceof DOMDocument) {
  55                 throw new ServerException('Tried to load malformed XML as DOM');
  56             } elseif (!$this->fromDom($dom)) {
  57                 throw new ServerException('Could not load MagicEnvelope from DOM');
  58             }
  59         }
  60     }
  61
  62     /**
  63      * Retrieve Salmon keypair first by checking local database, but
  64      * if it's not found, attempt discovery if it has been requested.
  65      *
  66      * @param Profile $profile      The profile we're looking up keys for.
  67      * @param boolean $discovery    Network discovery if no local cache?
  68      */
  69     public function getKeyPair(Profile $profile, $discovery=false) {
  70         $magicsig = Magicsig::getKV('user_id', $profile->id);
  71
  72         if ($discovery && !$magicsig instanceof Magicsig) {
  73             // Throws exception on failure, but does not try to _load_ the keypair string.
  74             $keypair = $this->discoverKeyPair($profile);
  75
  76             $magicsig = new Magicsig();
  77             $magicsig->user_id = $profile->id;
  78             $magicsig->importKeys($keypair);
  79         } elseif (!$magicsig instanceof Magicsig) { // No discovery request, so we'll give up.
  80             throw new ServerException(sprintf('No public key found for profile (id==%d)', $profile->id));
  81         }
  82
  83         assert($magicsig->publicKey instanceof Crypt_RSA);
  84
  85         return $magicsig;
  86     }
  87
  88     /**
  89      * Get the Salmon keypair from a URI, uses XRD Discovery etc. Reasonably
  90      * you'll only get the public key ;)
  91      *
  92      * The string will (hopefully) be formatted as described in Magicsig specification:
  93      * https://salmon-protocol.googlecode.com/svn/trunk/draft-panzer-magicsig-01.html#anchor13
  94      *
  95      * @return string formatted as Magicsig keypair
  96      */
  97     public function discoverKeyPair(Profile $profile)
  98     {
  99         $signer_uri = $profile->getUri();
 100         if (empty($signer_uri)) {
 101             throw new ServerException(sprintf('Profile missing URI (id==%d)', $profile->id));
 102         }
 103
 104         $disco = new Discovery();
 105
 106         // Throws exception on lookup problems
 107         $xrd = $disco->lookup($signer_uri);
 108
 109         $link = $xrd->get(Magicsig::PUBLICKEYREL);
 110         if (is_null($link)) {
 111             // TRANS: Exception.
 112             throw new Exception(_m('Unable to locate signer public key.'));
 113         }
 114
 115         // We have a public key element, let's hope it has proper key data.
 116         $keypair = false;
 117         $parts = explode(',', $link->href);
 118         if (count($parts) == 2) {
 119             $keypair = $parts[1];
 120         } else {
 121             // Backwards compatibility check for separator bug in 0.9.0
 122             $parts = explode(';', $link->href);
 123             if (count($parts) == 2) {
 124                 $keypair = $parts[1];
 125             }
 126         }
 127
 128         if ($keypair === false) {
 129             // For debugging clarity. Keypair did not pass count()-check above.
 130             // TRANS: Exception when public key was not properly formatted.
 131             throw new Exception(_m('Incorrectly formatted public key element.'));
 132         }
 133
 134         return $keypair;
 135     }
 136
 137     /**
 138      * The current MagicEnvelope spec as used in StatusNet 0.9.7 and later
 139      * includes both the original data and some signing metadata fields as
 140      * the input plaintext for the signature hash.
 141      *
 142      * @return string
 143      */
 144     public function signingText() {
 145         return implode('.', array($this->data, // this field is pre-base64'd
 146                             Magicsig::base64_url_encode($this->data_type),
 147                             Magicsig::base64_url_encode($this->encoding),
 148                             Magicsig::base64_url_encode($this->alg)));
 149     }
 150
 151     /**
 152      *
 153      * @param <type> $text
 154      * @param <type> $mimetype
 155      * @param Magicsig $magicsig    Magicsig with private key available.
 156      *
 157      * @return MagicEnvelope object with all properties set
 158      *
 159      * @throws Exception of various kinds on signing failure
 160      */
 161     public function signMessage($text, $mimetype, Magicsig $magicsig)
 162     {
 163         assert($magicsig->privateKey instanceof Crypt_RSA);
 164
 165         // Prepare text and metadata for signing
 166         $this->data = Magicsig::base64_url_encode($text);
 167         $this->data_type = $mimetype;
 168         $this->encoding  = self::ENCODING;
 169         $this->alg       = $magicsig->getName();
 170
 171         // Get the actual signature
 172         $this->sig = $magicsig->sign($this->signingText());
 173     }
 174
 175     /**
 176      * Create an <me:env> XML representation of the envelope.
 177      *
 178      * @return string representation of XML document
 179      */
 180     public function toXML() {
 181         $xs = new XMLStringer();
 182         $xs->startXML();
 183         $xs->elementStart('me:env', array('xmlns:me' => self::NS));
 184         $xs->element('me:data', array('type' => $this->data_type), $this->data);
 185         $xs->element('me:encoding', null, $this->encoding);
 186         $xs->element('me:alg', null, $this->alg);
 187         $xs->element('me:sig', null, $this->getSignature());
 188         $xs->elementEnd('me:env');
 189
 190         $string =  $xs->getString();
 191         return $string;
 192     }
 193
 194     /*
 195      * Extract the contained XML payload, and insert a copy of the envelope
 196      * signature data as an <me:provenance> section.
 197      *
 198      * @return DOMDocument of Atom entry
 199      *
 200      * @fixme in case of XML parsing errors, this will spew to the error log or output
 201      */
 202     public function getPayload()
 203     {
 204         $dom = new DOMDocument();
 205         if (!$dom->loadXML(Magicsig::base64_url_decode($this->data))) {
 206             throw new ServerException('Malformed XML in Salmon payload');
 207         }
 208
 209         switch ($this->data_type) {
 210         case 'application/atom+xml':
 211             if ($dom->documentElement->namespaceURI !== Activity::ATOM
 212                     || $dom->documentElement->tagName !== 'entry') {
 213                 throw new ServerException(_m('Salmon post must be an Atom entry.'));
 214             }
 215             $prov = $dom->createElementNS(self::NS, 'me:provenance');
 216             $prov->setAttribute('xmlns:me', self::NS);
 217             $data = $dom->createElementNS(self::NS, 'me:data', $this->data);
 218             $data->setAttribute('type', $this->data_type);
 219             $prov->appendChild($data);
 220             $enc = $dom->createElementNS(self::NS, 'me:encoding', $this->encoding);
 221             $prov->appendChild($enc);
 222             $alg = $dom->createElementNS(self::NS, 'me:alg', $this->alg);
 223             $prov->appendChild($alg);
 224             $sig = $dom->createElementNS(self::NS, 'me:sig', $this->getSignature());
 225             $prov->appendChild($sig);
 226
 227             $dom->documentElement->appendChild($prov);
 228             break;
 229         default:
 230             throw new ServerException('Unknown Salmon payload data type');
 231         }
 232         return $dom;
 233     }
 234
 235     public function getSignature()
 236     {
 237         return $this->sig;
 238     }
 239
 240     /**
 241      * Find the author URI referenced in the payload Atom entry.
 242      *
 243      * @return string URI for author
 244      * @throws ServerException on failure
 245      */
 246     public function getAuthorUri() {
 247         $doc = $this->getPayload();
 248
 249         $authors = $doc->documentElement->getElementsByTagName('author');
 250         foreach ($authors as $author) {
 251             $uris = $author->getElementsByTagName('uri');
 252             foreach ($uris as $uri) {
 253                 return $uri->nodeValue;
 254             }
 255         }
 256         throw new ServerException('No author URI found in Salmon payload data');
 257     }
 258
 259     /**
 260      * Attempt to verify cryptographic signing for parsed envelope data.
 261      * Requires network access to retrieve public key referenced by the envelope signer.
 262      *
 263      * Details of failure conditions are dumped to output log and not exposed to caller.
 264      *
 265      * @param Profile $profile profile used to get locally cached public signature key
 266      *                         or if necessary perform discovery on.
 267      *
 268      * @return boolean
 269      */
 270     public function verify(Profile $profile)
 271     {
 272         if ($this->alg != 'RSA-SHA256') {
 273             common_debug("Salmon error: bad algorithm");
 274             return false;
 275         }
 276
 277         if ($this->encoding != self::ENCODING) {
 278             common_debug("Salmon error: bad encoding");
 279             return false;
 280         }
 281
 282         try {
 283             $magicsig = $this->getKeyPair($profile, true);    // Do discovery too if necessary
 284         } catch (Exception $e) {
 285             common_debug("Salmon error: ".$e->getMessage());
 286             return false;
 287         }
 288
 289         return $magicsig->verify($this->signingText(), $this->getSignature());
 290     }
 291
 292     /**
 293      * Extract envelope data from an XML document containing an <me:env> or <me:provenance> element.
 294      *
 295      * @param DOMDocument $dom
 296      * @return mixed associative array of envelope data, or false on unrecognized input
 297      *
 298      * @fixme may give fatal errors if some elements are missing
 299      */
 300     protected function fromDom(DOMDocument $dom)
 301     {
 302         $env_element = $dom->getElementsByTagNameNS(self::NS, 'env')->item(0);
 303         if (!$env_element) {
 304             $env_element = $dom->getElementsByTagNameNS(self::NS, 'provenance')->item(0);
 305         }
 306
 307         if (!$env_element) {
 308             return false;
 309         }
 310
 311         $data_element = $env_element->getElementsByTagNameNS(self::NS, 'data')->item(0);
 312         $sig_element = $env_element->getElementsByTagNameNS(self::NS, 'sig')->item(0);
 313
 314         $this->data      = preg_replace('/\s/', '', $data_element->nodeValue);
 315         $this->data_type = $data_element->getAttribute('type');
 316         $this->encoding  = $env_element->getElementsByTagNameNS(self::NS, 'encoding')->item(0)->nodeValue;
 317         $this->alg       = $env_element->getElementsByTagNameNS(self::NS, 'alg')->item(0)->nodeValue;
 318         $this->sig       = preg_replace('/\s/', '', $sig_element->nodeValue);
 319         return true;
 320     }
 321
 322     /**
 323      * Encode the given string as a signed MagicEnvelope XML document,
 324      * using the keypair for the given local user profile. We can of
 325      * course not sign a remote profile's slap, since we don't have the
 326      * private key.
 327      *
 328      * Side effects: will create and store a keypair on-demand if one
 329      * hasn't already been generated for this user. This can be very slow
 330      * on some systems.
 331      *
 332      * @param string $text XML fragment to sign, assumed to be Atom
 333      * @param User $user User who cryptographically signs $text
 334      *
 335      * @return MagicEnvelope object complete with signature
 336      *
 337      * @throws Exception on bad profile input or key generation problems
 338      */
 339     public static function signAsUser($text, User $user)
 340     {
 341         // Find already stored key
 342         $magicsig = Magicsig::getKV('user_id', $user->id);
 343         if (!$magicsig instanceof Magicsig) {
 344             $magicsig = Magicsig::generate($user);
 345         }
 346         assert($magicsig instanceof Magicsig);
 347         assert($magicsig->privateKey instanceof Crypt_RSA);
 348
 349         $magic_env = new MagicEnvelope();
 350         $magic_env->signMessage($text, 'application/atom+xml', $magicsig);
 351
 352         return $magic_env;
 353     }
 354 }