mirror of
https://github.com/status-im/nim-json-rpc.git
synced 2025-02-24 18:28:10 +00:00
Merge pull request #21 from status-im/TransportAgnosticRpcNoDSL
Transport agnostic RPC
This commit is contained in:
commit
35faf61d40
4
.gitignore
vendored
4
.gitignore
vendored
@ -14,3 +14,7 @@
|
|||||||
# Ignore the nimcache folders
|
# Ignore the nimcache folders
|
||||||
nimcache/
|
nimcache/
|
||||||
|
|
||||||
|
# Ignore editor settings
|
||||||
|
.vscode
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,354 +0,0 @@
|
|||||||
import json, tables, strutils, options, macros #, chronicles
|
|
||||||
import asyncdispatch2
|
|
||||||
import jsonmarshal
|
|
||||||
|
|
||||||
export asyncdispatch2, json, jsonmarshal
|
|
||||||
|
|
||||||
# Temporarily disable logging
|
|
||||||
macro debug(body: varargs[untyped]): untyped = newStmtList()
|
|
||||||
macro info(body: varargs[untyped]): untyped = newStmtList()
|
|
||||||
macro error(body: varargs[untyped]): untyped = newStmtList()
|
|
||||||
|
|
||||||
#logScope:
|
|
||||||
# topics = "RpcServer"
|
|
||||||
|
|
||||||
type
|
|
||||||
RpcJsonError* = enum rjeInvalidJson, rjeVersionError, rjeNoMethod, rjeNoId
|
|
||||||
|
|
||||||
RpcJsonErrorContainer* = tuple[err: RpcJsonError, msg: string]
|
|
||||||
|
|
||||||
# Procedure signature accepted as an RPC call by server
|
|
||||||
RpcProc* = proc (params: JsonNode): Future[JsonNode]
|
|
||||||
|
|
||||||
RpcServer* = ref object
|
|
||||||
servers*: seq[StreamServer]
|
|
||||||
procs*: TableRef[string, RpcProc]
|
|
||||||
|
|
||||||
RpcProcError* = ref object of Exception
|
|
||||||
code*: int
|
|
||||||
data*: JsonNode
|
|
||||||
|
|
||||||
RpcBindError* = object of Exception
|
|
||||||
RpcAddressUnresolvableError* = object of Exception
|
|
||||||
|
|
||||||
const
|
|
||||||
JSON_PARSE_ERROR* = -32700
|
|
||||||
INVALID_REQUEST* = -32600
|
|
||||||
METHOD_NOT_FOUND* = -32601
|
|
||||||
INVALID_PARAMS* = -32602
|
|
||||||
INTERNAL_ERROR* = -32603
|
|
||||||
SERVER_ERROR* = -32000
|
|
||||||
|
|
||||||
maxRequestLength = 1024 * 128
|
|
||||||
|
|
||||||
jsonErrorMessages*: array[RpcJsonError, (int, string)] =
|
|
||||||
[
|
|
||||||
(JSON_PARSE_ERROR, "Invalid JSON"),
|
|
||||||
(INVALID_REQUEST, "JSON 2.0 required"),
|
|
||||||
(INVALID_REQUEST, "No method requested"),
|
|
||||||
(INVALID_REQUEST, "No id specified")
|
|
||||||
]
|
|
||||||
|
|
||||||
# Utility functions
|
|
||||||
# TODO: Move outside server
|
|
||||||
func `%`*(p: Port): JsonNode = %(p.int)
|
|
||||||
|
|
||||||
# Json state checking
|
|
||||||
|
|
||||||
template jsonValid*(jsonString: string, node: var JsonNode): (bool, string) =
|
|
||||||
var
|
|
||||||
valid = true
|
|
||||||
msg = ""
|
|
||||||
try: node = parseJson(line)
|
|
||||||
except:
|
|
||||||
valid = false
|
|
||||||
msg = getCurrentExceptionMsg()
|
|
||||||
debug "Cannot process json", json = jsonString, msg = msg
|
|
||||||
(valid, msg)
|
|
||||||
|
|
||||||
proc checkJsonErrors*(line: string,
|
|
||||||
node: var JsonNode): Option[RpcJsonErrorContainer] =
|
|
||||||
## Tries parsing line into node, if successful checks required fields
|
|
||||||
## Returns: error state or none
|
|
||||||
let res = jsonValid(line, node)
|
|
||||||
if not res[0]:
|
|
||||||
return some((rjeInvalidJson, res[1]))
|
|
||||||
if not node.hasKey("id"):
|
|
||||||
return some((rjeNoId, ""))
|
|
||||||
if node{"jsonrpc"} != %"2.0":
|
|
||||||
return some((rjeVersionError, ""))
|
|
||||||
if not node.hasKey("method"):
|
|
||||||
return some((rjeNoMethod, ""))
|
|
||||||
return none(RpcJsonErrorContainer)
|
|
||||||
|
|
||||||
# Json reply wrappers
|
|
||||||
|
|
||||||
proc wrapReply*(id: JsonNode, value: JsonNode, error: JsonNode): string =
|
|
||||||
let node = %{"jsonrpc": %"2.0", "result": value, "error": error, "id": id}
|
|
||||||
return $node & "\c\l"
|
|
||||||
|
|
||||||
proc sendError*(client: StreamTransport, code: int, msg: string, id: JsonNode,
|
|
||||||
data: JsonNode = newJNull()) {.async.} =
|
|
||||||
## Send error message to client
|
|
||||||
let error = %{"code": %(code), "id": id, "message": %msg, "data": data}
|
|
||||||
debug "Error generated", error = error, id = id
|
|
||||||
var res = wrapReply(id, newJNull(), error)
|
|
||||||
result = client.write(res)
|
|
||||||
|
|
||||||
proc sendJsonError*(state: RpcJsonError, client: StreamTransport, id: JsonNode,
|
|
||||||
data = newJNull()) {.async.} =
|
|
||||||
## Send client response for invalid json state
|
|
||||||
let errMsgs = jsonErrorMessages[state]
|
|
||||||
await client.sendError(errMsgs[0], errMsgs[1], id, data)
|
|
||||||
|
|
||||||
# Server message processing
|
|
||||||
proc processMessage(server: RpcServer, client: StreamTransport,
|
|
||||||
line: string) {.async.} =
|
|
||||||
var
|
|
||||||
node: JsonNode
|
|
||||||
# set up node and/or flag errors
|
|
||||||
jsonErrorState = checkJsonErrors(line, node)
|
|
||||||
|
|
||||||
if jsonErrorState.isSome:
|
|
||||||
let errState = jsonErrorState.get
|
|
||||||
var id =
|
|
||||||
if errState.err == rjeInvalidJson or errState.err == rjeNoId:
|
|
||||||
newJNull()
|
|
||||||
else:
|
|
||||||
node["id"]
|
|
||||||
await errState.err.sendJsonError(client, id, %errState.msg)
|
|
||||||
else:
|
|
||||||
let
|
|
||||||
methodName = node["method"].str
|
|
||||||
id = node["id"]
|
|
||||||
|
|
||||||
if not server.procs.hasKey(methodName):
|
|
||||||
await client.sendError(METHOD_NOT_FOUND, "Method not found", %id,
|
|
||||||
%(methodName & " is not a registered method."))
|
|
||||||
else:
|
|
||||||
let callRes = await server.procs[methodName](node["params"])
|
|
||||||
var res = wrapReply(id, callRes, newJNull())
|
|
||||||
discard await client.write(res)
|
|
||||||
|
|
||||||
proc processClient(server: StreamServer, client: StreamTransport) {.async, gcsafe.} =
|
|
||||||
var rpc = getUserData[RpcServer](server)
|
|
||||||
while true:
|
|
||||||
let line = await client.readLine(maxRequestLength)
|
|
||||||
if line == "":
|
|
||||||
client.close()
|
|
||||||
break
|
|
||||||
|
|
||||||
debug "Processing client", addresss = client.remoteAddress(), line
|
|
||||||
|
|
||||||
let future = processMessage(rpc, client, line)
|
|
||||||
yield future
|
|
||||||
if future.failed:
|
|
||||||
if future.readError of RpcProcError:
|
|
||||||
let err = future.readError.RpcProcError
|
|
||||||
await client.sendError(err.code, err.msg, err.data)
|
|
||||||
elif future.readError of ValueError:
|
|
||||||
let err = future.readError[].ValueError
|
|
||||||
await client.sendError(INVALID_PARAMS, err.msg, %"")
|
|
||||||
else:
|
|
||||||
await client.sendError(SERVER_ERROR,
|
|
||||||
"Error: Unknown error occurred", %"")
|
|
||||||
|
|
||||||
proc newRpcServer*(addresses: openarray[TransportAddress]): RpcServer =
|
|
||||||
## Create new server and assign it to addresses ``addresses``.
|
|
||||||
result = RpcServer()
|
|
||||||
result.procs = newTable[string, RpcProc]()
|
|
||||||
result.servers = newSeq[StreamServer]()
|
|
||||||
|
|
||||||
for item in addresses:
|
|
||||||
try:
|
|
||||||
info "Creating server on ", address = $item
|
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
|
||||||
udata = result)
|
|
||||||
result.servers.add(server)
|
|
||||||
except:
|
|
||||||
error "Failed to create server", address = $item, message = getCurrentExceptionMsg()
|
|
||||||
|
|
||||||
if len(result.servers) == 0:
|
|
||||||
# Server was not bound, critical error.
|
|
||||||
# TODO: Custom RpcException error
|
|
||||||
raise newException(RpcBindError, "Unable to create server!")
|
|
||||||
|
|
||||||
proc newRpcServer*(addresses: openarray[string]): RpcServer =
|
|
||||||
## Create new server and assign it to addresses ``addresses``.
|
|
||||||
var
|
|
||||||
tas4: seq[TransportAddress]
|
|
||||||
tas6: seq[TransportAddress]
|
|
||||||
baddrs: seq[TransportAddress]
|
|
||||||
|
|
||||||
for a in addresses:
|
|
||||||
# Attempt to resolve `address` for IPv4 address space.
|
|
||||||
try:
|
|
||||||
tas4 = resolveTAddress(a, IpAddressFamily.IPv4)
|
|
||||||
except:
|
|
||||||
discard
|
|
||||||
|
|
||||||
# Attempt to resolve `address` for IPv6 address space.
|
|
||||||
try:
|
|
||||||
tas6 = resolveTAddress(a, IpAddressFamily.IPv6)
|
|
||||||
except:
|
|
||||||
discard
|
|
||||||
|
|
||||||
for r in tas4:
|
|
||||||
baddrs.add(r)
|
|
||||||
for r in tas6:
|
|
||||||
baddrs.add(r)
|
|
||||||
|
|
||||||
if len(baddrs) == 0:
|
|
||||||
# Addresses could not be resolved, critical error.
|
|
||||||
raise newException(RpcAddressUnresolvableError, "Unable to get address!")
|
|
||||||
|
|
||||||
result = newRpcServer(baddrs)
|
|
||||||
|
|
||||||
proc newRpcServer*(address = "localhost", port: Port = Port(8545)): RpcServer =
|
|
||||||
var
|
|
||||||
tas4: seq[TransportAddress]
|
|
||||||
tas6: seq[TransportAddress]
|
|
||||||
|
|
||||||
# Attempt to resolve `address` for IPv4 address space.
|
|
||||||
try:
|
|
||||||
tas4 = resolveTAddress(address, port, IpAddressFamily.IPv4)
|
|
||||||
except:
|
|
||||||
discard
|
|
||||||
|
|
||||||
# Attempt to resolve `address` for IPv6 address space.
|
|
||||||
try:
|
|
||||||
tas6 = resolveTAddress(address, port, IpAddressFamily.IPv6)
|
|
||||||
except:
|
|
||||||
discard
|
|
||||||
|
|
||||||
if len(tas4) == 0 and len(tas6) == 0:
|
|
||||||
# Address was not resolved, critical error.
|
|
||||||
raise newException(RpcAddressUnresolvableError,
|
|
||||||
"Address " & address & " could not be resolved!")
|
|
||||||
|
|
||||||
result = RpcServer()
|
|
||||||
result.procs = newTable[string, RpcProc]()
|
|
||||||
result.servers = newSeq[StreamServer]()
|
|
||||||
for item in tas4:
|
|
||||||
try:
|
|
||||||
info "Creating server for address", ip4address = $item
|
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
|
||||||
udata = result)
|
|
||||||
result.servers.add(server)
|
|
||||||
except:
|
|
||||||
error "Failed to create server for address", address = $item
|
|
||||||
|
|
||||||
for item in tas6:
|
|
||||||
try:
|
|
||||||
info "Server created", ip6address = $item
|
|
||||||
var server = createStreamServer(item, processClient, {ReuseAddr},
|
|
||||||
udata = result)
|
|
||||||
result.servers.add(server)
|
|
||||||
except:
|
|
||||||
error "Failed to create server", address = $item
|
|
||||||
|
|
||||||
if len(result.servers) == 0:
|
|
||||||
# Server was not bound, critical error.
|
|
||||||
raise newException(RpcBindError,
|
|
||||||
"Could not setup server on " & address & ":" & $int(port))
|
|
||||||
|
|
||||||
proc start*(server: RpcServer) =
|
|
||||||
## Start the RPC server.
|
|
||||||
for item in server.servers:
|
|
||||||
item.start()
|
|
||||||
|
|
||||||
proc stop*(server: RpcServer) =
|
|
||||||
## Stop the RPC server.
|
|
||||||
for item in server.servers:
|
|
||||||
item.stop()
|
|
||||||
|
|
||||||
proc close*(server: RpcServer) =
|
|
||||||
## Cleanup resources of RPC server.
|
|
||||||
for item in server.servers:
|
|
||||||
item.close()
|
|
||||||
|
|
||||||
# Server registration and RPC generation
|
|
||||||
|
|
||||||
proc register*(server: RpcServer, name: string, rpc: RpcProc) =
|
|
||||||
## Add a name/code pair to the RPC server.
|
|
||||||
server.procs[name] = rpc
|
|
||||||
|
|
||||||
proc unRegisterAll*(server: RpcServer) =
|
|
||||||
# Remove all remote procedure calls from this server.
|
|
||||||
server.procs.clear
|
|
||||||
|
|
||||||
proc makeProcName(s: string): string =
|
|
||||||
result = ""
|
|
||||||
for c in s:
|
|
||||||
if c.isAlphaNumeric: result.add c
|
|
||||||
|
|
||||||
proc hasReturnType(params: NimNode): bool =
|
|
||||||
if params != nil and params.len > 0 and params[0] != nil and
|
|
||||||
params[0].kind != nnkEmpty:
|
|
||||||
result = true
|
|
||||||
|
|
||||||
macro rpc*(server: RpcServer, path: string, body: untyped): untyped =
|
|
||||||
## Define a remote procedure call.
|
|
||||||
## Input and return parameters are defined using the ``do`` notation.
|
|
||||||
## For example:
|
|
||||||
## .. code-block:: nim
|
|
||||||
## myServer.rpc("path") do(param1: int, param2: float) -> string:
|
|
||||||
## result = $param1 & " " & $param2
|
|
||||||
## ```
|
|
||||||
## Input parameters are automatically marshalled from json to Nim types,
|
|
||||||
## and output parameters are automatically marshalled to json for transport.
|
|
||||||
result = newStmtList()
|
|
||||||
let
|
|
||||||
parameters = body.findChild(it.kind == nnkFormalParams)
|
|
||||||
# all remote calls have a single parameter: `params: JsonNode`
|
|
||||||
paramsIdent = newIdentNode"params"
|
|
||||||
# procs are generated from the stripped path
|
|
||||||
pathStr = $path
|
|
||||||
# strip non alphanumeric
|
|
||||||
procNameStr = pathStr.makeProcName
|
|
||||||
# public rpc proc
|
|
||||||
procName = newIdentNode(procNameStr)
|
|
||||||
# when parameters present: proc that contains our rpc body
|
|
||||||
doMain = newIdentNode(procNameStr & "DoMain")
|
|
||||||
# async result
|
|
||||||
res = newIdentNode("result")
|
|
||||||
var
|
|
||||||
setup = jsonToNim(parameters, paramsIdent)
|
|
||||||
procBody = if body.kind == nnkStmtList: body else: body.body
|
|
||||||
|
|
||||||
if parameters.hasReturnType:
|
|
||||||
let returnType = parameters[0]
|
|
||||||
|
|
||||||
# delegate async proc allows return and setting of result as native type
|
|
||||||
result.add(quote do:
|
|
||||||
proc `doMain`(`paramsIdent`: JsonNode): Future[`returnType`] {.async.} =
|
|
||||||
`setup`
|
|
||||||
`procBody`
|
|
||||||
)
|
|
||||||
|
|
||||||
if returnType == ident"JsonNode":
|
|
||||||
# `JsonNode` results don't need conversion
|
|
||||||
result.add( quote do:
|
|
||||||
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
|
||||||
`res` = await `doMain`(`paramsIdent`)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
result.add(quote do:
|
|
||||||
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
|
||||||
`res` = %await `doMain`(`paramsIdent`)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
# no return types, inline contents
|
|
||||||
result.add(quote do:
|
|
||||||
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
|
||||||
`setup`
|
|
||||||
`procBody`
|
|
||||||
)
|
|
||||||
result.add( quote do:
|
|
||||||
`server`.register(`path`, `procName`)
|
|
||||||
)
|
|
||||||
|
|
||||||
when defined(nimDumpRpcs):
|
|
||||||
echo "\n", pathStr, ": ", result.repr
|
|
||||||
|
|
||||||
# TODO: Allow cross checking between client signatures and server calls
|
|
@ -1,5 +1,5 @@
|
|||||||
packageName = "eth_rpc"
|
packageName = "json_rpc"
|
||||||
version = "0.0.1"
|
version = "0.0.2"
|
||||||
author = "Status Research & Development GmbH"
|
author = "Status Research & Development GmbH"
|
||||||
description = "Ethereum remote procedure calls"
|
description = "Ethereum remote procedure calls"
|
||||||
license = "Apache License 2.0"
|
license = "Apache License 2.0"
|
@ -1,31 +1,38 @@
|
|||||||
import tables, json, macros
|
import tables, json, macros
|
||||||
import asyncdispatch2
|
import asyncdispatch2
|
||||||
|
from strutils import toLowerAscii
|
||||||
import jsonmarshal
|
import jsonmarshal
|
||||||
|
export asyncdispatch2
|
||||||
|
|
||||||
type
|
type
|
||||||
RpcClient* = ref object
|
RpcClient*[T, A] = ref object
|
||||||
transp: StreamTransport
|
|
||||||
awaiting: Table[string, Future[Response]]
|
awaiting: Table[string, Future[Response]]
|
||||||
address: TransportAddress
|
transport: T
|
||||||
|
address: A
|
||||||
nextId: int64
|
nextId: int64
|
||||||
|
|
||||||
Response* = tuple[error: bool, result: JsonNode]
|
Response* = tuple[error: bool, result: JsonNode]
|
||||||
|
|
||||||
const maxRequestLength = 1024 * 128
|
const defaultMaxRequestLength* = 1024 * 128
|
||||||
|
|
||||||
proc newRpcClient*(): RpcClient =
|
proc newRpcClient*[T, A]: RpcClient[T, A] =
|
||||||
## Creates a new ``RpcClient`` instance.
|
## Creates a new ``RpcClient`` instance.
|
||||||
result = RpcClient(awaiting: initTable[string, Future[Response]](), nextId: 1)
|
result = RpcClient[T, A](awaiting: initTable[string, Future[Response]](), nextId: 1)
|
||||||
|
|
||||||
proc call*(self: RpcClient, name: string,
|
proc call*(self: RpcClient, name: string,
|
||||||
params: JsonNode): Future[Response] {.async.} =
|
params: JsonNode): Future[Response] {.async.} =
|
||||||
## Remotely calls the specified RPC method.
|
## Remotely calls the specified RPC method.
|
||||||
let id = $self.nextId
|
let id = $self.nextId
|
||||||
self.nextId.inc
|
self.nextId.inc
|
||||||
var msg = $ %{"jsonrpc": %"2.0", "method": %name, "params": params,
|
var
|
||||||
|
value =
|
||||||
|
$ %{"jsonrpc": %"2.0",
|
||||||
|
"method": %name,
|
||||||
|
"params": params,
|
||||||
"id": %id} & "\c\l"
|
"id": %id} & "\c\l"
|
||||||
let res = await self.transp.write(msg)
|
let res = await self.transport.write(value)
|
||||||
# TODO: Add actions when not full packet was send, e.g. disconnect peer.
|
# TODO: Add actions when not full packet was send, e.g. disconnect peer.
|
||||||
assert(res == len(msg))
|
assert(res == len(value))
|
||||||
|
|
||||||
# completed by processMessage.
|
# completed by processMessage.
|
||||||
var newFut = newFuture[Response]()
|
var newFut = newFuture[Response]()
|
||||||
@ -33,10 +40,8 @@ proc call*(self: RpcClient, name: string,
|
|||||||
self.awaiting[id] = newFut
|
self.awaiting[id] = newFut
|
||||||
result = await newFut
|
result = await newFut
|
||||||
|
|
||||||
template handleRaise[T](fut: Future[T], errType: typedesc, msg: string) =
|
template asyncRaise[T](fut: Future[T], errType: typedesc, msg: string) =
|
||||||
# complete future before raising
|
fut.fail(newException(errType, msg))
|
||||||
fut.complete((true, %msg))
|
|
||||||
raise newException(errType, msg)
|
|
||||||
|
|
||||||
macro checkGet(node: JsonNode, fieldName: string,
|
macro checkGet(node: JsonNode, fieldName: string,
|
||||||
jKind: static[JsonNodeKind]): untyped =
|
jKind: static[JsonNodeKind]): untyped =
|
||||||
@ -58,17 +63,18 @@ macro checkGet(node: JsonNode, fieldName: string,
|
|||||||
else: discard
|
else: discard
|
||||||
|
|
||||||
proc processMessage(self: RpcClient, line: string) =
|
proc processMessage(self: RpcClient, line: string) =
|
||||||
let node = parseJson(line)
|
# Note: this doesn't use any transport code so doesn't need to be differentiated.
|
||||||
|
let
|
||||||
|
node = parseJson(line) # TODO: Check errors
|
||||||
|
id = checkGet(node, "id", JString)
|
||||||
|
|
||||||
# TODO: Use more appropriate exception objects
|
|
||||||
let id = checkGet(node, "id", JString)
|
|
||||||
if not self.awaiting.hasKey(id):
|
if not self.awaiting.hasKey(id):
|
||||||
raise newException(ValueError,
|
raise newException(ValueError,
|
||||||
"Cannot find message id \"" & node["id"].str & "\"")
|
"Cannot find message id \"" & node["id"].str & "\"")
|
||||||
|
|
||||||
let version = checkGet(node, "jsonrpc", JString)
|
let version = checkGet(node, "jsonrpc", JString)
|
||||||
if version != "2.0":
|
if version != "2.0":
|
||||||
self.awaiting[id].handleRaise(ValueError,
|
self.awaiting[id].asyncRaise(ValueError,
|
||||||
"Unsupported version of JSON, expected 2.0, received \"" & version & "\"")
|
"Unsupported version of JSON, expected 2.0, received \"" & version & "\"")
|
||||||
|
|
||||||
let errorNode = node{"error"}
|
let errorNode = node{"error"}
|
||||||
@ -79,31 +85,34 @@ proc processMessage(self: RpcClient, line: string) =
|
|||||||
self.awaiting.del(id)
|
self.awaiting.del(id)
|
||||||
# TODO: actions on unable find result node
|
# TODO: actions on unable find result node
|
||||||
else:
|
else:
|
||||||
self.awaiting[id].complete((true, errorNode))
|
self.awaiting[id].fail(newException(ValueError, $errorNode))
|
||||||
self.awaiting.del(id)
|
self.awaiting.del(id)
|
||||||
|
|
||||||
proc connect*(self: RpcClient, address: string, port: Port): Future[void]
|
proc processData(client: RpcClient) {.async.} =
|
||||||
|
|
||||||
proc processData(self: RpcClient) {.async.} =
|
|
||||||
while true:
|
while true:
|
||||||
let line = await self.transp.readLine(maxRequestLength)
|
var value = await client.transport.readLine(defaultMaxRequestLength)
|
||||||
if line == "":
|
if value == "":
|
||||||
# transmission ends
|
# transmission ends
|
||||||
self.transp.close()
|
client.transport.close
|
||||||
break
|
break
|
||||||
|
|
||||||
processMessage(self, line)
|
client.processMessage(value)
|
||||||
# async loop reconnection and waiting
|
# async loop reconnection and waiting
|
||||||
self.transp = await connect(self.address)
|
client.transport = await connect(client.address)
|
||||||
|
|
||||||
proc connect*(self: RpcClient, address: string, port: Port) {.async.} =
|
type RpcStreamClient* = RpcClient[StreamTransport, TransportAddress]
|
||||||
# TODO: `address` hostname can be resolved to many IP addresses, we are using
|
|
||||||
# first one, but maybe it would be better to iterate over all IP addresses
|
proc connect*(client: RpcStreamClient, address: string, port: Port) {.async.} =
|
||||||
# and try to establish connection until it will not be established.
|
|
||||||
let addresses = resolveTAddress(address, port)
|
let addresses = resolveTAddress(address, port)
|
||||||
self.transp = await connect(addresses[0])
|
client.transport = await connect(addresses[0])
|
||||||
self.address = addresses[0]
|
client.address = addresses[0]
|
||||||
asyncCheck processData(self)
|
asyncCheck processData(client)
|
||||||
|
|
||||||
|
proc newRpcStreamClient*(): RpcStreamClient =
|
||||||
|
## Create new server and assign it to addresses ``addresses``.
|
||||||
|
result = newRpcClient[StreamTransport, TransportAddress]()
|
||||||
|
|
||||||
|
# Signature processing
|
||||||
|
|
||||||
proc createRpcProc(procName, parameters, callBody: NimNode): NimNode =
|
proc createRpcProc(procName, parameters, callBody: NimNode): NimNode =
|
||||||
# parameters come as a tree
|
# parameters come as a tree
|
240
json_rpc/router.nim
Normal file
240
json_rpc/router.nim
Normal file
@ -0,0 +1,240 @@
|
|||||||
|
import
|
||||||
|
json, tables, asyncdispatch2, jsonmarshal, strutils, macros,
|
||||||
|
chronicles, options
|
||||||
|
export asyncdispatch2, json, jsonmarshal
|
||||||
|
|
||||||
|
type
|
||||||
|
RpcJsonError* = enum rjeInvalidJson, rjeVersionError, rjeNoMethod, rjeNoId, rjeNoParams
|
||||||
|
RpcJsonErrorContainer* = tuple[err: RpcJsonError, msg: string]
|
||||||
|
|
||||||
|
# Procedure signature accepted as an RPC call by server
|
||||||
|
RpcProc* = proc(input: JsonNode): Future[JsonNode] {.gcsafe.}
|
||||||
|
|
||||||
|
RpcProcError* = ref object of Exception
|
||||||
|
code*: int
|
||||||
|
data*: JsonNode
|
||||||
|
|
||||||
|
RpcBindError* = object of Exception
|
||||||
|
RpcAddressUnresolvableError* = object of Exception
|
||||||
|
|
||||||
|
RpcRouter* = object
|
||||||
|
procs*: TableRef[string, RpcProc]
|
||||||
|
|
||||||
|
const
|
||||||
|
methodField = "method"
|
||||||
|
paramsField = "params"
|
||||||
|
jsonRpcField = "jsonrpc"
|
||||||
|
idField = "id"
|
||||||
|
resultField = "result"
|
||||||
|
errorField = "error"
|
||||||
|
codeField = "code"
|
||||||
|
messageField = "message"
|
||||||
|
dataField = "data"
|
||||||
|
messageTerminator = "\c\l"
|
||||||
|
|
||||||
|
JSON_PARSE_ERROR* = -32700
|
||||||
|
INVALID_REQUEST* = -32600
|
||||||
|
METHOD_NOT_FOUND* = -32601
|
||||||
|
INVALID_PARAMS* = -32602
|
||||||
|
INTERNAL_ERROR* = -32603
|
||||||
|
SERVER_ERROR* = -32000
|
||||||
|
|
||||||
|
defaultMaxRequestLength* = 1024 * 128
|
||||||
|
jsonErrorMessages*: array[RpcJsonError, (int, string)] =
|
||||||
|
[
|
||||||
|
(JSON_PARSE_ERROR, "Invalid JSON"),
|
||||||
|
(INVALID_REQUEST, "JSON 2.0 required"),
|
||||||
|
(INVALID_REQUEST, "No method requested"),
|
||||||
|
(INVALID_REQUEST, "No id specified"),
|
||||||
|
(INVALID_PARAMS, "No parameters specified")
|
||||||
|
]
|
||||||
|
|
||||||
|
proc newRpcRouter*: RpcRouter =
|
||||||
|
result.procs = newTable[string, RpcProc]()
|
||||||
|
|
||||||
|
proc register*(router: var RpcRouter, path: string, call: RpcProc) =
|
||||||
|
router.procs.add(path, call)
|
||||||
|
|
||||||
|
proc clear*(router: var RpcRouter) = router.procs.clear
|
||||||
|
|
||||||
|
proc hasMethod*(router: RpcRouter, methodName: string): bool = router.procs.hasKey(methodName)
|
||||||
|
|
||||||
|
template isEmpty(node: JsonNode): bool = node.isNil or node.kind == JNull
|
||||||
|
|
||||||
|
# Json state checking
|
||||||
|
|
||||||
|
template jsonValid*(jsonString: string, node: var JsonNode): (bool, string) =
|
||||||
|
var
|
||||||
|
valid = true
|
||||||
|
msg = ""
|
||||||
|
try: node = parseJson(line)
|
||||||
|
except:
|
||||||
|
valid = false
|
||||||
|
msg = getCurrentExceptionMsg()
|
||||||
|
debug "Cannot process json", json = jsonString, msg = msg
|
||||||
|
(valid, msg)
|
||||||
|
|
||||||
|
proc checkJsonState*(line: string,
|
||||||
|
node: var JsonNode): Option[RpcJsonErrorContainer] =
|
||||||
|
## Tries parsing line into node, if successful checks required fields
|
||||||
|
## Returns: error state or none
|
||||||
|
let res = jsonValid(line, node)
|
||||||
|
if not res[0]:
|
||||||
|
return some((rjeInvalidJson, res[1]))
|
||||||
|
if not node.hasKey(idField):
|
||||||
|
return some((rjeNoId, ""))
|
||||||
|
let jVer = node{jsonRpcField}
|
||||||
|
if jVer != nil and jVer.kind != JNull and jVer != %"2.0":
|
||||||
|
return some((rjeVersionError, ""))
|
||||||
|
if not node.hasKey(methodField):
|
||||||
|
return some((rjeNoMethod, ""))
|
||||||
|
if not node.hasKey(paramsField):
|
||||||
|
return some((rjeNoParams, ""))
|
||||||
|
return none(RpcJsonErrorContainer)
|
||||||
|
|
||||||
|
# Json reply wrappers
|
||||||
|
|
||||||
|
proc wrapReply*(id: JsonNode, value: JsonNode, error: JsonNode): JsonNode =
|
||||||
|
return %{jsonRpcField: %"2.0", resultField: value, errorField: error, idField: id}
|
||||||
|
|
||||||
|
proc wrapError*(code: int, msg: string, id: JsonNode,
|
||||||
|
data: JsonNode = newJNull()): JsonNode =
|
||||||
|
# Create standardised error json
|
||||||
|
result = %{codeField: %(code), idField: id, messageField: %msg, dataField: data}
|
||||||
|
debug "Error generated", error = result, id = id
|
||||||
|
|
||||||
|
proc route*(router: RpcRouter, node: JsonNode): Future[JsonNode] {.async, gcsafe.} =
|
||||||
|
## Assumes correct setup of node
|
||||||
|
let
|
||||||
|
methodName = node[methodField].str
|
||||||
|
id = node[idField]
|
||||||
|
rpcProc = router.procs.getOrDefault(methodName)
|
||||||
|
|
||||||
|
if rpcProc.isNil:
|
||||||
|
let
|
||||||
|
methodNotFound = %(methodName & " is not a registered RPC method.")
|
||||||
|
error = wrapError(METHOD_NOT_FOUND, "Method not found", id, methodNotFound)
|
||||||
|
result = wrapReply(id, newJNull(), error)
|
||||||
|
else:
|
||||||
|
let
|
||||||
|
jParams = node[paramsField]
|
||||||
|
res = await rpcProc(jParams)
|
||||||
|
result = wrapReply(id, res, newJNull())
|
||||||
|
|
||||||
|
proc route*(router: RpcRouter, data: string): Future[string] {.async, gcsafe.} =
|
||||||
|
## Route to RPC from string data. Data is expected to be able to be converted to Json.
|
||||||
|
## Returns string of Json from RPC result/error node
|
||||||
|
var
|
||||||
|
node: JsonNode
|
||||||
|
# parse json node and/or flag missing fields and errors
|
||||||
|
jsonErrorState = checkJsonState(data, node)
|
||||||
|
|
||||||
|
if jsonErrorState.isSome:
|
||||||
|
let errState = jsonErrorState.get
|
||||||
|
var id =
|
||||||
|
if errState.err == rjeInvalidJson or errState.err == rjeNoId:
|
||||||
|
newJNull()
|
||||||
|
else:
|
||||||
|
node["id"]
|
||||||
|
let
|
||||||
|
errMsg = jsonErrorMessages[errState.err]
|
||||||
|
res = wrapError(code = errMsg[0], msg = errMsg[1], id = id)
|
||||||
|
# return error state as json
|
||||||
|
result = $res & messageTerminator
|
||||||
|
else:
|
||||||
|
let res = await router.route(node)
|
||||||
|
result = $res & messageTerminator
|
||||||
|
|
||||||
|
proc tryRoute*(router: RpcRouter, data: JsonNode, fut: var Future[JsonNode]): bool =
|
||||||
|
## Route to RPC, returns false if the method or params cannot be found.
|
||||||
|
## Expects json input and returns json output.
|
||||||
|
let
|
||||||
|
jPath = data{methodField}
|
||||||
|
jParams = data{paramsField}
|
||||||
|
if jPath.isEmpty or jParams.isEmpty:
|
||||||
|
return false
|
||||||
|
|
||||||
|
let
|
||||||
|
path = jPath.getStr
|
||||||
|
rpc = router.procs.getOrDefault(path)
|
||||||
|
if rpc != nil:
|
||||||
|
fut = rpc(jParams)
|
||||||
|
return true
|
||||||
|
|
||||||
|
proc makeProcName(s: string): string =
|
||||||
|
result = ""
|
||||||
|
for c in s:
|
||||||
|
if c.isAlphaNumeric: result.add c
|
||||||
|
|
||||||
|
proc hasReturnType(params: NimNode): bool =
|
||||||
|
if params != nil and params.len > 0 and params[0] != nil and
|
||||||
|
params[0].kind != nnkEmpty:
|
||||||
|
result = true
|
||||||
|
|
||||||
|
macro rpc*(server: RpcRouter, path: string, body: untyped): untyped =
|
||||||
|
## Define a remote procedure call.
|
||||||
|
## Input and return parameters are defined using the ``do`` notation.
|
||||||
|
## For example:
|
||||||
|
## .. code-block:: nim
|
||||||
|
## myServer.rpc("path") do(param1: int, param2: float) -> string:
|
||||||
|
## result = $param1 & " " & $param2
|
||||||
|
## ```
|
||||||
|
## Input parameters are automatically marshalled from json to Nim types,
|
||||||
|
## and output parameters are automatically marshalled to json for transport.
|
||||||
|
result = newStmtList()
|
||||||
|
let
|
||||||
|
parameters = body.findChild(it.kind == nnkFormalParams)
|
||||||
|
# all remote calls have a single parameter: `params: JsonNode`
|
||||||
|
paramsIdent = newIdentNode"params"
|
||||||
|
# procs are generated from the stripped path
|
||||||
|
pathStr = $path
|
||||||
|
# strip non alphanumeric
|
||||||
|
procNameStr = pathStr.makeProcName
|
||||||
|
# public rpc proc
|
||||||
|
procName = newIdentNode(procNameStr)
|
||||||
|
# when parameters present: proc that contains our rpc body
|
||||||
|
doMain = newIdentNode(procNameStr & "DoMain")
|
||||||
|
# async result
|
||||||
|
res = newIdentNode("result")
|
||||||
|
var
|
||||||
|
setup = jsonToNim(parameters, paramsIdent)
|
||||||
|
procBody = if body.kind == nnkStmtList: body else: body.body
|
||||||
|
errTrappedBody = quote do:
|
||||||
|
try:
|
||||||
|
`procBody`
|
||||||
|
except:
|
||||||
|
debug "Error occurred within RPC ", path = `path`, errorMessage = getCurrentExceptionMsg()
|
||||||
|
if parameters.hasReturnType:
|
||||||
|
let returnType = parameters[0]
|
||||||
|
|
||||||
|
# delegate async proc allows return and setting of result as native type
|
||||||
|
result.add(quote do:
|
||||||
|
proc `doMain`(`paramsIdent`: JsonNode): Future[`returnType`] {.async.} =
|
||||||
|
`setup`
|
||||||
|
`errTrappedBody`
|
||||||
|
)
|
||||||
|
|
||||||
|
if returnType == ident"JsonNode":
|
||||||
|
# `JsonNode` results don't need conversion
|
||||||
|
result.add( quote do:
|
||||||
|
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
||||||
|
`res` = await `doMain`(`paramsIdent`)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
result.add(quote do:
|
||||||
|
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
||||||
|
`res` = %await `doMain`(`paramsIdent`)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# no return types, inline contents
|
||||||
|
result.add(quote do:
|
||||||
|
proc `procName`(`paramsIdent`: JsonNode): Future[JsonNode] {.async.} =
|
||||||
|
`setup`
|
||||||
|
`errTrappedBody`
|
||||||
|
)
|
||||||
|
result.add( quote do:
|
||||||
|
`server`.register(`path`, `procName`)
|
||||||
|
)
|
||||||
|
|
||||||
|
when defined(nimDumpRpcs):
|
||||||
|
echo "\n", pathStr, ": ", result.repr
|
37
json_rpc/server.nim
Normal file
37
json_rpc/server.nim
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
import json, tables, options, macros
|
||||||
|
import asyncdispatch2, router
|
||||||
|
import jsonmarshal
|
||||||
|
|
||||||
|
export asyncdispatch2, json, jsonmarshal, router
|
||||||
|
|
||||||
|
type
|
||||||
|
RpcServer*[S] = ref object
|
||||||
|
servers*: seq[S]
|
||||||
|
router*: RpcRouter
|
||||||
|
|
||||||
|
proc newRpcServer*[S](): RpcServer[S] =
|
||||||
|
new result
|
||||||
|
result.router = newRpcRouter()
|
||||||
|
result.servers = @[]
|
||||||
|
|
||||||
|
template rpc*(server: RpcServer, path: string, body: untyped): untyped =
|
||||||
|
server.router.rpc(path, body)
|
||||||
|
|
||||||
|
template hasMethod*(server: RpcServer, methodName: string): bool = server.router.hasMethod(methodName)
|
||||||
|
|
||||||
|
# Wrapper for message processing
|
||||||
|
|
||||||
|
proc route*[T](server: RpcServer[T], line: string): Future[string] {.async, gcsafe.} =
|
||||||
|
result = await server.router.route(line)
|
||||||
|
|
||||||
|
# Server registration
|
||||||
|
|
||||||
|
proc register*(server: RpcServer, name: string, rpc: RpcProc) =
|
||||||
|
## Add a name/code pair to the RPC server.
|
||||||
|
server.router.addRoute(name, rpc)
|
||||||
|
|
||||||
|
proc unRegisterAll*(server: RpcServer) =
|
||||||
|
# Remove all remote procedure calls from this server.
|
||||||
|
server.router.clear
|
||||||
|
|
||||||
|
|
156
json_rpc/sockettransport.nim
Normal file
156
json_rpc/sockettransport.nim
Normal file
@ -0,0 +1,156 @@
|
|||||||
|
import server, json, chronicles
|
||||||
|
|
||||||
|
proc sendError*[T](transport: T, code: int, msg: string, id: JsonNode,
|
||||||
|
data: JsonNode = newJNull()) {.async.} =
|
||||||
|
## Send error message to client
|
||||||
|
let error = wrapError(code, msg, id, data)
|
||||||
|
var value = $wrapReply(id, newJNull(), error)
|
||||||
|
result = transport.write(value)
|
||||||
|
|
||||||
|
proc processClient(server: StreamServer, transport: StreamTransport) {.async, gcsafe.} =
|
||||||
|
## Process transport data to the RPC server
|
||||||
|
var rpc = getUserData[RpcServer[StreamTransport]](server)
|
||||||
|
while true:
|
||||||
|
var
|
||||||
|
maxRequestLength = defaultMaxRequestLength
|
||||||
|
value = await transport.readLine(defaultMaxRequestLength)
|
||||||
|
if value == "":
|
||||||
|
transport.close
|
||||||
|
break
|
||||||
|
|
||||||
|
debug "Processing message", address = transport.remoteAddress(), line = value
|
||||||
|
|
||||||
|
let future = rpc.route(value)
|
||||||
|
yield future
|
||||||
|
if future.failed:
|
||||||
|
if future.readError of RpcProcError:
|
||||||
|
let err = future.readError.RpcProcError
|
||||||
|
await transport.sendError(err.code, err.msg, err.data)
|
||||||
|
elif future.readError of ValueError:
|
||||||
|
let err = future.readError[].ValueError
|
||||||
|
await transport.sendError(INVALID_PARAMS, err.msg, %"")
|
||||||
|
else:
|
||||||
|
await transport.sendError(SERVER_ERROR,
|
||||||
|
"Error: Unknown error occurred", %"")
|
||||||
|
else:
|
||||||
|
let res = await future
|
||||||
|
result = transport.write(res)
|
||||||
|
|
||||||
|
# Utility functions for setting up servers using stream transport addresses
|
||||||
|
|
||||||
|
proc addStreamServer*(server: RpcServer[StreamServer], address: TransportAddress) =
|
||||||
|
try:
|
||||||
|
info "Creating server on ", address = $address
|
||||||
|
var transportServer = createStreamServer(address, processClient, {ReuseAddr}, udata = server)
|
||||||
|
server.servers.add(transportServer)
|
||||||
|
except:
|
||||||
|
error "Failed to create server", address = $address, message = getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
if len(server.servers) == 0:
|
||||||
|
# Server was not bound, critical error.
|
||||||
|
raise newException(RpcBindError, "Unable to create server!")
|
||||||
|
|
||||||
|
proc addStreamServers*(server: RpcServer[StreamServer], addresses: openarray[TransportAddress]) =
|
||||||
|
for item in addresses:
|
||||||
|
server.addStreamServer(item)
|
||||||
|
|
||||||
|
proc addStreamServer*(server: RpcServer[StreamServer], address: string) =
|
||||||
|
## Create new server and assign it to addresses ``addresses``.
|
||||||
|
var
|
||||||
|
tas4: seq[TransportAddress]
|
||||||
|
tas6: seq[TransportAddress]
|
||||||
|
added = 0
|
||||||
|
|
||||||
|
# Attempt to resolve `address` for IPv4 address space.
|
||||||
|
try:
|
||||||
|
tas4 = resolveTAddress(address, IpAddressFamily.IPv4)
|
||||||
|
except:
|
||||||
|
discard
|
||||||
|
|
||||||
|
# Attempt to resolve `address` for IPv6 address space.
|
||||||
|
try:
|
||||||
|
tas6 = resolveTAddress(address, IpAddressFamily.IPv6)
|
||||||
|
except:
|
||||||
|
discard
|
||||||
|
|
||||||
|
for r in tas4:
|
||||||
|
server.addStreamServer(r)
|
||||||
|
added.inc
|
||||||
|
for r in tas6:
|
||||||
|
server.addStreamServer(r)
|
||||||
|
added.inc
|
||||||
|
|
||||||
|
if added == 0:
|
||||||
|
# Addresses could not be resolved, critical error.
|
||||||
|
raise newException(RpcAddressUnresolvableError, "Unable to get address!")
|
||||||
|
|
||||||
|
proc addStreamServers*(server: RpcServer[StreamServer], addresses: openarray[string]) =
|
||||||
|
for address in addresses:
|
||||||
|
server.addStreamServer(address)
|
||||||
|
|
||||||
|
proc addStreamServer*(server: RpcServer[StreamServer], address: string, port: Port) =
|
||||||
|
var
|
||||||
|
tas4: seq[TransportAddress]
|
||||||
|
tas6: seq[TransportAddress]
|
||||||
|
added = 0
|
||||||
|
|
||||||
|
# Attempt to resolve `address` for IPv4 address space.
|
||||||
|
try:
|
||||||
|
tas4 = resolveTAddress(address, port, IpAddressFamily.IPv4)
|
||||||
|
except:
|
||||||
|
discard
|
||||||
|
|
||||||
|
# Attempt to resolve `address` for IPv6 address space.
|
||||||
|
try:
|
||||||
|
tas6 = resolveTAddress(address, port, IpAddressFamily.IPv6)
|
||||||
|
except:
|
||||||
|
discard
|
||||||
|
|
||||||
|
if len(tas4) == 0 and len(tas6) == 0:
|
||||||
|
# Address was not resolved, critical error.
|
||||||
|
raise newException(RpcAddressUnresolvableError,
|
||||||
|
"Address " & address & " could not be resolved!")
|
||||||
|
|
||||||
|
for r in tas4:
|
||||||
|
server.addStreamServer(r)
|
||||||
|
added.inc
|
||||||
|
for r in tas6:
|
||||||
|
server.addStreamServer(r)
|
||||||
|
added.inc
|
||||||
|
|
||||||
|
if len(server.servers) == 0:
|
||||||
|
# Server was not bound, critical error.
|
||||||
|
raise newException(RpcBindError,
|
||||||
|
"Could not setup server on " & address & ":" & $int(port))
|
||||||
|
|
||||||
|
type RpcStreamServer* = RpcServer[StreamServer]
|
||||||
|
|
||||||
|
proc newRpcStreamServer*(addresses: openarray[TransportAddress]): RpcStreamServer =
|
||||||
|
## Create new server and assign it to addresses ``addresses``.
|
||||||
|
result = newRpcServer[StreamServer]()
|
||||||
|
result.addStreamServers(addresses)
|
||||||
|
|
||||||
|
proc newRpcStreamServer*(addresses: openarray[string]): RpcStreamServer =
|
||||||
|
## Create new server and assign it to addresses ``addresses``.
|
||||||
|
result = newRpcServer[StreamServer]()
|
||||||
|
result.addStreamServers(addresses)
|
||||||
|
|
||||||
|
proc newRpcStreamServer*(address = "localhost", port: Port = Port(8545)): RpcStreamServer =
|
||||||
|
# Create server on specified port
|
||||||
|
result = newRpcServer[StreamServer]()
|
||||||
|
result.addStreamServer(address, port)
|
||||||
|
|
||||||
|
proc start*(server: RpcStreamServer) =
|
||||||
|
## Start the RPC server.
|
||||||
|
for item in server.servers:
|
||||||
|
item.start()
|
||||||
|
|
||||||
|
proc stop*(server: RpcStreamServer) =
|
||||||
|
## Stop the RPC server.
|
||||||
|
for item in server.servers:
|
||||||
|
item.stop()
|
||||||
|
|
||||||
|
proc close*(server: RpcStreamServer) =
|
||||||
|
## Cleanup resources of RPC server.
|
||||||
|
for item in server.servers:
|
||||||
|
item.close()
|
@ -1,3 +1,3 @@
|
|||||||
import eth-rpc / client
|
import json_rpc / client
|
||||||
export client
|
export client
|
||||||
|
|
||||||
|
@ -1,2 +1,2 @@
|
|||||||
import eth-rpc / server
|
import json_rpc / server
|
||||||
export server
|
export server
|
||||||
|
2
rpcsockets.nim
Normal file
2
rpcsockets.nim
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
import json_rpc / [server, sockettransport]
|
||||||
|
export server, sockettransport
|
@ -1,3 +1,3 @@
|
|||||||
import
|
import
|
||||||
testrpcmacro, testserverclient, testethcalls, testerrors
|
testrpcmacro, testserverclient, testethcalls #, testerrors
|
||||||
|
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
include ../ eth-rpc / client
|
include ../ json_rpc / client
|
||||||
|
|
||||||
proc nextId*(self: RpcClient): int64 = self.nextId
|
proc nextId*(self: RpcClient): int64 = self.nextId
|
||||||
|
|
||||||
@ -9,7 +9,7 @@ proc rawCall*(self: RpcClient, name: string,
|
|||||||
self.nextId.inc
|
self.nextId.inc
|
||||||
|
|
||||||
var s = msg & "\c\l"
|
var s = msg & "\c\l"
|
||||||
let res = await self.transp.write(s)
|
let res = await self.transport.write(s)
|
||||||
assert res == len(s)
|
assert res == len(s)
|
||||||
|
|
||||||
# completed by processMessage.
|
# completed by processMessage.
|
||||||
|
157
tests/ethhexstrings.nim
Normal file
157
tests/ethhexstrings.nim
Normal file
@ -0,0 +1,157 @@
|
|||||||
|
type
|
||||||
|
HexQuantityStr* = distinct string
|
||||||
|
HexDataStr* = distinct string
|
||||||
|
|
||||||
|
# Hex validation
|
||||||
|
|
||||||
|
template stripLeadingZeros(value: string): string =
|
||||||
|
var cidx = 0
|
||||||
|
# ignore the last character so we retain '0' on zero value
|
||||||
|
while cidx < value.len - 1 and value[cidx] == '0':
|
||||||
|
cidx.inc
|
||||||
|
value[cidx .. ^1]
|
||||||
|
|
||||||
|
proc encodeQuantity*(value: SomeUnsignedInt): string =
|
||||||
|
var hValue = value.toHex.stripLeadingZeros
|
||||||
|
result = "0x" & hValue
|
||||||
|
|
||||||
|
template hasHexHeader*(value: string | HexDataStr | HexQuantityStr): bool =
|
||||||
|
template strVal: untyped = value.string
|
||||||
|
if strVal != "" and strVal[0] == '0' and strVal[1] in {'x', 'X'} and strVal.len > 2: true
|
||||||
|
else: false
|
||||||
|
|
||||||
|
template isHexChar*(c: char): bool =
|
||||||
|
if c notin {'0'..'9'} and
|
||||||
|
c notin {'a'..'f'} and
|
||||||
|
c notin {'A'..'F'}: false
|
||||||
|
else: true
|
||||||
|
|
||||||
|
proc validate*(value: HexQuantityStr): bool =
|
||||||
|
template strVal: untyped = value.string
|
||||||
|
if not value.hasHexHeader:
|
||||||
|
return false
|
||||||
|
# No leading zeros
|
||||||
|
if strVal[2] == '0': return false
|
||||||
|
for i in 2..<strVal.len:
|
||||||
|
let c = strVal[i]
|
||||||
|
if not c.isHexChar:
|
||||||
|
return false
|
||||||
|
return true
|
||||||
|
|
||||||
|
proc validate*(value: HexDataStr): bool =
|
||||||
|
template strVal: untyped = value.string
|
||||||
|
if not value.hasHexHeader:
|
||||||
|
return false
|
||||||
|
# Leading zeros are allowed
|
||||||
|
for i in 2..<strVal.len:
|
||||||
|
let c = strVal[i]
|
||||||
|
if not c.isHexChar:
|
||||||
|
return false
|
||||||
|
# Must be even number of digits
|
||||||
|
if strVal.len mod 2 != 0: return false
|
||||||
|
return true
|
||||||
|
|
||||||
|
# Initialisation
|
||||||
|
|
||||||
|
template hexDataStr*(value: string): HexDataStr = value.HexDataStr
|
||||||
|
template hexQuantityStr*(value: string): HexQuantityStr = value.HexQuantityStr
|
||||||
|
|
||||||
|
# Converters
|
||||||
|
|
||||||
|
import json
|
||||||
|
from ../rpcserver import expect
|
||||||
|
|
||||||
|
proc `%`*(value: HexDataStr): JsonNode =
|
||||||
|
if not value.validate:
|
||||||
|
raise newException(ValueError, "HexDataStr: Invalid hex for Ethereum: " & value.string)
|
||||||
|
else:
|
||||||
|
result = %(value.string)
|
||||||
|
|
||||||
|
proc `%`*(value: HexQuantityStr): JsonNode =
|
||||||
|
if not value.validate:
|
||||||
|
raise newException(ValueError, "HexQuantityStr: Invalid hex for Ethereum: " & value.string)
|
||||||
|
else:
|
||||||
|
result = %(value.string)
|
||||||
|
|
||||||
|
proc fromJson*(n: JsonNode, argName: string, result: var HexDataStr) =
|
||||||
|
# Note that '0x' is stripped after validation
|
||||||
|
n.kind.expect(JString, argName)
|
||||||
|
let hexStr = n.getStr()
|
||||||
|
if not hexStr.hexDataStr.validate:
|
||||||
|
raise newException(ValueError, "Parameter \"" & argName & "\" value is not valid as a Ethereum data \"" & hexStr & "\"")
|
||||||
|
result = hexStr[2..hexStr.high].hexDataStr
|
||||||
|
|
||||||
|
proc fromJson*(n: JsonNode, argName: string, result: var HexQuantityStr) =
|
||||||
|
# Note that '0x' is stripped after validation
|
||||||
|
n.kind.expect(JString, argName)
|
||||||
|
let hexStr = n.getStr()
|
||||||
|
if not hexStr.hexQuantityStr.validate:
|
||||||
|
raise newException(ValueError, "Parameter \"" & argName & "\" value is not valid as an Ethereum hex quantity \"" & hexStr & "\"")
|
||||||
|
result = hexStr[2..hexStr.high].hexQuantityStr
|
||||||
|
|
||||||
|
# testing
|
||||||
|
|
||||||
|
when isMainModule:
|
||||||
|
import unittest
|
||||||
|
suite "Hex quantity":
|
||||||
|
test "Empty string":
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = ""
|
||||||
|
x = hexQuantityStr source
|
||||||
|
check %x == %source
|
||||||
|
test "Even length":
|
||||||
|
let
|
||||||
|
source = "0x123"
|
||||||
|
x = hexQuantityStr source
|
||||||
|
check %x == %source
|
||||||
|
test "Odd length":
|
||||||
|
let
|
||||||
|
source = "0x123"
|
||||||
|
x = hexQuantityStr"0x123"
|
||||||
|
check %x == %source
|
||||||
|
test "Missing header":
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "1234"
|
||||||
|
x = hexQuantityStr source
|
||||||
|
check %x != %source
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "01234"
|
||||||
|
x = hexQuantityStr source
|
||||||
|
check %x != %source
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "x1234"
|
||||||
|
x = hexQuantityStr source
|
||||||
|
check %x != %source
|
||||||
|
|
||||||
|
suite "Hex data":
|
||||||
|
test "Even length":
|
||||||
|
let
|
||||||
|
source = "0x1234"
|
||||||
|
x = hexDataStr source
|
||||||
|
check %x == %source
|
||||||
|
test "Odd length":
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "0x123"
|
||||||
|
x = hexDataStr source
|
||||||
|
check %x != %source
|
||||||
|
test "Missing header":
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "1234"
|
||||||
|
x = hexDataStr source
|
||||||
|
check %x != %source
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "01234"
|
||||||
|
x = hexDataStr source
|
||||||
|
check %x != %source
|
||||||
|
expect ValueError:
|
||||||
|
let
|
||||||
|
source = "x1234"
|
||||||
|
x = hexDataStr source
|
||||||
|
check %x != %source
|
@ -1,4 +1,4 @@
|
|||||||
import ../rpcserver, nimcrypto, json, stint, strutils, ethtypes, stintjson
|
import ../rpcserver, nimcrypto, json, stint, strutils, ethtypes, stintjson, ethhexstrings
|
||||||
|
|
||||||
#[
|
#[
|
||||||
For details on available RPC calls, see: https://github.com/ethereum/wiki/wiki/JSON-RPC
|
For details on available RPC calls, see: https://github.com/ethereum/wiki/wiki/JSON-RPC
|
||||||
@ -31,19 +31,16 @@ proc addEthRpcs*(server: RpcServer) =
|
|||||||
## Returns the current client version.
|
## Returns the current client version.
|
||||||
result = "Nimbus-RPC-Test"
|
result = "Nimbus-RPC-Test"
|
||||||
|
|
||||||
server.rpc("web3_sha3") do(data: string) -> string:
|
server.rpc("web3_sha3") do(data: HexDataStr) -> HexDataStr:
|
||||||
## Returns Keccak-256 (not the standardized SHA3-256) of the given data.
|
## Returns Keccak-256 (not the standardized SHA3-256) of the given data.
|
||||||
##
|
##
|
||||||
## data: the data to convert into a SHA3 hash.
|
## data: the data to convert into a SHA3 hash.
|
||||||
## Returns the SHA3 result of the given string.
|
## Returns the SHA3 result of the given string.
|
||||||
# TODO: Capture error on malformed input
|
# TODO: Capture error on malformed input
|
||||||
var rawData: seq[byte]
|
var rawData: seq[byte]
|
||||||
if data.len > 2 and data[0] == '0' and data[1] in ['x', 'X']:
|
rawData = data.string.fromHex
|
||||||
rawData = data[2..data.high].fromHex
|
|
||||||
else:
|
|
||||||
rawData = data.fromHex
|
|
||||||
# data will have 0x prefix
|
# data will have 0x prefix
|
||||||
result = "0x" & $keccak_256.digest(rawData)
|
result = hexDataStr "0x" & $keccak_256.digest(rawData)
|
||||||
|
|
||||||
server.rpc("net_version") do() -> string:
|
server.rpc("net_version") do() -> string:
|
||||||
## Returns string of the current network id:
|
## Returns string of the current network id:
|
||||||
|
@ -3,11 +3,11 @@
|
|||||||
allow unchecked and unformatted calls.
|
allow unchecked and unformatted calls.
|
||||||
]#
|
]#
|
||||||
|
|
||||||
import unittest, debugclient, ../rpcserver
|
import unittest, debugclient, ../rpcsockets
|
||||||
import strformat, chronicles
|
import strformat, chronicles
|
||||||
|
|
||||||
var server = newRpcServer("localhost", 8547.Port)
|
var server = newRpcStreamServer("localhost", 8547.Port)
|
||||||
var client = newRpcClient()
|
var client = newRpcStreamClient()
|
||||||
|
|
||||||
server.start()
|
server.start()
|
||||||
waitFor client.connect("localhost", Port(8547))
|
waitFor client.connect("localhost", Port(8547))
|
||||||
@ -15,6 +15,10 @@ waitFor client.connect("localhost", Port(8547))
|
|||||||
server.rpc("rpc") do(a: int, b: int):
|
server.rpc("rpc") do(a: int, b: int):
|
||||||
result = %(&"a: {a}, b: {b}")
|
result = %(&"a: {a}, b: {b}")
|
||||||
|
|
||||||
|
server.rpc("makeError"):
|
||||||
|
if true:
|
||||||
|
raise newException(ValueError, "Test")
|
||||||
|
|
||||||
proc testMissingRpc: Future[Response] {.async.} =
|
proc testMissingRpc: Future[Response] {.async.} =
|
||||||
var fut = client.call("phantomRpc", %[])
|
var fut = client.call("phantomRpc", %[])
|
||||||
result = await fut
|
result = await fut
|
||||||
@ -33,22 +37,44 @@ proc testMalformed: Future[Response] {.async.} =
|
|||||||
if fut.finished: result = fut.read()
|
if fut.finished: result = fut.read()
|
||||||
else: result = (true, %"Timeout")
|
else: result = (true, %"Timeout")
|
||||||
|
|
||||||
|
proc testRaise: Future[Response] {.async.} =
|
||||||
|
var fut = client.call("rpcMakeError", %[])
|
||||||
|
result = await fut
|
||||||
|
|
||||||
suite "RPC Errors":
|
suite "RPC Errors":
|
||||||
# Note: We don't expect a exceptions for most of the tests,
|
# Note: We don't expect a exceptions for most of the tests,
|
||||||
# because the server should respond with the error in json
|
# because the server should respond with the error in json
|
||||||
test "Missing RPC":
|
test "Missing RPC":
|
||||||
|
#expect ValueError:
|
||||||
|
try:
|
||||||
let res = waitFor testMissingRpc()
|
let res = waitFor testMissingRpc()
|
||||||
check res.error == true and
|
check res.error == true and
|
||||||
res.result["message"] == %"Method not found" and
|
res.result["message"] == %"Method not found" and
|
||||||
res.result["data"] == %"phantomRpc is not a registered method."
|
res.result["data"] == %"phantomRpc is not a registered method."
|
||||||
|
except:
|
||||||
|
echo "Error ", getCurrentExceptionMsg()
|
||||||
|
|
||||||
test "Incorrect json version":
|
test "Incorrect json version":
|
||||||
|
#expect ValueError:
|
||||||
|
try:
|
||||||
let res = waitFor testInvalidJsonVer()
|
let res = waitFor testInvalidJsonVer()
|
||||||
check res.error == true and res.result["message"] == %"JSON 2.0 required"
|
check res.error == true and res.result["message"] == %"JSON 2.0 required"
|
||||||
|
except:
|
||||||
|
echo "Error ", getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
test "Raising exceptions":
|
||||||
|
#expect ValueError:
|
||||||
|
try:
|
||||||
|
let res = waitFor testRaise()
|
||||||
|
except:
|
||||||
|
echo "Error ", getCurrentExceptionMsg()
|
||||||
|
|
||||||
test "Malformed json":
|
test "Malformed json":
|
||||||
# TODO: We time out here because the server won't be able to
|
# TODO: We time out here because the server won't be able to
|
||||||
# find an id to return to us, so we cannot complete the future.
|
# find an id to return to us, so we cannot complete the future.
|
||||||
|
try:
|
||||||
let res = waitFor testMalformed()
|
let res = waitFor testMalformed()
|
||||||
check res.error == true and res.result == %"Timeout"
|
check res.error == true and res.result == %"Timeout"
|
||||||
|
except:
|
||||||
|
echo "Error ", getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
@ -1,14 +1,14 @@
|
|||||||
import unittest, json, tables
|
import unittest, json, tables
|
||||||
import ../rpcclient, ../rpcserver
|
import ../rpcclient, ../rpcsockets
|
||||||
import stint, ethtypes, ethprocs, stintjson
|
import stint, ethtypes, ethprocs, stintjson, nimcrypto, ethhexstrings, chronicles
|
||||||
|
|
||||||
from os import getCurrentDir, DirSep
|
from os import getCurrentDir, DirSep
|
||||||
from strutils import rsplit
|
from strutils import rsplit
|
||||||
template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]
|
template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]
|
||||||
|
|
||||||
var
|
var
|
||||||
server = newRpcServer("localhost", Port(8546))
|
server = newRpcStreamServer("localhost", Port(8546))
|
||||||
client = newRpcClient()
|
client = newRpcStreamClient()
|
||||||
|
|
||||||
## Generate Ethereum server RPCs
|
## Generate Ethereum server RPCs
|
||||||
server.addEthRpcs()
|
server.addEthRpcs()
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
import unittest, json, tables
|
import unittest, json, tables, chronicles
|
||||||
import ../rpcserver
|
import ../rpcsockets
|
||||||
|
|
||||||
type
|
type
|
||||||
# some nested types to check object parsing
|
# some nested types to check object parsing
|
||||||
@ -27,7 +27,7 @@ let
|
|||||||
},
|
},
|
||||||
"c": %1.23}
|
"c": %1.23}
|
||||||
|
|
||||||
var s = newRpcServer(["localhost:8545"])
|
var s = newRpcStreamServer(["localhost:8545"])
|
||||||
|
|
||||||
# RPC definitions
|
# RPC definitions
|
||||||
|
|
||||||
@ -67,14 +67,14 @@ s.rpc("rpc.testreturns") do() -> int:
|
|||||||
suite "Server types":
|
suite "Server types":
|
||||||
|
|
||||||
test "On macro registration":
|
test "On macro registration":
|
||||||
check s.procs.hasKey("rpc.simplepath")
|
check s.hasMethod("rpc.simplepath")
|
||||||
check s.procs.hasKey("rpc.differentparams")
|
check s.hasMethod("rpc.differentparams")
|
||||||
check s.procs.hasKey("rpc.arrayparam")
|
check s.hasMethod("rpc.arrayparam")
|
||||||
check s.procs.hasKey("rpc.seqparam")
|
check s.hasMethod("rpc.seqparam")
|
||||||
check s.procs.hasKey("rpc.objparam")
|
check s.hasMethod("rpc.objparam")
|
||||||
check s.procs.hasKey("rpc.returntypesimple")
|
check s.hasMethod("rpc.returntypesimple")
|
||||||
check s.procs.hasKey("rpc.returntypecomplex")
|
check s.hasMethod("rpc.returntypecomplex")
|
||||||
check s.procs.hasKey("rpc.testreturns")
|
check s.hasMethod("rpc.testreturns")
|
||||||
|
|
||||||
test "Simple paths":
|
test "Simple paths":
|
||||||
let r = waitFor rpcSimplePath(%[])
|
let r = waitFor rpcSimplePath(%[])
|
||||||
|
@ -1,8 +1,8 @@
|
|||||||
import unittest, json
|
import unittest, json, chronicles
|
||||||
import ../rpcclient, ../rpcserver
|
import ../rpcclient, ../rpcsockets
|
||||||
|
|
||||||
var srv = newRpcServer(["localhost:8545"])
|
var srv = newRpcStreamServer(["localhost:8545"])
|
||||||
var client = newRpcClient()
|
var client = newRpcStreamClient()
|
||||||
|
|
||||||
# Create RPC on server
|
# Create RPC on server
|
||||||
srv.rpc("myProc") do(input: string, data: array[0..3, int]):
|
srv.rpc("myProc") do(input: string, data: array[0..3, int]):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user