File: FingerprintManager.cpp

package info (click to toggle)
sslsniff 0.6-1
  • links: PTS
  • area: main
  • in suites: squeeze
  • size: 624 kB
  • ctags: 348
  • sloc: cpp: 2,067; sh: 994; makefile: 63; xml: 12
file content (75 lines) | stat: -rw-r--r-- 2,801 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
 * Copyright (c) 2002-2009 Moxie Marlinspike
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

#include "FingerprintManager.hpp"
#include "util/Util.hpp"

FingerprintManager* FingerprintManager::manager = new FingerprintManager();

FingerprintManager* FingerprintManager::getInstance() {
  return manager;
}

void FingerprintManager::setValidUserAgents(std::string &validAgents) {
  if (!validAgents.empty()) {
    std::string delimiters(" ,");
    Util::tokenizeString(validAgents, delimiters, this->validAgents);
  }
}

void FingerprintManager::setUserAgentFor(ip::address &address, std::string &userAgent) {
  //  std::cerr << "FM: Got Fingerprint (" << address << "):" << userAgent << std::endl;
  fingerprints[address] = userAgent;  
}

bool FingerprintManager::isValidWildcardTarget(ip::address &address) {
  std::string *userAgent = getUserAgentFor(address);

  return 
    isValidTarget(address) &&
    userAgent != NULL      &&
    !(userAgent->empty())  &&
    userAgent->find("Firefox") != std::string::npos;
}

bool FingerprintManager::isValidTarget(ip::address &address) {
  if (validAgents.size() == 0) return true; // We're not fingerprinting.
  
  std::string* userAgent = getUserAgentFor(address);
  if (userAgent == NULL || userAgent->empty()) return false;

  std::vector<std::string>::iterator iter = validAgents.begin();

  while (iter != validAgents.end()) {
    if      (*iter == "ff" && (userAgent->find("Firefox") != std::string::npos) && (userAgent->find("Firefox/3.5") == std::string::npos))    return true;
    else if (*iter == "ie" && (userAgent->find("MSIE") != std::string::npos))       return true;
    else if (*iter == "safari" && (userAgent->find("Safari") != std::string::npos)) return true;
    else if (*iter == "opera" && (userAgent->find("Opera") != std::string::npos))   return true;

    iter++;
  }

  return false;
}

std::string* FingerprintManager::getUserAgentFor(ip::address &address) {
  std::map<ip::address,std::string>::iterator iter = fingerprints.find(address);
  if( iter != fingerprints.end() ) return &(iter->second);
  else                             return NULL;
}