Skip to content

Commit

Permalink
Added an openssl wrapper, ssl module and smtp module.
Browse files Browse the repository at this point in the history
  • Loading branch information
dom96 committed Oct 23, 2010
1 parent 56e963b commit d68782c
Show file tree
Hide file tree
Showing 3 changed files with 643 additions and 0 deletions.
81 changes: 81 additions & 0 deletions lib/impure/ssl.nim
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
#
#
# Nimrod's Runtime Library
# (c) Copyright 2010 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* = object
ssl: PSSL
bio: PBIO

proc connect*(sock: var TSecureSocket, address: string, port: int, certResult: var Int) =
## Connects to the specified `address` on the specified `port`. `certResult` will become the result of the certificate validation.
SslLoadErrorStrings()
ERR_load_BIO_strings()

assert(SSL_library_init() == 1)

var ctx = SSL_CTX_new(SSLv23_client_method())
if ctx == nil:
ERR_print_errors_fp(stderr)
assert(False)

#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)
assert(BIO_get_ssl(sock.bio, addr(sock.ssl)) != 0)

assert(BIO_set_conn_hostname(sock.bio, address & ":" & $port) == 1)

if BIO_do_connect(sock.bio) <= 0:
ERR_print_errors_fp(stderr)
OSError()

certResult = SSL_get_verify_result(sock.ssl)

proc recvLine*(sock: TSecureSocket, line: var String): 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, 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, c)


proc send*(sock: TSecureSocket, data: string) =
## Writes `data` to the socket.
if BIO_write(sock.bio, data, data.len()) <= 0:
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

176 changes: 176 additions & 0 deletions lib/pure/smtp.nim
Original file line number Diff line number Diff line change
@@ -0,0 +1,176 @@
#
#
# Nimrod's Runtime Library
# (c) Copyright 2010 Dominik Picheta
#
# See the file "copying.txt", included in this
# distribution, for details about the copyright.
#

## This module implements the SMTP client protocol as specified by RFC 5321,
## this can be used to send mail to any SMTP Server.
##
## This module also implements the protocol used to format messages, as specified by RFC 2822.
##
## Example gmail use:
##
##
## .. code-block:: Nimrod
## var msg = createMessage("Hello from Nimrod's SMTP", "Hello!.\n Is this awesome or what?", @["foo@gmail.com"])
## var smtp = connect("smtp.gmail.com", 465, True, True)
## smtp.auth("username", "password")
## smtp.sendmail("username@gmail.com", @["foo@gmail.com"], $msg)
##

import sockets, strutils, strtabs, ssl, base64

type
TSMTP* = object
sock: TSocket
sslSock: TSecureSocket
ssl: Bool
debug: Bool

TMessage* = object
msgTo: seq[String]
msgCc: seq[String]
msgSubject: String
msgOtherHeaders: PStringTable
msgBody: String

EInvalidReply* = object of EBase

proc debugSend(smtp: TSMTP, cmd: String) =
if smtp.debug:
echo("C:" & cmd)
if not smtp.ssl:
smtp.sock.send(cmd)
else:
smtp.sslSock.send(cmd)

proc debugRecv(smtp: TSMTP): String =
var line = ""
var ret = False
if not smtp.ssl:
ret = smtp.sock.recvLine(line)
else:
ret = smtp.sslSock.recvLine(line)
if ret:
if smtp.debug:
echo("S:" & line)
return line
else:
echo("S-Warning: recvLine failed.")
return ""

proc quitExcpt(smtp: TSMTP, msg: String) =
smtp.debugSend("QUIT")
raise newException(EInvalidReply, msg)

proc checkReply(smtp: TSMTP, reply: string) =
var line = smtp.debugRecv()
if not line.startswith(reply):
quitExcpt(smtp, "Expected " & reply & " reply, got: " & line)

proc connect*(address: String, port: int = 25, ssl: bool = False, debug: bool = False): TSMTP =
## Establishes a connection with a SMTP server.
## May fail with EInvalidReply or with a socket errors.

