mirror of
https://github.com/nim-lang/Nim.git
synced 2025-12-29 17:34:43 +00:00
95 lines
2.4 KiB
Nim
Executable File
95 lines
2.4 KiB
Nim
Executable File
#
|
|
#
|
|
# Nimrod's Runtime Library
|
|
# (c) Copyright 2012 Dominik Picheta
|
|
#
|
|
# See the file "copying.txt", included in this
|
|
# distribution, for details about the copyright.
|
|
#
|
|
|
|
## This module provides an easy to use sockets-style
|
|
## nimrod interface to the OpenSSL library.
|
|
|
|
import openssl, strutils, os
|
|
|
|
type
|
|
TSecureSocket* {.final.} = object
|
|
ssl: PSSL
|
|
bio: PBIO
|
|
|
|
proc connect*(sock: var TSecureSocket, address: string,
|
|
port: int): Int =
|
|
## Connects to the specified `address` on the specified `port`.
|
|
## Returns the result of the certificate validation.
|
|
SslLoadErrorStrings()
|
|
ERR_load_BIO_strings()
|
|
|
|
if SSL_library_init() != 1:
|
|
OSError()
|
|
|
|
var ctx = SSL_CTX_new(SSLv23_client_method())
|
|
if ctx == nil:
|
|
ERR_print_errors_fp(stderr)
|
|
OSError()
|
|
|
|
#if SSL_CTX_load_verify_locations(ctx,
|
|
# "/tmp/openssl-0.9.8e/certs/vsign1.pem", NIL) == 0:
|
|
# echo("Failed load verify locations")
|
|
# ERR_print_errors_fp(stderr)
|
|
|
|
sock.bio = BIO_new_ssl_connect(ctx)
|
|
if BIO_get_ssl(sock.bio, addr(sock.ssl)) == 0:
|
|
OSError()
|
|
|
|
if BIO_set_conn_hostname(sock.bio, address & ":" & $port) != 1:
|
|
OSError()
|
|
|
|
if BIO_do_connect(sock.bio) <= 0:
|
|
ERR_print_errors_fp(stderr)
|
|
OSError()
|
|
|
|
result = SSL_get_verify_result(sock.ssl)
|
|
|
|
proc recvLine*(sock: TSecureSocket, line: var TaintedString): bool =
|
|
## Acts in a similar fashion to the `recvLine` in the sockets module.
|
|
## Returns false when no data is available to be read.
|
|
## `Line` must be initialized and not nil!
|
|
setLen(line.string, 0)
|
|
while True:
|
|
var c: array[0..0, char]
|
|
var n = BIO_read(sock.bio, c, c.len)
|
|
if n <= 0: return False
|
|
if c[0] == '\r':
|
|
n = BIO_read(sock.bio, c, c.len)
|
|
if n > 0 and c[0] == '\L':
|
|
return True
|
|
elif n <= 0:
|
|
return False
|
|
elif c[0] == '\L': return True
|
|
add(line.string, c)
|
|
|
|
|
|
proc send*(sock: TSecureSocket, data: string) =
|
|
## Writes `data` to the socket.
|
|
if BIO_write(sock.bio, data, data.len()) <= 0:
|
|
OSError()
|
|
|
|
proc close*(sock: TSecureSocket) =
|
|
## Closes the socket
|
|
if BIO_free(sock.bio) <= 0:
|
|
ERR_print_errors_fp(stderr)
|
|
OSError()
|
|
|
|
when isMainModule:
|
|
var s: TSecureSocket
|
|
echo connect(s, "smtp.gmail.com", 465)
|
|
|
|
#var buffer: array[0..255, char]
|
|
#echo BIO_read(bio, buffer, buffer.len)
|
|
var buffer: string = ""
|
|
|
|
echo s.recvLine(buffer)
|
|
echo buffer
|
|
echo buffer.len
|
|
|