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
  30 class MagicEnvelope
  31 {
  32     const ENCODING = 'base64url';
  33
  34     const NS = 'http://salmon-protocol.org/ns/magic-env';
  35
  36     private function normalizeUser($user_id)
  37     {
  38         if (substr($user_id, 0, 5) == 'http:' ||
  39             substr($user_id, 0, 6) == 'https:' ||
  40             substr($user_id, 0, 5) == 'acct:') {
  41             return $user_id;
  42         }
  43
  44         if (strpos($user_id, '@') !== FALSE) {
  45             return 'acct:' . $user_id;
  46         }
  47
  48         return 'http://' . $user_id;
  49     }
  50
  51     public function getKeyPair($signer_uri)
  52     {
  53         $disco = new Discovery();
  54
  55         try {
  56             $xrd = $disco->lookup($signer_uri);
  57         } catch (Exception $e) {
  58             return false;
  59         }
  60         if ($xrd->links) {
  61             if ($link = Discovery::getService($xrd->links, Magicsig::PUBLICKEYREL)) {
  62                 $keypair = false;
  63                 $parts = explode(',', $link['href']);
  64                 if (count($parts) == 2) {
  65                     $keypair = $parts[1];
  66                 } else {
  67                     // Backwards compatibility check for separator bug in 0.9.0
  68                     $parts = explode(';', $link['href']);
  69                     if (count($parts) == 2) {
  70                         $keypair = $parts[1];
  71                     }
  72                 }
  73
  74                 if ($keypair) {
  75                     return $keypair;
  76                 }
  77             }
  78         }
  79         // TRANS: Exception.
  80         throw new Exception(_m('Unable to locate signer public key.'));
  81     }
  82
  83
  84     public function signMessage($text, $mimetype, $keypair)
  85     {
  86         $signature_alg = Magicsig::fromString($keypair);
  87         $armored_text = Magicsig::base64_url_encode($text);
  88
  89         return array(
  90             'data' => $armored_text,
  91             'encoding' => MagicEnvelope::ENCODING,
  92             'data_type' => $mimetype,
  93             'sig' => $signature_alg->sign($armored_text),
  94             'alg' => $signature_alg->getName()
  95         );
  96     }
  97
  98     public function toXML($env) {
  99         $xs = new XMLStringer();
 100         $xs->startXML();
 101         $xs->elementStart('me:env', array('xmlns:me' => MagicEnvelope::NS));
 102         $xs->element('me:data', array('type' => $env['data_type']), $env['data']);
 103         $xs->element('me:encoding', null, $env['encoding']);
 104         $xs->element('me:alg', null, $env['alg']);
 105         $xs->element('me:sig', null, $env['sig']);
 106         $xs->elementEnd('me:env');
 107
 108         $string =  $xs->getString();
 109         common_debug($string);
 110         return $string;
 111     }
 112
 113     public function unfold($env)
 114     {
 115         $dom = new DOMDocument();
 116         $dom->loadXML(Magicsig::base64_url_decode($env['data']));
 117
 118         if ($dom->documentElement->tagName != 'entry') {
 119             return false;
 120         }
 121
 122         $prov = $dom->createElementNS(MagicEnvelope::NS, 'me:provenance');
 123         $prov->setAttribute('xmlns:me', MagicEnvelope::NS);
 124         $data = $dom->createElementNS(MagicEnvelope::NS, 'me:data', $env['data']);
 125         $data->setAttribute('type', $env['data_type']);
 126         $prov->appendChild($data);
 127         $enc = $dom->createElementNS(MagicEnvelope::NS, 'me:encoding', $env['encoding']);
 128         $prov->appendChild($enc);
 129         $alg = $dom->createElementNS(MagicEnvelope::NS, 'me:alg', $env['alg']);
 130         $prov->appendChild($alg);
 131         $sig = $dom->createElementNS(MagicEnvelope::NS, 'me:sig', $env['sig']);
 132         $prov->appendChild($sig);
 133
 134         $dom->documentElement->appendChild($prov);
 135
 136         return $dom->saveXML();
 137     }
 138
 139     public function getAuthor($text) {
 140         $doc = new DOMDocument();
 141         if (!$doc->loadXML($text)) {
 142             return FALSE;
 143         }
 144
 145         if ($doc->documentElement->tagName == 'entry') {
 146             $authors = $doc->documentElement->getElementsByTagName('author');
 147             foreach ($authors as $author) {
 148                 $uris = $author->getElementsByTagName('uri');
 149                 foreach ($uris as $uri) {
 150                     return $this->normalizeUser($uri->nodeValue);
 151                 }
 152             }
 153         }
 154     }
 155
 156     public function checkAuthor($text, $signer_uri)
 157     {
 158         return ($this->getAuthor($text) == $signer_uri);
 159     }
 160
 161     public function verify($env)
 162     {
 163         if ($env['alg'] != 'RSA-SHA256') {
 164             common_log(LOG_DEBUG, "Salmon error: bad algorithm");
 165             return false;
 166         }
 167
 168         if ($env['encoding'] != MagicEnvelope::ENCODING) {
 169             common_log(LOG_DEBUG, "Salmon error: bad encoding");
 170             return false;
 171         }
 172
 173         $text = Magicsig::base64_url_decode($env['data']);
 174         $signer_uri = $this->getAuthor($text);
 175
 176         try {
 177             $keypair = $this->getKeyPair($signer_uri);
 178         } catch (Exception $e) {
 179             common_log(LOG_DEBUG, "Salmon error: ".$e->getMessage());
 180             return false;
 181         }
 182
 183         $verifier = Magicsig::fromString($keypair);
 184
 185         if (!$verifier) {
 186             common_log(LOG_DEBUG, "Salmon error: unable to parse keypair");
 187             return false;
 188         }
 189
 190         return $verifier->verify($env['data'], $env['sig']);
 191     }
 192
 193     public function parse($text)
 194     {
 195         $dom = DOMDocument::loadXML($text);
 196         return $this->fromDom($dom);
 197     }
 198
 199     public function fromDom($dom)
 200     {
 201         $env_element = $dom->getElementsByTagNameNS(MagicEnvelope::NS, 'env')->item(0);
 202         if (!$env_element) {
 203             $env_element = $dom->getElementsByTagNameNS(MagicEnvelope::NS, 'provenance')->item(0);
 204         }
 205
 206         if (!$env_element) {
 207             return false;
 208         }
 209
 210         $data_element = $env_element->getElementsByTagNameNS(MagicEnvelope::NS, 'data')->item(0);
 211         $sig_element = $env_element->getElementsByTagNameNS(MagicEnvelope::NS, 'sig')->item(0);
 212         return array(
 213             'data' => preg_replace('/\s/', '', $data_element->nodeValue),
 214             'data_type' => $data_element->getAttribute('type'),
 215             'encoding' => $env_element->getElementsByTagNameNS(MagicEnvelope::NS, 'encoding')->item(0)->nodeValue,
 216             'alg' => $env_element->getElementsByTagNameNS(MagicEnvelope::NS, 'alg')->item(0)->nodeValue,
 217             'sig' => preg_replace('/\s/', '', $sig_element->nodeValue),
 218         );
 219     }
 220 }