src/FriendList.cpp

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2008 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2002-2008 Merkur ( devs@emule-project.net / http://www.emule-project.net )
   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
  27 #include "FriendList.h" // Interface
  28
  29 #include <common/DataFileVersion.h>
  30
  31 #include "amule.h"                      // Needed for theApp: let it first or fail under win32
  32 #include "ClientList.h"         // Needed for CClientList
  33 #include "updownclient.h"       // Needed for CUpDownClient
  34 #include "Friend.h"             // Needed for CFriend
  35 #include "CFile.h"
  36 #include "Logger.h"
  37
  38 CFriendList::CFriendList()
  39 {
  40         LoadList();
  41 }
  42
  43 CFriendList::~CFriendList()
  44 {
  45         SaveList();
  46
  47         while ( m_FriendList.size() ) {
  48                 delete m_FriendList.front();
  49                 m_FriendList.pop_front();
  50         }
  51 }
  52
  53
  54 void CFriendList::AddFriend(CFriend* toadd)
  55 {
  56         m_FriendList.push_back(toadd);
  57         SaveList();
  58 }
  59
  60
  61 void CFriendList::AddFriend(const CMD4Hash& userhash, uint32 lastSeen, uint32 lastUsedIP, uint32 lastUsedPort, uint32 lastChatted, const wxString& name)
  62 {
  63         CFriend* NewFriend = new CFriend( userhash, lastSeen, lastUsedIP, lastUsedPort, lastChatted, name);
  64
  65         AddFriend( NewFriend );
  66 }
  67
  68
  69 void CFriendList::AddFriend(CUpDownClient* toadd)
  70 {
  71         if ( toadd->IsFriend() ) {
  72                 return;
  73         }
  74
  75         CFriend* NewFriend = new CFriend( toadd );
  76         toadd->SetFriend(NewFriend);
  77
  78         AddFriend( NewFriend );
  79 }
  80
  81
  82 void CFriendList::RemoveFriend(const CMD4Hash& userhash, uint32 lastUsedIP, uint32 lastUsedPort)
  83 {
  84         CFriend* toremove = FindFriend(userhash, lastUsedIP, lastUsedPort);
  85         if (toremove) {
  86                 if ( toremove->GetLinkedClient() ){
  87                         toremove->GetLinkedClient()->SetFriendSlot(false);
  88                         toremove->GetLinkedClient()->SetFriend(NULL);
  89                         toremove->UnLinkClient();
  90                 }
  91
  92                 m_FriendList.remove(toremove);
  93
  94                 SaveList();
  95
  96                 delete toremove;
  97         }
  98 }
  99
 100 void CFriendList::LoadList()
 101 {
 102         CPath metfile = CPath(theApp->ConfigDir + wxT("emfriends.met"));
 103
 104         if (!metfile.FileExists()) {
 105                 return;
 106         }
 107
 108         CFile file;
 109         try {
 110                 if ( file.Open(metfile) ) {
 111                         if ( file.ReadUInt8() /*header*/ == MET_HEADER ) {
 112                                 uint32 nRecordsNumber = file.ReadUInt32();
 113                                 for (uint32 i = 0; i < nRecordsNumber; i++) {
 114                                         CFriend* Record = new CFriend();
 115                                         Record->LoadFromFile(&file);
 116                                         m_FriendList.push_back(Record);
 117                                 }
 118                         }
 119                 } else {
 120                         AddLogLineM(false, _("Failed to open friend list file 'emfriends.met' for reading!"));
 121                 }
 122         } catch (const CInvalidPacket& e) {
 123                 AddDebugLogLineM(true, logGeneral, wxT("Invalid entry in friend list, file may be corrupt: ") + e.what());
 124         } catch (const CSafeIOException& e) {
 125                 AddDebugLogLineM(true, logGeneral, wxT("IO error while reading 'emfriends.met': ") + e.what());
 126         }
 127
 128 }
 129
 130
 131 void CFriendList::SaveList()
 132 {
 133         CFile file;
 134         if (file.Create(theApp->ConfigDir + wxT("emfriends.met"), true)) {
 135                 try {
 136                         file.WriteUInt8(MET_HEADER);
 137                         file.WriteUInt32(m_FriendList.size());
 138
 139                         for (FriendList::iterator it = m_FriendList.begin(); it != m_FriendList.end(); ++it) {
 140                                 (*it)->WriteToFile(&file);
 141                         }
 142                 } catch (const CIOFailureException& e) {
 143                         AddDebugLogLineM(true, logGeneral, wxT("IO failure while saving 'emfriends.met': ") + e.what());
 144                 }
 145         } else {
 146                 AddLogLineM(false, _("Failed to open friend list file 'emfriends.met' for writing!"));
 147         }
 148 }
 149
 150
 151 CFriend* CFriendList::FindFriend(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort)
 152 {
 153
 154         for(FriendList::iterator it = m_FriendList.begin(); it != m_FriendList.end(); ++it) {
 155
 156                 CFriend* cur_friend = *it;
 157                 // to avoid that unwanted clients become a friend, we have to distinguish between friends with
 158                 // a userhash and of friends which are identified by IP+port only.
 159                 if ( !userhash.IsEmpty() && cur_friend->HasHash() ) {
 160                         // check for a friend which has the same userhash as the specified one
 161                         if (cur_friend->GetUserHash() == userhash) {
 162                                 return cur_friend;
 163                         }
 164                 } else if (cur_friend->GetIP() == dwIP && cur_friend->GetPort() == nPort) {
 165                                 return cur_friend;
 166                 }
 167         }
 168
 169         return NULL;
 170 }
 171
 172
 173 bool CFriendList::IsAlreadyFriend( uint32 dwLastUsedIP, uint32 nLastUsedPort )
 174 {
 175         return (FindFriend( CMD4Hash(), dwLastUsedIP, nLastUsedPort ) != NULL);
 176 }
 177
 178
 179 void CFriendList::RemoveAllFriendSlots()
 180 {
 181         for(FriendList::iterator it = m_FriendList.begin(); it != m_FriendList.end(); ++it) {
 182                 CFriend* cur_friend = *it;
 183                 if ( cur_friend->GetLinkedClient() ) {
 184                                 cur_friend->GetLinkedClient()->SetFriendSlot(false);
 185                 }
 186         }
 187 }
 188
 189 void    CFriendList::RequestSharedFileList(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort) {
 190         CFriend* cur_friend = FindFriend(userhash, dwIP, nPort);
 191         if (cur_friend) {
 192                 CUpDownClient* client = cur_friend->GetLinkedClient();
 193                 if (!client) {
 194                         client = new CUpDownClient(cur_friend->GetPort(), cur_friend->GetIP(), 0, 0, 0, true, true);
 195                         client->SetUserName(cur_friend->GetName());
 196                         theApp->clientlist->AddClient(client);
 197                 }
 198                 client->RequestSharedFileList();
 199         }
 200 }
 201
 202 void CFriendList::SetFriendSlot(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort, bool new_state) {
 203         CFriend* cur_friend = FindFriend(userhash, dwIP, nPort);
 204         if (cur_friend && cur_friend->GetLinkedClient()) {
 205                 RemoveAllFriendSlots();
 206                 cur_friend->GetLinkedClient()->SetFriendSlot(new_state);
 207         }
 208 }
 209
 210 void CFriendList::StartChatSession(const CMD4Hash& userhash, uint32 dwIP, uint16 nPort) {
 211         CFriend* friend_client = FindFriend(userhash, dwIP, nPort);
 212         if (friend_client) {
 213                 CUpDownClient* client = friend_client->GetLinkedClient();
 214                 if (!client) {
 215                         client = new CUpDownClient(friend_client->GetPort(), friend_client->GetIP(), 0, 0, 0, true, true);
 216                         client->SetIP(friend_client->GetIP());
 217                         client->SetUserName(friend_client->GetName());
 218                         theApp->clientlist->AddClient(client);
 219                         friend_client->LinkClient(client);
 220                 }
 221         } else {
 222                 printf("CRITICAL - no client on StartChatSession\n");
 223         }
 224
 225 }
 226
 227 void CFriendList::UpdateFriendName(const CMD4Hash& userhash, const wxString& name, uint32 dwIP, uint16 nPort) {
 228         CFriend* friend_client = FindFriend(userhash, dwIP, nPort);
 229         friend_client->SetName(name);
 230         SaveList();
 231 }
 232 // File_checked_for_headers