From 11989ded0a677679f88735ce0e5ddb2b786eb6fb Mon Sep 17 00:00:00 2001
From: Frank Ploss <frank@fqxp.de>
Date: Sat, 19 Dec 2015 12:58:13 +0100
Subject: [PATCH] Add TorControlClient, only checking protocol version until
 now

---
 extension.js  | 117 +++++++++++++++++++++++++++++++++++++++++++++++---
 metadata.json |   5 +--
 2 files changed, 111 insertions(+), 11 deletions(-)

diff --git a/extension.js b/extension.js
index 48eb57b..0977ed6 100644
--- a/extension.js
+++ b/extension.js
@@ -1,23 +1,123 @@
+// ex: set sw=4
 'use strict';
 
+const Gio = imports.gi.Gio;
 const Lang = imports.lang;
 const St = imports.gi.St;
 const Main = imports.ui.main;
 const PanelMenu = imports.ui.panelMenu;
 
-const IndicatorName = "Tor";
-const DisabledIcon = 'my-caffeine-off-symbolic';
-const EnabledIcon = 'my-caffeine-on-symbolic';
 const TorIcon = 'tor';
 
 let torButton;
+let torControlClient;
+
+const TorControlClient = new Lang.Class({
+    Name: 'TorControlClient',
+
+    _init: function() {
+        this._connect();
+        this._updateProtocolInfo();
+        this._ensureProtocolCompatibility();
+    },
+
+    close: function() {
+        this._outputStream.close(null);
+        this._inputStream.close(null);
+    },
+
+    _connect: function() {
+        this._socketClient = new Gio.SocketClient();
+        this._connection = this._socketClient.connect_to_host('127.0.0.1:9051', null, null);
+        this._inputStream = new Gio.DataInputStream({base_stream: this._connection.get_input_stream()});
+        this._outputStream = new Gio.DataOutputStream({base_stream: this._connection.get_output_stream()});
+    },
+
+    _updateProtocolInfo: function() {
+        var reply = this._runCommand('PROTOCOLINFO');
+
+        if (reply.statusCode != 250) {
+            throw "Could not read protocol info";
+        }
+
+        var protocolInfoVersion;
+        var authMethods = [];
+        var authCookieFile;
+
+        for (let i = 0; i < reply.replyLines.length; i++) {
+            let tokens = reply.replyLines[i].split(' ');
+
+            switch (tokens[0]) {
+                case 'PROTOCOLINFO':
+                    protocolInfoVersion = tokens[1];
+                    break;
+                case 'AUTH':
+                    let methodsArg = tokens[1].split('=');
+                    authMethods = methodsArg[1].split(',');
+
+                    if (authMethods.indexOf('COOKIE') != -1 || authMethods.indexOf('SAFECOOKIE') != -1) {
+                        let cookieArg = tokens[2].split('=');
+                        authCookieFile = cookieArg[1];
+                    }
+                    break;
+            }
+        }
+
+        this._protocolInfo = {
+            protocolInfoVersion: protocolInfoVersion,
+            authMethods: authMethods,
+            authCookieFile: authCookieFile
+        }
+    },
+
+    _ensureProtocolCompatibility: function() {
+        if (this._protocolInfo.protocolInfoVersion != 1) {
+            throw 'Cannot handle tor protocol version ' + this._protocolInfo.protocolInfoVersion;
+        }
+    },
+
+    _runCommand: function(cmd) {
+        this._outputStream.put_string(cmd + '\n', null);
+        this._outputStream.flush(null);
+
+        var statusCode;
+        var replyLines = [];
+
+        do {
+            let line = this._readLine();
+            var reply = this._parseLine(line);
+
+            statusCode = reply.statusCode;
+            replyLines.push(reply.replyLine);
+        } while (reply.isMidReplyLine);
+
+        return {
+            statusCode: statusCode,
+            replyLines: replyLines
+        }
+    },
+
+    _readLine: function() {
+        return this._inputStream.read_line(null, null)[0].toString().trim();
+    },
+
+    _parseLine: function(line) {
+        return {
+            statusCode: parseInt(line.substr(0, 3)),
+            isMidReplyLine: (line[3] == '-'),
+            replyLine: line.substring(4)
+        }
+    }
+});
 
 const TorButton = new Lang.Class({
-    Name: IndicatorName,
+    Name: 'TorButton',
     Extends: PanelMenu.Button,
 
-    _init: function(metadata, params) {
-        this.parent(null, IndicatorName);
+    _init: function(torControlClient) {
+        this.parent(null, this.Name);
+
+        this._torControlClient = torControlClient;
 
         this._icon = new St.Icon({
             icon_name: TorIcon,
@@ -39,11 +139,14 @@ function init(extensionMeta) {
 }
 
 function enable() {
+    torControlClient = new TorControlClient();
     torButton = new TorButton();
-    Main.panel.addToStatusArea(IndicatorName, torButton);
+    Main.panel.addToStatusArea(torButton.Name, torButton);
 }
 
 function disable() {
     torButton.destroy();
     torButton = null;
+    torControlClient.close();
+    torControlClient = null;
 }
diff --git a/metadata.json b/metadata.json
index fe2a4e9..b19d1c8 100644
--- a/metadata.json
+++ b/metadata.json
@@ -2,10 +2,7 @@
   "description": "Change tor identity",
   "name": "Tor",
   "shell-version": [
-    "3.10",
-    "3.12",
-    "3.14",
-    "3.16"
+    "3.14"
   ],
   "url": "",
   "uuid": "change-tor-identity@fqxp.de",