/* * This file is part of Transdroid * * Transdroid 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. * * Transdroid 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 Transdroid. If not, see . * */ package org.transdroid.daemon.adapters.qBittorrent; import com.android.internal.http.multipart.FilePart; import com.android.internal.http.multipart.MultipartEntity; import com.android.internal.http.multipart.Part; import org.apache.http.HttpEntity; import org.apache.http.HttpResponse; import org.apache.http.NameValuePair; import org.apache.http.client.entity.UrlEncodedFormEntity; import org.apache.http.client.methods.HttpPost; import org.apache.http.cookie.Cookie; import org.apache.http.impl.client.DefaultHttpClient; import org.apache.http.message.BasicNameValuePair; import org.apache.http.protocol.HTTP; import org.json.JSONArray; import org.json.JSONException; import org.json.JSONObject; import org.transdroid.core.gui.log.Log; import org.transdroid.daemon.Daemon; import org.transdroid.daemon.DaemonException; import org.transdroid.daemon.DaemonException.ExceptionType; import org.transdroid.daemon.DaemonSettings; import org.transdroid.daemon.IDaemonAdapter; import org.transdroid.daemon.Label; import org.transdroid.daemon.Priority; import org.transdroid.daemon.Torrent; import org.transdroid.daemon.TorrentDetails; import org.transdroid.daemon.TorrentFile; import org.transdroid.daemon.TorrentStatus; import org.transdroid.daemon.task.AddByFileTask; import org.transdroid.daemon.task.AddByMagnetUrlTask; import org.transdroid.daemon.task.AddByUrlTask; import org.transdroid.daemon.task.DaemonTask; import org.transdroid.daemon.task.DaemonTaskFailureResult; import org.transdroid.daemon.task.DaemonTaskResult; import org.transdroid.daemon.task.DaemonTaskSuccessResult; import org.transdroid.daemon.task.GetFileListTask; import org.transdroid.daemon.task.GetFileListTaskSuccessResult; import org.transdroid.daemon.task.GetStatsTask; import org.transdroid.daemon.task.GetStatsTaskSuccessResult; import org.transdroid.daemon.task.GetTorrentDetailsTask; import org.transdroid.daemon.task.GetTorrentDetailsTaskSuccessResult; import org.transdroid.daemon.task.RemoveTask; import org.transdroid.daemon.task.RetrieveTask; import org.transdroid.daemon.task.RetrieveTaskSuccessResult; import org.transdroid.daemon.task.SetDownloadLocationTask; import org.transdroid.daemon.task.SetFilePriorityTask; import org.transdroid.daemon.task.SetLabelTask; import org.transdroid.daemon.task.SetTransferRatesTask; import org.transdroid.daemon.util.HttpHelper; import java.io.File; import java.io.FileNotFoundException; import java.io.UnsupportedEncodingException; import java.net.URI; import java.util.ArrayList; import java.util.Collections; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map; /** * The daemon adapter for the qBittorrent torrent client. * * @author erickok */ public class QBittorrentAdapter implements IDaemonAdapter { private static final String LOG_NAME = "qBittorrent daemon"; private DaemonSettings settings; private DefaultHttpClient httpclient; private int version = -1; private long lastAuthTime = -1; private int qbNoPriority = 0; private int qbLowPriority = 1; private int qbNormalPriority = 2; private int qbHighPriority = 7; public QBittorrentAdapter(DaemonSettings settings) { this.settings = settings; } private synchronized void ensureVersion(Log log) { // Still need to retrieve the API and qBittorrent version numbers from the server? if (version > 0) return; // Since 4.1, API v2 is used. Since qBittorrent 3.2, API v1 is used. Otherwise we use unofficial legacy json endpoints. try { String versionText = ""; try { // Try v2 API first, which returns version number in 'v4.1.9' format versionText = makeRequest(log, "/api/v2/app/version").substring(1); } catch (Exception e1) { // Try v1 API, which returns version number in 'v3.2.0' format try { versionText = makeRequest(log, "/version/qbittorrent").substring(1); } catch (Exception e2) { // Legacy mode; format is something like 'qBittorrent v2.9.7 (Web UI)' or 'qBittorrent v3.0.0-alpha5 (Web UI)' String about = makeRequest(log, "/about.html"); String aboutStartText = "qBittorrent v"; String aboutEndText = " (Web UI)"; int aboutStart = about.indexOf(aboutStartText); int aboutEnd = about.indexOf(aboutEndText); if (aboutStart >= 0 && aboutEnd > aboutStart) { versionText = about.substring(aboutStart + aboutStartText.length(), aboutEnd); } } } version = parseVersionNumber(versionText); if (version >= 30200) { qbNormalPriority = 6; } } catch (Exception e) { // Unable to establish version number; assume an old version by setting it to version 1 version = 10000; qbNormalPriority = 2; } } private int parseVersionNumber(String versionText) { // String found: now parse a version like 2.9.7 as a number like 20907 (allowing 10 places for each .) int version = -1; String[] parts = versionText.split("\\."); if (parts.length > 0) { version = Integer.parseInt(parts[0]) * 100 * 100; if (parts.length > 1) { version += Integer.parseInt(parts[1]) * 100; if (parts.length > 2) { // For the last part only read until a non-numeric character is read // For example version 3.0.0-alpha5 is read as version code 30000 String numbers = ""; for (char c : parts[2].toCharArray()) { if (Character.isDigit(c)) // Still a number; add it to the numbers string numbers += Character.toString(c); else { // No longer reading numbers; stop reading break; } } version += Integer.parseInt(numbers); } } } return version; } private synchronized void ensureAuthenticated(Log log) throws DaemonException { // Have we already authenticated? Check if we have the cookie that we need if (isAuthenticated()) { return; } final BasicNameValuePair usernameParam = new BasicNameValuePair("username", settings.getUsername()); final BasicNameValuePair passwordParam = new BasicNameValuePair("password", settings.getPassword()); // Try qBittorrent 4.1 API v2 first try { makeRequest(log, "/api/v2/auth/login", usernameParam, passwordParam); lastAuthTime = System.currentTimeMillis(); } catch (DaemonException ignored) { } // If still not authenticated, try the qBittorrent 3.2 API v1 endpoint if (!isAuthenticated()) { try { makeRequest(log, "/login", usernameParam, passwordParam); lastAuthTime = System.currentTimeMillis(); } catch (DaemonException ignored) { } } if (!isAuthenticated()) { throw new DaemonException(ExceptionType.AuthenticationFailure, "Server rejected our login"); } } private boolean isAuthenticated() { if (System.currentTimeMillis() - lastAuthTime > MAX_SESSION_TIME) { return false; } List cookies = httpclient.getCookieStore().getCookies(); for (Cookie c : cookies) { if (c.getName().equals("SID")) { // And here it is! Okay, no need authenticate again. return true; } } return false; } @Override public DaemonTaskResult executeTask(Log log, DaemonTask task) { try { initialise(); ensureAuthenticated(log); ensureVersion(log); switch (task.getMethod()) { case Retrieve: // Request all torrents from server String path; if (version >= 40100) { path = "/api/v2/torrents/info"; } else if (version >= 30200) { path = "/query/torrents"; } else if (version >= 30000) { path = "/json/torrents"; } else { path = "/json/events"; } JSONArray allTorrentsResult = new JSONArray(makeRequest(log, path)); final List torrentsList = parseJsonTorrents(allTorrentsResult); JSONArray allLabelsResult = null; if (version >= 40100) { allLabelsResult = new JSONObject(makeRequest(log, "/api/v2/torrents/categories")).names(); } // either version < 40100 or no labels if (allLabelsResult == null) { allLabelsResult = new JSONArray(); } final List