src/MagnetURI.cpp

   1 //                                                       -*- C++ -*-
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2011 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2007-2011 Dévai Tamás ( gonosztopi@amule.org )
   6 //
   7 // Any parts of this program derived from the xMule, lMule or eMule project,
   8 // or contributed by third-party developers are copyrighted by their
   9 // respective authors.
  10 //
  11 // This program is free software; you can redistribute it and/or modify
  12 // it under the terms of the GNU General Public License as published by
  13 // the Free Software Foundation; either version 2 of the License, or
  14 // (at your option) any later version.
  15 //
  16 // This program is distributed in the hope that it will be useful,
  17 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 // GNU General Public License for more details.
  20 //
  21 // You should have received a copy of the GNU General Public License
  22 // along with this program; if not, write to the Free Software
  23 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  24 //
  25
  26 #include "MagnetURI.h"
  27
  28
  29 #ifdef USE_STD_STRING
  30 #       ifdef _T
  31 #               undef   _T
  32 #       endif
  33 #       ifdef _C
  34 #               undef _C
  35 #       endif
  36 #       define  _T(str) str
  37 #       define  _C(ch)  ch
  38 #else
  39         // wx/chartype.h defines _T
  40 #       define  _C(ch)  wxChar(ch)
  41 #endif
  42
  43
  44 CMagnetURI::CMagnetURI(const STRING& uri)
  45 {
  46         if (uri.compare(0, 7, _T("magnet:")) == 0) {
  47                 size_t start = uri.find(_C('?'));
  48                 if (start == STRING::npos) start = uri.length();
  49                 while (start < uri.length() - 1) {
  50                         size_t end = uri.find(_C('&'), start + 1);
  51                         if (end == STRING::npos) end = uri.length();
  52                         size_t pos = uri.find(_C('='), start + 1);
  53                         if (pos == STRING::npos) pos = uri.length();
  54                         if (pos < end) {
  55                                 m_fields.push_back(Field_Type(uri.substr(start + 1, pos - start - 1), uri.substr(pos + 1, end - pos - 1)));
  56                         }
  57                         start = end;
  58                 }
  59         }
  60 }
  61
  62 STRING CMagnetURI::GetLink() const
  63 {
  64         STRING retval(_T("magnet:"));
  65
  66         for (List_Type::const_iterator it = m_fields.begin(); it != m_fields.end(); ++it) {
  67                 if (it == m_fields.begin()) {
  68                         retval.append(1, _C('?'));
  69                 } else {
  70                         retval.append(1, _C('&'));
  71                 }
  72                 retval.append(it->first);
  73                 retval.append(1, _C('='));
  74                 retval.append(it->second);
  75         }
  76         return retval;
  77 }
  78
  79 CMagnetURI::Value_List CMagnetURI::GetField(const STRING& name) const
  80 {
  81         Value_List retval;
  82
  83         for (List_Type::const_iterator it = m_fields.begin(); it != m_fields.end(); ++it) {
  84                 if (it->first.compare(name) == 0) {
  85                         retval.push_back(it->second);
  86                 }
  87         }
  88         return retval;
  89 }
  90
  91
  92 bool CMagnetED2KConverter::CanConvertToED2K() const
  93 {
  94         bool has_urn = false;
  95         bool has_xl = false;
  96
  97         for (List_Type::const_iterator it = m_fields.begin(); it != m_fields.end(); ++it) {
  98                 if (it->first.compare(_T("xl")) == 0) {
  99                         has_xl = true;
 100                         continue;
 101                 }
 102                 if (it->first.compare(_T("xt")) == 0) {
 103                         if ((it->second.compare(0, 9, _T("urn:ed2k:")) == 0) ||
 104                             (it->second.compare(0, 13, _T("urn:ed2khash:")) == 0))
 105                         {
 106                                 has_urn = true;
 107                                 continue;
 108                         }
 109                 }
 110                 if (has_urn && has_xl) break;
 111         }
 112         return has_urn && has_xl;
 113 }
 114
 115 STRING CMagnetED2KConverter::GetED2KLink() const
 116 {
 117         if (CanConvertToED2K()) {
 118                 STRING dn;
 119                 STRING hash;
 120                 STRING len(GetField(_T("xl")).front());
 121
 122                 Value_List dn_val = GetField(_T("dn"));
 123                 if (!dn_val.empty()) {
 124                         dn = dn_val.front();
 125                 } else {
 126                         // This should never happen. Has anyone seen a link without a file name?
 127                         // Just in case, assign a reasonable(?) name to that unnamed file.
 128                         dn = _T("FileName.ext");
 129                 }
 130                 Value_List urn_list = GetField(_T("xt"));
 131                 // Use the first ed2k-hash found.
 132                 for (Value_List::iterator it = urn_list.begin(); it != urn_list.end(); ++it) {
 133                         if (it->compare(0, 9, _T("urn:ed2k:")) == 0) {
 134                                 hash = it->substr(9);
 135                                 break;
 136                         } else if (it->compare(0, 13, _T("urn:ed2khash:")) == 0) {
 137                                 hash = it->substr(13);
 138                                 break;
 139                         }
 140                 }
 141                 return STRING(_T("ed2k://|file|")).append(dn).append(1, _C('|')).append(len).append(1, _C('|')).append(hash).append(_T("|/"));
 142         } else {
 143                 return STRING();
 144         }
 145 }