module Network.Socks5
( SocksConf(..)
, defaultSocksConf
, socksConnectAddr
, socksConnectName
, socksConnectTo
, socksConnectWith
) where
import Control.Monad
import Control.Exception
import Network.Socket
import Network.BSD
import Network.Socks5.Command
import Network.Socks5.Types
import Network
import System.IO
data SocksConf = SocksConf
{ socksHost :: String
, socksPort :: PortNumber
, socksVersion :: Int
}
defaultSocksConf host port = SocksConf host port 5
withSocks sock sockaddr f = do
connect sock sockaddr
r <- socks5Establish sock [SocksMethodNone]
when (r == SocksMethodNotAcceptable) $ error "cannot connect with no socks method of authentication"
f
socksConnectAddr :: Socket -> SockAddr -> SockAddr -> IO ()
socksConnectAddr sock sockserver destaddr = withSocks sock sockserver $ do
case destaddr of
SockAddrInet p h -> socks5ConnectIPV4 sock h p >> return ()
SockAddrInet6 p _ h _ -> socks5ConnectIPV6 sock h p >> return ()
_ -> error "unsupported unix sockaddr type"
socksConnectName :: Socket -> SockAddr -> String -> PortNumber -> IO ()
socksConnectName sock sockserver destination port = withSocks sock sockserver $ do
_ <- socks5ConnectDomainName sock destination port
return ()
socksConnectWith :: SocksConf
-> String
-> PortID
-> IO Socket
socksConnectWith sockConf desthost destport = do
dport <- resolvePortID destport
proto <- getProtocolNumber "tcp"
bracketOnError (socket AF_INET Stream proto) sClose $ \sock -> do
he <- getHostByName $ socksHost sockConf
let sockaddr = SockAddrInet (socksPort sockConf) (hostAddress he)
socksConnectName sock sockaddr desthost dport
return sock
socksConnectTo :: String -> PortID -> String -> PortID -> IO Handle
socksConnectTo sockshost socksport desthost destport = do
sport <- resolvePortID socksport
let socksConf = defaultSocksConf sockshost sport
sock <- socksConnectWith socksConf desthost destport
socketToHandle sock ReadWriteMode
resolvePortID (Service serv) = getServicePortNumber serv
resolvePortID (PortNumber n) = return n
resolvePortID _ = error "unsupported unix PortID"