src/govpn/cmd/govpn-client/proxy.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2015 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package main
  20
  21 import (
  22         "bufio"
  23         "encoding/base64"
  24         "io"
  25         "log"
  26         "net"
  27         "net/http"
  28 )
  29
  30 func proxyTCP() (io.Writer, chan []byte, chan struct{}) {
  31         proxyAddr, err := net.ResolveTCPAddr("tcp", *proxyAddr)
  32         if err != nil {
  33                 log.Fatalln("Can not resolve proxy address:", err)
  34         }
  35         conn, err := net.DialTCP("tcp", nil, proxyAddr)
  36         if err != nil {
  37                 log.Fatalln("Can not connect to proxy:", err)
  38         }
  39         req := "CONNECT " + *remoteAddr + " HTTP/1.1\n"
  40         req += "Host: " + *remoteAddr + "\n"
  41         if *proxyAuth != "" {
  42                 req += "Proxy-Authorization: Basic "
  43                 req += base64.StdEncoding.EncodeToString([]byte(*proxyAuth)) + "\n"
  44         }
  45         req += "\n"
  46         conn.Write([]byte(req))
  47         resp, err := http.ReadResponse(
  48                 bufio.NewReader(conn),
  49                 &http.Request{Method: "CONNECT"},
  50         )
  51         if err != nil || resp.StatusCode != http.StatusOK {
  52                 log.Fatalln("Unexpected response from proxy")
  53         }
  54         sink := make(chan []byte)
  55         ready := make(chan struct{})
  56         go handleTCP(conn, sink, ready)
  57         go func() { ready <- struct{}{} }()
  58         return TCPSender{conn}, sink, ready
  59 }