echo.hs

writeAll :: Socket -> String -> IO ()
writeAll socket s = do
  sent <- send socket s
  case drop sent s of
    "" -> return ()
    s -> writeAll socket s

echo :: Socket -> IO ()
echo client = do
  input <- recv client 4096
  if null input
    then close client
    else writeAll client input >> echo client