| Copyright | (c) Lars Petersen 2015 |
|---|---|
| License | MIT |
| Maintainer | info@lars-petersen.net |
| Stability | experimental |
| Safe Haskell | None |
| Language | Haskell2010 |
System.Socket
Contents
Description
{-# LANGUAGE OverloadedStrings #-}
module Main where
import Control.Exception ( bracket, catch )
import Control.Monad ( forever )
import System.Socket
import System.Socket.Family.Inet6
import System.Socket.Type.Stream
import System.Socket.Protocol.TCP
main :: IO ()
main = bracket
( socket :: IO (Socket Inet6 Stream TCP) )
( \s-> do
close s
putStrLn "Listening socket closed."
)
( \s-> do
setSocketOption s (ReuseAddress True)
setSocketOption s (V6Only False)
bind s (SocketAddressInet6 inet6Any 8080 0 0)
listen s 5
putStrLn "Listening socket ready..."
forever $ acceptAndHandle s `catch` \e-> print (e :: SocketException)
)
acceptAndHandle :: Socket Inet6 Stream TCP -> IO ()
acceptAndHandle s = bracket
( accept s )
( \(p, addr)-> do
close p
putStrLn $ "Closed connection to " ++ show addr
)
( \(p, addr)-> do
putStrLn $ "Accepted connection from " ++ show addr
sendAll p "Hello world!" msgNoSignal
)- data Socket f t p
- class Storable (SocketAddress f) => Family f where
- data SocketAddress f
- class Type t where
- class Protocol p where
- socket :: (Family f, Type t, Protocol p) => IO (Socket f t p)
- connect :: Family f => Socket f t p -> SocketAddress f -> IO ()
- bind :: Family f => Socket f t p -> SocketAddress f -> IO ()
- listen :: Socket f t p -> Int -> IO ()
- accept :: Family f => Socket f t p -> IO (Socket f t p, SocketAddress f)
- send :: Socket f t p -> ByteString -> MessageFlags -> IO Int
- sendTo :: Family f => Socket f t p -> ByteString -> MessageFlags -> SocketAddress f -> IO Int
- receive :: Socket f t p -> Int -> MessageFlags -> IO ByteString
- receiveFrom :: Family f => Socket f t p -> Int -> MessageFlags -> IO (ByteString, SocketAddress f)
- close :: Socket f t p -> IO ()
- class SocketOption o where
- data Error = Error SocketException
- data ReuseAddress = ReuseAddress Bool
- data KeepAlive = KeepAlive Bool
- data AddressInfo f t p = AddressInfo {}
- class Family f => HasAddressInfo f where
- data NameInfo = NameInfo {}
- class Family f => HasNameInfo f where
- newtype MessageFlags = MessageFlags CInt
- msgEndOfRecord :: MessageFlags
- msgNoSignal :: MessageFlags
- msgOutOfBand :: MessageFlags
- msgWaitAll :: MessageFlags
- data AddressInfoFlags
- aiAddressConfig :: AddressInfoFlags
- aiAll :: AddressInfoFlags
- aiCanonicalName :: AddressInfoFlags
- aiNumericHost :: AddressInfoFlags
- aiNumericService :: AddressInfoFlags
- aiPassive :: AddressInfoFlags
- aiV4Mapped :: AddressInfoFlags
- data NameInfoFlags
- niNameRequired :: NameInfoFlags
- niDatagram :: NameInfoFlags
- niNoFullyQualifiedDomainName :: NameInfoFlags
- niNumericHost :: NameInfoFlags
- niNumericService :: NameInfoFlags
- newtype SocketException = SocketException CInt
- eOk :: SocketException
- eInterrupted :: SocketException
- eBadFileDescriptor :: SocketException
- eInvalid :: SocketException
- ePipe :: SocketException
- eWouldBlock :: SocketException
- eAgain :: SocketException
- eNotSocket :: SocketException
- eDestinationAddressRequired :: SocketException
- eMessageSize :: SocketException
- eProtocolType :: SocketException
- eNoProtocolOption :: SocketException
- eProtocolNotSupported :: SocketException
- eSocketTypeNotSupported :: SocketException
- eOperationNotSupported :: SocketException
- eProtocolFamilyNotSupported :: SocketException
- eAddressFamilyNotSupported :: SocketException
- eAddressInUse :: SocketException
- eAddressNotAvailable :: SocketException
- eNetworkDown :: SocketException
- eNetworkUnreachable :: SocketException
- eNetworkReset :: SocketException
- eConnectionAborted :: SocketException
- eConnectionReset :: SocketException
- eNoBufferSpace :: SocketException
- eIsConnected :: SocketException
- eNotConnected :: SocketException
- eShutdown :: SocketException
- eTooManyReferences :: SocketException
- eTimedOut :: SocketException
- eConnectionRefused :: SocketException
- eHostDown :: SocketException
- eHostUnreachable :: SocketException
- eAlready :: SocketException
- eInProgress :: SocketException
- newtype AddressInfoException = AddressInfoException CInt
- eaiAgain :: AddressInfoException
- eaiBadFlags :: AddressInfoException
- eaiFail :: AddressInfoException
- eaiFamily :: AddressInfoException
- eaiMemory :: AddressInfoException
- eaiNoName :: AddressInfoException
- eaiSocketType :: AddressInfoException
- eaiService :: AddressInfoException
- eaiSystem :: AddressInfoException
Socket
A generic socket type. Use socket to create a new socket.
The socket is just an MVar-wrapped file descriptor.
The Socket constructor is exported trough the unsafe
module in order to make this library easily extensible, but it is usually
not necessary nor advised to work directly on the file descriptor.
If you do, the following rules must be obeyed:
- Make sure not to deadlock. Use
withMVaror similar. - The lock must not be held during a blocking call. This would make it impossible to send and receive simultaneously or to close the socket.
- The lock must be held when calling operations that use the file descriptor. Otherwise the socket might get closed or even reused by another thread/capability which might result in reading from or writing on a totally different socket. This is a security nightmare!
- The socket is non-blocking and all the code relies on that assumption.
You need to use GHC's eventing mechanism primitives to block until
something happens. The former rules forbid to use
threadWaitReadas it does not separate between registering the file descriptor (for which the lock must be held) and the actual waiting (for which you must not hold the lock). Also see this thread and read the library code to see how the problem is currently circumvented.
Family
class Storable (SocketAddress f) => Family f where #
The address Family determines the network protocol to use.
The most common address families are Inet (IPv4)
and Inet6 (IPv6).
Minimal complete definition
Associated Types
data SocketAddress f #
The SocketAddress type is a data family.
This allows to provide different data constructors depending on the socket
family without knowing all of them in advance or the need to extend this
core library.
SocketAddressInet inetLoopback 8080 :: SocketAddress Inet SocketAddressInet6 inet6Loopback 8080 0 0 :: SocketAddress Inet6
Methods
familyNumber :: f -> CInt #
The number designating this Family on the specific platform. This
method is only exported for implementing extension libraries.
This function shall yield the values of constants like AF_INET, AF_INET6 etc.
Type
The Type determines properties of the transport layer and the semantics
of basic socket operations.
The instances supplied by this library are Raw
(no transport layer), Stream
(for unframed binary streams, e.g. TCP),
Datagram (for datagrams
of limited length, e.g. UDP) and
SequentialPacket (for framed messages of arbitrary
length, e.g. SCTP).
Minimal complete definition
Methods
typeNumber :: t -> CInt #
This number designates this Type on the specific platform. This
method is only exported for implementing extension libraries.
The function shall yield the values of constants like SOCK_STREAM,
SOCK_DGRAM etc.
Protocol
The Protocol determines the transport protocol to use.
Use Default to let the operating system choose
a transport protocol compatible with the socket's Type.
Minimal complete definition
Methods
protocolNumber :: p -> CInt #
This number designates this Protocol on the specific platform. This
method is only exported for implementing extension libraries.
The function shall yield the values of constants like IPPROTO_TCP,
IPPROTO_UDP etc.
Operations
socket
socket :: (Family f, Type t, Protocol p) => IO (Socket f t p) #
Creates a new socket.
Whereas the underlying POSIX socket operation takes 3 parameters, this library encodes this information in the type variables. This rules out several kinds of errors and especially simplifies the handling of addresses (by using associated data families). Examples:
-- create an IPv4-UDP-datagram socket sock <- socket :: IO (Socket Inet Datagram UDP) -- create an IPv6-TCP-streaming socket sock6 <- socket :: IO (Socket Inet6 Stream TCP) -- create an IPv6-streaming socket with default protocol (usually TCP) sock6 <- socket :: IO (Socket Inet6 Strem Default)
- This operation sets up a finalizer that automatically closes the socket
when the garbage collection decides to collect it. This is just a
fail-safe. You might still run out of file descriptors as there's
no guarantee about when the finalizer is run. You're advised to
manually
closethe socket when it's no longer needed. If possible, usebracketto reliably close the socket descriptor on exception or regular termination of your computation:
result <- bracket (socket :: IO (Socket Inet6 Stream TCP)) close $ \sock-> do somethingWith sock -- your computation here return somethingelse
- This operation configures the socket non-blocking to work seamlessly with the runtime system's event notification mechanism.
- This operation can safely deal with asynchronous exceptions without leaking file descriptors.
- This operation throws
SocketExceptions. Consult yourman socketfor details and specific errors.
connect
connect :: Family f => Socket f t p -> SocketAddress f -> IO () #
Connects to a remote address.
- This operation returns as soon as a connection has been established (as if the socket were blocking). The connection attempt has either failed or succeeded after this operation threw an exception or returned.
- The operation throws
SocketExceptions. Callingconnecton aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned.
bind
bind :: Family f => Socket f t p -> SocketAddress f -> IO () #
Bind a socket to an address.
- Calling
bindon aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned. - It is assumed that
bindnever blocks and thereforeeInProgress,eAlreadyandeInterrupteddon't occur. This assumption is supported by the fact that the Linux manpage doesn't mention any of these errors, the Posix manpage doesn't mention the last one and even MacOS' implementation will never fail with any of these when the socket is configured non-blocking as argued here. - This operation throws
SocketExceptions. Consult yourmanpage for details and specificerrnos.
listen
listen :: Socket f t p -> Int -> IO () #
Starts listening and queueing connection requests on a connection-mode socket. The second parameter determines the backlog size.
- Calling
listenon aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned. - The second parameter is called backlog and sets a limit on how many
unaccepted connections the transport implementation shall queue. A value
of
0leaves the decision to the implementation. - This operation throws
SocketExceptions. Consult yourman listenfor details and specific errors.
accept
accept :: Family f => Socket f t p -> IO (Socket f t p, SocketAddress f) #
Accept a new connection.
- Calling
accepton aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned. - This operation configures the new socket non-blocking. It uses
accept4(when available) in order to accept and set the socket non-blocking with a single system call. - This operation sets up a finalizer for the new socket that automatically
closes the new socket when the garbage collection decides to collect it.
This is just a fail-safe. You might still run out of file descriptors as
there's no guarantee about when the finalizer is run. You're advised to
manually
closethe socket when it's no longer needed. - This operation throws
SocketExceptions. - This operation catches
eAgain,eWouldBlockandeInterruptedinternally and retries automatically.
send, sendTo
send :: Socket f t p -> ByteString -> MessageFlags -> IO Int #
Send data.
- Calling
sendon aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned. - The operation returns the number of bytes sent. On
DatagramandSequentialPacketsockets certain assurances on atomicity exist andeAgainoreWouldBlockare thrown until the whole message would fit into the send buffer. - This operation throws
SocketExceptions. Consultman sendfor details and specific errors. eAgain,eWouldBlockandeInterruptedand handled internally and won't be thrown. For performance reasons the operation first tries a write on the socket and then waits when it goteAgainoreWouldBlock.
sendTo :: Family f => Socket f t p -> ByteString -> MessageFlags -> SocketAddress f -> IO Int #
Like send, but allows to specify a destination address.
receive, receiveFrom
receive :: Socket f t p -> Int -> MessageFlags -> IO ByteString #
Receive data.
- The operation takes a buffer size in bytes a first parameter which
limits the maximum length of the returned
ByteString. - When an empty
ByteStringis returned this usally (protocol specific) means that the peer gracefully closed the connection. The user is advised to check for and handle this case. - Calling
receiveon aclosed socket throwseBadFileDescriptoreven if the former file descriptor has been reassigned. - This operation throws
SocketExceptions. Consultman recvfor details and specific errors. eAgain,eWouldBlockandeInterruptedand handled internally and won't be thrown. For performance reasons the operation first tries a read on the socket and then waits when it goteAgainoreWouldBlockuntil the socket is signaled to be readable.
receiveFrom :: Family f => Socket f t p -> Int -> MessageFlags -> IO (ByteString, SocketAddress f) #
Like receive, but additionally yields the peer address.
close
close :: Socket f t p -> IO () #
Closes a socket.
- This operation is idempotent and thus can be performed more than once without throwing an exception. If it throws an exception it is presumably a not recoverable situation and the process should exit.
- This operation does not block.
- This operation wakes up all threads that are currently blocking on this
socket. All other threads are guaranteed to not block on operations on this socket in the future.
Threads that perform operations other than
closeon this socket will fail witheBadFileDescriptorafter the socket has been closed (closereplaces theFdin theMVarwith-1to reliably avoid use-after-free situations). - This operation potentially throws
SocketExceptions (onlyEIOis documented).eInterruptedis catched internally and retried automatically, so won't be thrown.
Options
class SocketOption o where #
SocketOptions allow to read and write certain properties of a socket.
- Each option shall have a corresponding data type that models the data associated with the socket option.
- Use
unsafeGetSocketOptionandunsafeSetSocketOptionin order to implement custom socket options.
Minimal complete definition
Methods
getSocketOption :: Socket f t p -> IO o #
Get a specific SocketOption.
- This operation throws
SocketExceptions. Consultman getsockoptfor details and specific errors.
setSocketOption :: Socket f t p -> o -> IO () #
Set a specific SocketOption.
- This operation throws
SocketExceptions. Consultman setsockoptfor details and specific errors.
Error
Reports the last error that occured on the socket.
- Also known as
SO_ERROR. - The operation
setSocketOptionalways throwseInvalidfor this option. - Use with care in the presence of concurrency!
Constructors
| Error SocketException |
ReuseAddress
data ReuseAddress #
Allows or disallows the reuse of a local address in a bind call.
- Also known as
SO_REUSEADDR. - This is particularly useful when experiencing
eAddressInUseexceptions.
Constructors
| ReuseAddress Bool |
Instances
KeepAlive
When enabled the protocol checks in a protocol-specific manner if the other end is still alive.
- Also known as
SO_KEEPALIVE.
Name Resolution
getAddressInfo
data AddressInfo f t p #
Constructors
| AddressInfo | |
Fields | |
Instances
| Eq (SocketAddress f) => Eq (AddressInfo f t p) # | |
| Show (SocketAddress f) => Show (AddressInfo f t p) # | |
class Family f => HasAddressInfo f where #
This class is for address families that support name resolution.
Minimal complete definition
Methods
getAddressInfo :: (Type t, Protocol p) => Maybe ByteString -> Maybe ByteString -> AddressInfoFlags -> IO [AddressInfo f t p] #
Maps names to addresses (i.e. by DNS lookup).
The operation throws AddressInfoExceptions.
Contrary to the underlying getaddrinfo operation this wrapper is
typesafe and thus only returns records that match the address, type
and protocol encoded in the type. This is the price we have to pay
for typesafe sockets and extensibility.
If you need different types of records, you need to start several
queries. If you want to connect to both IPv4 and IPV6 addresses use
aiV4Mapped and use IPv6-sockets.
getAddressInfo (Just "www.haskell.org") (Just "https") mempty :: IO [AddressInfo Inet Stream TCP]
> [AddressInfo {addressInfoFlags = AddressInfoFlags 0, socketAddress = SocketAddressInet {inetAddress = InetAddress 162.242.239.16, inetPort = InetPort 443}, canonicalName = Nothing}]> getAddressInfo (Just "www.haskell.org") (Just "80") aiV4Mapped :: IO [AddressInfo Inet6 Stream TCP]
[AddressInfo {
addressInfoFlags = AddressInfoFlags 8,
socketAddress = SocketAddressInet6 {inet6Address = Inet6Address 2400:cb00:2048:0001:0000:0000:6ca2:cc3c, inet6Port = Inet6Port 80, inet6FlowInfo = Inet6FlowInfo 0, inet6ScopeId = Inet6ScopeId 0},
canonicalName = Nothing }]> getAddressInfo (Just "darcs.haskell.org") Nothing aiV4Mapped :: IO [AddressInfo Inet6 Stream TCP]
[AddressInfo {
addressInfoFlags = AddressInfoFlags 8,
socketAddress = SocketAddressInet6 {inet6Address = Inet6Address 0000:0000:0000:0000:0000:ffff:17fd:e1ad, inet6Port = Inet6Port 0, inet6FlowInfo = Inet6FlowInfo 0, inet6ScopeId = Inet6ScopeId 0},
canonicalName = Nothing }]
> getAddressInfo (Just "darcs.haskell.org") Nothing mempty :: IO [AddressInfo Inet6 Stream TCP]
*** Exception: AddressInfoException "Name or service not known"Instances
getNameInfo
class Family f => HasNameInfo f where #
This class is for address families that support reverse name resolution.
Minimal complete definition
Methods
getNameInfo :: SocketAddress f -> NameInfoFlags -> IO NameInfo #
(Reverse-)map an address back to a human-readable host- and service name.
The operation throws AddressInfoExceptions.
> getNameInfo (SocketAddressInet inetLoopback 80) mempty
NameInfo {hostName = "localhost.localdomain", serviceName = "http"}Instances
Flags
MessageFlags
newtype MessageFlags #
Use the Monoid instance to combine several flags:
mconcat [msgNoSignal, msgWaitAll]
Use the Bits instance to check whether a flag is set:
if flags .&. msgEndOfRecord /= mempty then ...
Constructors
| MessageFlags CInt |
Instances
msgEndOfRecord :: MessageFlags #
MSG_EOR
MSG_NOSIGNAL
Suppresses the generation of PIPE signals when writing to a socket
that is no longer connected.
Although this flag is POSIX, it is not available on all platforms. Try
msgNoSignal /= mempty
in order to check whether this flag is defined on a certain platform. It is safe to just use this constant even if it might not have effect on a certain target platform. The platform independence of this flag is therefore fulfilled to some extent.
Some more explanation on the platform specific behaviour:
- Linux defines and supports
MSG_NOSIGNALand properly suppresses the generation of broken pipe-related signals. - Windows does not define it, but does not generate signals either.
- OSX does not define it, but generates
PIPEsignals. The GHC runtime ignores them if you don't hook them explicitly. The non-portable socket optionSO_NOSIGPIPEmay be used disable signals on a per-socket basis.
msgOutOfBand :: MessageFlags #
MSG_OOB
MSG_WAITALL
AddressInfoFlags
data AddressInfoFlags #
Use the Monoid instance to combine several flags:
mconcat [aiAddressConfig, aiV4Mapped]
Instances
aiAddressConfig :: AddressInfoFlags #
AI_ADDRCONFIG:
AI_ALL: Return both IPv4 (as v4-mapped IPv6 address) and IPv6 addresses
when aiV4Mapped is set independent of whether IPv6 addresses exist for
this name.
aiCanonicalName :: AddressInfoFlags #
AI_CANONNAME:
aiNumericHost :: AddressInfoFlags #
AI_NUMERICHOST:
aiNumericService :: AddressInfoFlags #
AI_NUMERICSERV:
aiPassive :: AddressInfoFlags #
AI_PASSIVE:
aiV4Mapped :: AddressInfoFlags #
AI_V4MAPPED: Return mapped IPv4 addresses if no IPv6 addresses could be found
or if aiAll flag is set.
NameInfoFlags
data NameInfoFlags #
Use the Monoid instance to combine several flags:
mconcat [niNameRequired, niNoFullyQualifiedDomainName]
Instances
niNameRequired :: NameInfoFlags #
NI_NAMEREQD: Throw an exception if the hostname cannot be determined.
niNoFullyQualifiedDomainName :: NameInfoFlags #
NI_NOFQDN: Return only the hostname part of the fully qualified domain name for local hosts.
niNumericHost :: NameInfoFlags #
NI_NUMERICHOST: Return the numeric form of the host address.
niNumericService :: NameInfoFlags #
NI_NUMERICSERV: Return the numeric form of the service address.
Exceptions
SocketException
newtype SocketException #
Contains the error code that can be matched against.
Hint: Use guards or MultiWayIf to match against specific exceptions:
if | e == eAddressInUse -> ... | e == eAddressNotAvailable -> ... | otherwise -> ...
Constructors
| SocketException CInt |
Instances
eOk :: SocketException #
SocketException "No error"
eInterrupted :: SocketException #
SocketException "Interrupted system call"
NOTE: This exception shall not be thrown by any public operation in this library, but is handled internally.
eBadFileDescriptor :: SocketException #
SocketException "Bad file descriptor"
SocketException "Invalid argument"
SocketException "Broken pipe"
eWouldBlock :: SocketException #
SocketException "Resource temporarily unavailable"
NOTE: This exception shall not be thrown by any public operation in this library, but is handled internally.
SocketException "Resource temporarily unavailable"
eNotSocket :: SocketException #
SocketException "Socket operation on non-socket"
NOTE: This should be ruled out by the type system.
eDestinationAddressRequired :: SocketException #
SocketException "Destination address required"
eMessageSize :: SocketException #
SocketException "Message too long"
eProtocolType :: SocketException #
SocketException "Protocol wrong type for socket"
eNoProtocolOption :: SocketException #
SocketException "Protocol not available"
eProtocolNotSupported :: SocketException #
SocketException "Protocol not supported"
eSocketTypeNotSupported :: SocketException #
SocketException "Socket type not supported"
eOperationNotSupported :: SocketException #
SocketException "Operation not supported"
eProtocolFamilyNotSupported :: SocketException #
SocketException "Protocol family not supported"
eAddressFamilyNotSupported :: SocketException #
SocketException "Address family not supported by protocol"
eAddressInUse :: SocketException #
SocketException "Address already in use"
eAddressNotAvailable :: SocketException #
SocketException "Cannot assign requested address"
eNetworkDown :: SocketException #
SocketException "Network is down"
eNetworkUnreachable :: SocketException #
SocketException "Network is unreachable"
eNetworkReset :: SocketException #
SocketException "Network dropped connection on reset"
eConnectionAborted :: SocketException #
SocketException "Software caused connection abort"
eConnectionReset :: SocketException #
SocketException "Connection reset by peer"
eNoBufferSpace :: SocketException #
SocketException "No buffer space available"
eIsConnected :: SocketException #
SocketException "Transport endpoint is already connected"
eNotConnected :: SocketException #
SocketException "Transport endpoint is not connected"
eShutdown :: SocketException #
SocketException "Cannot send after transport endpoint shutdown"
eTooManyReferences :: SocketException #
SocketException "Too many references: cannot splice"
eTimedOut :: SocketException #
SocketException "Connection timed out"
eConnectionRefused :: SocketException #
SocketException "Connection refused"
eHostDown :: SocketException #
SocketException "Host is down"
eHostUnreachable :: SocketException #
SocketException "No route to host"
SocketException "Operation already in progress"
NOTE: This exception shall not be thrown by any public operation in this library, but is handled internally.
eInProgress :: SocketException #
SocketException "Operation now in progress"
AddressInfoException
newtype AddressInfoException #
Contains the error code that can be matched against.
Hint: Use guards or MultiWayIf to match against specific exceptions:
if | e == eaiFail -> ... | e == eaiNoName -> ... | otherwise -> ...
Constructors
| AddressInfoException CInt |
eaiAgain :: AddressInfoException #
AddressInfoException "Temporary failure in name resolution"
eaiBadFlags :: AddressInfoException #
AddressInfoException "Bad value for ai_flags"
eaiFail :: AddressInfoException #
AddressInfoException "Non-recoverable failure in name resolution"
eaiFamily :: AddressInfoException #
AddressInfoException "ai_family not supported"
eaiMemory :: AddressInfoException #
AddressInfoException "Memory allocation failure"
eaiNoName :: AddressInfoException #
AddressInfoException "No such host is known"
eaiSocketType :: AddressInfoException #
AddressInfoException "ai_socktype not supported"
eaiService :: AddressInfoException #
AddressInfoException "Servname not supported for ai_socktype"
eaiSystem :: AddressInfoException #
AddressInfoException "System error"