diff lisp/w3/socks.el @ 14:9ee227acff29 r19-15b90

Import from CVS: tag r19-15b90
author cvs
date Mon, 13 Aug 2007 08:48:42 +0200
parents
children 0293115a14e9
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/lisp/w3/socks.el	Mon Aug 13 08:48:42 2007 +0200
@@ -0,0 +1,384 @@
+;;; socks.el --- A Socks v5 Client for Emacs
+;; Author: wmperry
+;; Created: 1996/12/14 06:59:31
+;; Version: 1.2
+;; Keywords: comm, firewalls
+
+;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+;;; Copyright (c) 1996 by William M. Perry (wmperry@cs.indiana.edu)
+;;;
+;;; This file is not part of GNU Emacs, but the same permissions apply.
+;;;
+;;; GNU Emacs 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 2, or (at your option)
+;;; any later version.
+;;;
+;;; GNU Emacs 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 GNU Emacs; see the file COPYING.  If not, write to
+;;; the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
+;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+;;;
+;;; This is an implementation of the SOCKS v5 protocol as defined in
+;;; RFC 1928.
+;;;
+;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+(require 'cl)
+
+(defconst socks-version 5)
+(defvar socks-debug nil)
+
+;; Common socks v5 commands
+(defconst socks-connect-command 1)
+(defconst socks-bind-command 2)
+(defconst socks-udp-associate-command 3)
+
+;; Miscellaneous other socks constants
+(defconst socks-authentication-null 0)
+(defconst socks-authentication-failure 255)
+
+;; Response codes
+(defconst socks-response-success               0)
+(defconst socks-response-general-failure       1)
+(defconst socks-response-access-denied         2)
+(defconst socks-response-network-unreachable   3)
+(defconst socks-response-host-unreachable      4)
+(defconst socks-response-connection-refused    5)
+(defconst socks-response-ttl-expired           6)
+(defconst socks-response-cmd-not-supported     7)
+(defconst socks-response-address-not-supported 8)
+
+(defvar socks-errors
+  '("Succeeded"
+    "General SOCKS server failure"
+    "Connection not allowed by ruleset"
+    "Network unreachable"
+    "Host unreachable"
+    "Connection refused"
+    "Time-to-live expired"
+    "Command not supported"
+    "Address type not supported"))
+
+;; The socks v5 address types
+(defconst socks-address-type-v4   1)
+(defconst socks-address-type-name 3)
+(defconst socks-address-type-v6   4)
+
+;; Base variables
+(defvar socks-host (or (getenv "SOCKS5_SERVER") "socks"))
+(defvar socks-port (or (getenv "SOCKS5_PORT")   1080))
+(defvar socks-timeout 5)
+(defvar socks-connections (make-hash-table :size 13))
+
+;; Miscellaneous stuff for authentication
+(defvar socks-authentication-methods nil)
+(defvar socks-username (user-login-name))
+(defvar socks-password nil)
+
+(defun socks-register-authentication-method (id desc callback)
+  (let ((old (assq id socks-authentication-methods)))
+    (if old
+	(setcdr old (cons desc callback))
+      (setq socks-authentication-methods
+	    (cons (cons id (cons desc callback))
+		  socks-authentication-methods)))))
+
+(defun socks-unregister-authentication-method (id)
+  (let ((old (assq id socks-authentication-methods)))
+    (if old
+	(setq socks-authentication-methods
+	      (delq old socks-authentication-methods)))))
+
+(socks-register-authentication-method 0 "No authentication" 'identity)
+
+(defun socks-build-auth-list ()
+  (let ((num 0)
+	(retval ""))
+    (mapcar
+     (function
+      (lambda (x)
+	(if (fboundp (cdr (cdr x)))
+	    (setq retval (format "%s%c" retval (car x))
+		  num (1+ num)))))
+     socks-authentication-methods)
+    (format "%c%s" num retval)))
+
+(defconst socks-state-waiting-for-auth 0)
+(defconst socks-state-submethod-negotiation 1)
+(defconst socks-state-authenticated 2)
+(defconst socks-state-waiting 3)
+(defconst socks-state-connected 4)
+
+(defmacro socks-wait-for-state-change (proc htable cur-state)
+  (`
+   (while (and (= (cl-gethash 'state (, htable)) (, cur-state))
+	       (memq (process-status (, proc)) '(run open)))
+     (accept-process-output (, proc) socks-timeout))))
+
+(defun socks-filter (proc string)
+  (let ((info (cl-gethash proc socks-connections))
+	state desired-len)
+    (or info (error "socks-filter called on non-SOCKS connection %S" proc))
+    (setq state (cl-gethash 'state info))
+    (cond
+     ((= state socks-state-waiting-for-auth)
+      (cl-puthash 'scratch (concat string (cl-gethash 'scratch info)) info)
+      (setq string (cl-gethash 'scratch info))
+      (if (< (length string) 2)
+	  nil				; We need to spin some more
+	(cl-puthash 'authtype (aref string 1) info)
+	(cl-puthash 'scratch (substring string 2 nil) info)
+	(cl-puthash 'state socks-state-submethod-negotiation info)))
+     ((= state socks-state-submethod-negotiation)
+      )
+     ((= state socks-state-authenticated)
+      )
+     ((= state socks-state-waiting)
+      (cl-puthash 'scratch (concat string (cl-gethash 'scratch info)) info)
+      (setq string (cl-gethash 'scratch info))
+      (if (< (length string) 4)
+	  nil
+	(setq desired-len
+	      (+ 6			; Standard socks header
+		 (cond
+		  ((= (aref string 3) socks-address-type-v4) 4)
+		  ((= (aref string 3) socks-address-type-v6) 16)
+		  ((= (aref string 3) socks-address-type-name)
+		   (if (< (length string) 5)
+		       255
+		     (+ 1 (aref string 4)))))))
+	(if (< (length string) desired-len)
+	    nil				; Need to spin some more
+	  (cl-puthash 'state socks-state-connected info)
+	  (cl-puthash 'reply (aref string 1) info)
+	  (cl-puthash 'response string info))))
+     ((= state socks-state-connected)
+      )
+     )
+    )
+  )
+
+(defun socks-open-connection (&optional host port)
+  (interactive)
+  (setq host (or host socks-host)
+	port (or port socks-port))
+  (save-excursion
+    (let ((proc (socks-original-open-network-stream "socks"
+						    nil
+						    host port))
+	  (info (make-hash-table :size 13))
+	  (authtype nil))
+
+      ;; Initialize process and info about the process
+      (set-process-filter proc 'socks-filter)
+      (process-kill-without-query proc)
+      (cl-puthash proc info socks-connections)
+      (cl-puthash 'state socks-state-waiting-for-auth info)
+      (cl-puthash 'authtype socks-authentication-failure info)
+
+      ;; Send what we think we can handle for authentication types
+      (process-send-string proc (format "%c%s" socks-version
+					(socks-build-auth-list)))
+
+      ;; Basically just do a select() until we change states.
+      (socks-wait-for-state-change proc info socks-state-waiting-for-auth)
+      (setq authtype (cl-gethash 'authtype info))
+      (cond
+       ((= authtype socks-authentication-null)
+	(and socks-debug (message "No authentication necessary")))
+       ((= authtype socks-authentication-failure)
+	(error "No acceptable authentication methods found."))
+       (t
+	(let* ((auth-type (char-int (cl-gethash 'authtype info)))
+	       (auth-handler (assoc auth-type socks-authentication-methods))
+	       (auth-func (and auth-handler (cdr (cdr auth-handler))))
+	       (auth-desc (and auth-handler (car (cdr auth-handler)))))
+	  (set-process-filter proc nil)
+	  (if (and auth-func (fboundp auth-func)
+		   (funcall auth-func proc))
+	      (message "Successfully authenticated using: %s" auth-desc)
+	    (delete-process proc)
+	    (error "Failed to use auth method: %s (%d)"
+		   (or auth-desc "Unknown") auth-type))
+	  )
+	)
+       )
+      (cl-puthash 'state socks-state-authenticated info)
+      (set-process-filter proc 'socks-filter)
+      proc)))
+
+(defun socks-send-command (proc command atype address port)
+  (let ((addr (case atype
+		(socks-address-type-v4 address)
+		(socks-address-type-v6 address)
+		(t
+		 (format "%c%s" (length address) address))))
+	(info (cl-gethash proc socks-connections)))
+    (or info (error "socks-send-command called on non-SOCKS connection %S"
+		    proc))
+    (cl-puthash 'state socks-state-waiting info)
+    (process-send-string proc
+			 (format 
+			  "%c%c%c%c%s%c%c"
+			  socks-version	; version 
+			  command	; command
+			  0		; reserved
+			  atype		; address type
+			  addr		; address
+			  (lsh port -8)	; port, high byte
+			  (- port (lsh (lsh port -8) 8)) ; port, low byte
+			  ))
+    (socks-wait-for-state-change proc info socks-state-waiting)
+    (if (= (cl-gethash 'reply info) socks-response-success)
+	nil				; Sweet sweet success!
+      (delete-process proc)
+      (error "%s" (nth (cl-gethash 'reply info) socks-errors)))
+    proc))
+
+
+;; Replacement functions for open-network-stream, etc.
+(defvar socks-noproxy nil
+  "*List of regexps matching hosts that we should not socksify connections to")
+
+(defun socks-find-route (host service)
+  (let ((route (cons socks-host socks-port))
+	(noproxy socks-noproxy))
+    (while noproxy
+      (if (string-match (car noproxy) host)
+	  (setq route nil
+		noproxy nil))
+      (setq noproxy (cdr noproxy)))
+    route))
+
+(if (fboundp 'socks-original-open-network-stream)
+    nil					; Do nothing, we've been here already
+  (fset 'socks-original-open-network-stream
+	(symbol-function 'open-network-stream))
+  (fset 'open-network-stream 'socks-open-network-stream))
+
+(defvar socks-services-file "/etc/services")
+(defvar socks-tcp-services (make-hash-table :size 13 :test 'equal))
+(defvar socks-udp-services (make-hash-table :size 13 :test 'equal))
+
+(defun socks-parse-services ()
+  (if (not (and (file-exists-p socks-services-file)
+		(file-readable-p socks-services-file)))
+      (error "Could not find services file: %s" socks-services-file))
+  (save-excursion
+    (clrhash socks-tcp-services)
+    (clrhash socks-udp-services)
+    (set-buffer (get-buffer-create " *socks-tmp*"))
+    (erase-buffer)
+    (insert-file-contents socks-services-file)
+    ;; Nuke comments
+    (goto-char (point-min))
+    (while (re-search-forward "#.*" nil t)
+      (replace-match ""))
+    ;; Nuke empty lines
+    (goto-char (point-min))
+    (while (re-search-forward "^[ \t\n]+" nil t)
+      (replace-match ""))
+    ;; Now find all the lines
+    (goto-char (point-min))
+    (let (name port type)
+      (while (re-search-forward "^\\([^ \t]+\\)[ \t]+\\([0-9]+\\)/\\([a-z]+\\)"
+				nil t)
+	(setq name (downcase (match-string 1))
+	      port (string-to-int (match-string 2))
+	      type (downcase (match-string 3)))
+	(cl-puthash name port (if (equal type "udp")
+			       socks-udp-services
+			     socks-tcp-services))))))
+
+(defun socks-find-services-entry (service &optional udp)
+  "Return the port # associated with SERVICE"
+  (if (= (hash-table-count socks-tcp-services) 0)
+      (socks-parse-services))
+  (cl-gethash (downcase service)
+	      (if udp socks-udp-services socks-tcp-services)))
+
+(defun socks-open-network-stream (name buffer host service)
+  (let* ((route (socks-find-route host service))
+	 proc info)
+    (if (not route)
+	(socks-original-open-network-stream name buffer host service)
+      (setq proc (socks-open-connection (car route) (cdr route))
+	    info (cl-gethash proc socks-connections))
+      (socks-send-command proc socks-connect-command
+			  socks-address-type-name
+			  host
+			  (if (stringp service)
+			      (socks-find-services-entry service)
+			    service))
+      (cl-puthash 'buffer buffer info)
+      (cl-puthash 'host host info)
+      (cl-puthash 'service host info)
+      (set-process-filter proc nil)
+      (set-process-buffer proc (if buffer (get-buffer-create buffer)))
+      proc)))
+
+;; Authentication modules go here
+
+;; Basic username/password authentication, ala RFC 1929
+;; To enable username/password authentication, uncomment the following
+;; lines:
+;;
+;; (socks-register-authentication-method 2 "Username/Password"
+;;			      'socks-username/password-auth)
+
+(defconst socks-username/password-auth-version 1)
+
+(if (not (fboundp 'char-int))
+    (fset 'char-int 'identity))
+
+(defun socks-username/password-auth-filter (proc str)
+  (let ((info (cl-gethash proc socks-connections))
+	state desired-len)
+    (or info (error "socks-filter called on non-SOCKS connection %S" proc))
+    (setq state (cl-gethash 'state info))
+    (cl-puthash 'scratch (concat (cl-gethash 'scratch info) str) info)
+    (if (< (length (cl-gethash 'scratch info)) 2)
+	nil
+      (cl-puthash 'password-auth-status (char-int
+					 (aref (cl-gethash 'scratch info) 1))
+		  info)
+      (cl-puthash 'state socks-state-authenticated info))))
+
+(defun socks-username/password-auth (proc)
+  (if (not socks-password)
+      (setq socks-password (read-passwd
+			    (format "Password for %s@%s: "
+				    socks-username socks-host))))
+  (let* ((info (cl-gethash proc socks-connections))
+	 (state (cl-gethash 'state info)))
+    (cl-puthash 'scratch "" info)
+    (set-process-filter proc 'socks-username/password-auth-filter)
+    (process-send-string proc
+			 (format "%c%c%s%c%s"
+				 socks-username/password-auth-version
+				 (length socks-username)
+				 socks-username
+				 (length socks-password)
+				 socks-password))
+    (socks-wait-for-state-change proc info state)
+    (= (cl-gethash 'password-auth-status info) 0)))
+
+
+;; More advanced GSS/API stuff, not yet implemented - volunteers?
+;; (socks-register-authentication-method 1 "GSS/API" 'socks-gssapi-auth)
+
+(defun socks-gssapi-auth (proc)
+  nil)
+
+
+;; CHAP stuff
+;; (socks-register-authentication-method 3 "CHAP" 'socks-chap-auth)
+(defun socks-chap-auth (proc)
+  nil)
+
+(provide 'socks)