if not ssl:
result.sock = socket()
result.sock.connect(address, TPort(port))
else:
result.ssl = True
var certResult: int
result.sslSock.connect(address, port, certResult)

result.debug = debug

result.checkReply("220")
result.debugSend("HELO " & address & "\c\L")
result.checkReply("250")

proc auth*(smtp: TSMTP, username, password: string) =
## Sends an AUTH command to the server to login as the `username` using `password`.
## May fail with EInvalidReply.

smtp.debugSend("AUTH LOGIN\c\L")
smtp.checkReply("334") # TODO: Check whether it's asking for the "Username:"
# i.e "334 VXNlcm5hbWU6"
smtp.debugSend(encode(username) & "\c\L")
smtp.checkReply("334") # TODO: Same as above, only "Password:" (I think?)

smtp.debugSend(encode(password) & "\c\L")
smtp.checkReply("235") # Check whether the authentification was successful.

proc sendmail*(smtp: TSMTP, fromaddr: string, toaddrs: seq[string], msg: string) =
## Sends `msg` from `fromaddr` to `toaddr`.
## Messages may be formed using ``createMessage`` by converting the TMessage into a string.

smtp.debugSend("MAIL FROM:<" & fromaddr & ">\c\L")
smtp.checkReply("250")
for address in items(toaddrs):
smtp.debugSend("RCPT TO:<" & address & ">\c\L")
smtp.checkReply("250")

# Send the message
smtp.debugSend("DATA " & "\c\L")
smtp.checkReply("354")
smtp.debugSend(msg & "\c\L")
smtp.debugSend(".\c\L")
smtp.checkReply("250")

# quit
smtp.debugSend("QUIT\c\L")

proc createMessage*(mSubject, mBody: String, mTo, mCc: seq[String],
otherHeaders: openarray[tuple[name, value: String]]): TMessage =
## Creates a new MIME compliant message.
result.msgTo = mTo
result.msgCc = mCc
result.msgSubject = mSubject
result.msgBody = mBody
result.msgOtherHeaders = newStringTable()
for n, v in items(otherHeaders):
result.msgOtherHeaders[n] = v

proc createMessage*(mSubject, mBody: String, mTo, mCc: seq[String] = @[]): TMessage =
## Alternate version of the above.
result.msgTo = mTo
result.msgCc = mCc
result.msgSubject = mSubject
result.msgBody = mBody
result.msgOtherHeaders = newStringTable()

proc `$`*(msg: TMessage): String =
result = ""
if msg.msgTo.len() > 0:
result = "TO: " & msg.msgTo.join(", ") & "\c\L"
if msg.msgCc.len() > 0:
result.add("CC: " & msg.msgTo.join(", ") & "\c\L")
# TODO: Folding? i.e when a line is too long, shorten it...
result.add("Subject: " & msg.msgSubject & "\c\L")
for key, value in pairs(msg.msgOtherHeaders):
result.add(key & ": " & value & "\c\L")

result.add("\c\L" & msg.msgBody)


when isMainModule:
#var msg = createMessage("Test subject!", "Hello, my name is dom96.\n What\'s yours?", @["dominik@localhost"])
#echo(msg)

#var smtp = connect("localhost", 25, False, True)
#smtp.sendmail("root@localhost", @["dominik@localhost"], $msg)

#echo(decode("a17sm3701420wbe.12"))

var msg = createMessage("Hello from Nimrod's SMTP!", "Hello!!!!.\n Is this awesome or what?", @["someone@yahoo.com", "someone@gmail.com"])
echo(msg)

var smtp = connect("smtp.gmail.com", 465, True, True)
smtp.auth("someone", "password")
smtp.sendmail("someone@gmail.com", @["someone@yahoo.com", "someone@gmail.com"], $msg)



Loading

0 comments on commit d68782c

Please sign in to comment.