Gazelle/classes/tracker.class.php

<?
// TODO: Turn this into a class with nice functions like update_user, delete_torrent, etc.
class Tracker {
  const STATS_MAIN = 0;
  const STATS_USER = 1;

  public static $Requests = array();

  /**
   * Send a GET request over a socket directly to the tracker
   * For example, Tracker::update_tracker('change_passkey', array('oldpasskey' => OLD_PASSKEY, 'newpasskey' => NEW_PASSKEY)) will send the request:
   * GET /tracker_32_char_secret_code/update?action=change_passkey&oldpasskey=OLD_PASSKEY&newpasskey=NEW_PASSKEY HTTP/1.1
   *
   * @param string $Action The action to send
   * @param array $Updates An associative array of key->value pairs to send to the tracker
   * @param boolean $ToIRC Sends a message to the channel #tracker with the GET URL.
   */
  public static function update_tracker($Action, $Updates, $ToIRC = false) {
    // Build request
    $Get = TRACKER_SECRET . "/update?action=$Action";
    foreach ($Updates as $Key => $Value) {
      $Get .= "&$Key=$Value";
    }

    $MaxAttempts = 3;
    // don't wait around if we're debugging
    if (DEBUG_MODE) {
      $MaxAttempts = 1;
    }

    $Err = false;
    if (self::send_request($Get, $MaxAttempts, $Err) === false) {
      send_irc("PRIVMSG #tracker :$MaxAttempts $Err $Get");
      if (G::$Cache->get_value('ocelot_error_reported') === false) {
        send_irc('PRIVMSG ' . ADMIN_CHAN . " :Failed to update ocelot: $Err : $Get");
        G::$Cache->cache_value('ocelot_error_reported', true, 3600);
      }
      return false;
    }
    return true;
  }

  /**
   * Get global peer stats from the tracker
   *
   * @return array(0 => $Leeching, 1 => $Seeding) or false if request failed
   */
  public static function global_peer_count() {
    $Stats = self::get_stats(self::STATS_MAIN);
    if (isset($Stats['leechers tracked']) && isset($Stats['seeders tracked'])) {
      $Leechers = $Stats['leechers tracked'];
      $Seeders = $Stats['seeders tracked'];
    } else {
      return false;
    }
    return array($Leechers, $Seeders);
  }

  /**
   * Get peer stats for a user from the tracker
   *
   * @param string $TorrentPass The user's pass key
   * @return array(0 => $Leeching, 1 => $Seeding) or false if the request failed
   */
  public static function user_peer_count($TorrentPass) {
    $Stats = self::get_stats(self::STATS_USER, array('key' => $TorrentPass));
    if ($Stats === false) {
      return false;
    }
    if (isset($Stats['leeching']) && isset($Stats['seeding'])) {
      $Leeching = $Stats['leeching'];
      $Seeding = $Stats['seeding'];
    } else {
      // User doesn't exist, but don't tell anyone
      $Leeching = $Seeding = 0;
    }
    return array($Leeching, $Seeding);
  }

  /**
   * Get whatever info the tracker has to report
   *
   * @return results from get_stats()
   */
  public static function info() {
    return self::get_stats(self::STATS_MAIN);
  }

  /**
   * Send a stats request to the tracker and process the results
   *
   * @param int $Type Stats type to get
   * @param array $Params Parameters required by stats type
   * @return array with stats in named keys or false if the request failed
   */
  private static function get_stats($Type, $Params = false) {
    if (!defined('TRACKER_REPORTKEY')) {
      return false;
    }
    $Get = TRACKER_REPORTKEY . '/report?';
    if ($Type === self::STATS_MAIN) {
      $Get .= 'get=stats';
    } elseif ($Type === self::STATS_USER && !empty($Params['key'])) {
      $Get .= "get=user&key=$Params[key]";
    } else {
      return false;
    }
    $Response = self::send_request($Get);
    if ($Response === false) {
      return false;
    }
    $Stats = array();
    foreach (explode("\n", $Response) as $Stat) {
      list($Val, $Key) = explode(" ", $Stat, 2);
      $Stats[$Key] = $Val;
    }
    return $Stats;
  }

  /**
   * Send a request to the tracker
   *
   * @param string $Path GET string to send to the tracker
   * @param int $MaxAttempts Maximum number of failed attempts before giving up
   * @param $Err Variable to use as storage for the error string if the request fails
   * @return tracker response message or false if the request failed
   */
  private static function send_request($Get, $MaxAttempts = 1, &$Err = false) {
    $Header = "GET /$Get HTTP/1.1\r\nConnection: Close\r\n\r\n";
    $Attempts = 0;
    $Sleep = 0;
    $Success = false;
    $StartTime = microtime(true);
    while (!$Success && $Attempts++ < $MaxAttempts) {
      if ($Sleep) {
        sleep($Sleep);
      }

      // spend some time retrying if we're not in DEBUG_MODE
      if (!DEBUG_MODE) {
        $Sleep = 6;
      }

      // Send request
      $File = fsockopen(TRACKER_HOST, TRACKER_PORT, $ErrorNum, $ErrorString);
      if ($File) {
        if (fwrite($File, $Header) === false) {
          $Err = "Failed to fwrite()";
          $Sleep = 3;
          continue;
        }
      } else {
        $Err = "Failed to fsockopen() - $ErrorNum - $ErrorString";
        continue;
      }

      // Check for response.
      $Response = '';
      while (!feof($File)) {
        $Response .= fread($File, 1024);
      }
      $DataStart = strpos($Response, "\r\n\r\n") + 4;
      $DataEnd = strrpos($Response, "\n");
      if ($DataEnd > $DataStart) {
        $Data = substr($Response, $DataStart, $DataEnd - $DataStart);
      } else {
        $Data = "";
      }
      $Status = substr($Response, $DataEnd + 1);
      if ($Status == "success") {
        $Success = true;
      }
    }
    $Request = array(
      'path' => substr($Get, strpos($Get, '/')),
      'response' => ($Success ? $Data : $Response),
      'status' => ($Success ? 'ok' : 'failed'),
      'time' => 1000 * (microtime(true) - $StartTime)
    );
    self::$Requests[] = $Request;
    if ($Success) {
      return $Data;
    }
    return false;
  }
}
?>