2015-01-06 11:13:57 +00:00
|
|
|
/*
|
|
|
|
This file is part of go-ethereum
|
|
|
|
|
|
|
|
go-ethereum is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
go-ethereum is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
/**
|
|
|
|
* @authors
|
|
|
|
* Jeffrey Wilcke <i@jev.io>
|
|
|
|
*/
|
2014-09-30 21:26:16 +00:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
2014-12-14 18:23:48 +00:00
|
|
|
"github.com/ethereum/go-ethereum/eth"
|
2014-10-23 13:01:27 +00:00
|
|
|
"github.com/ethereum/go-ethereum/ethutil"
|
2014-12-23 14:46:46 +00:00
|
|
|
"github.com/ethereum/go-ethereum/logger"
|
2014-10-23 13:01:27 +00:00
|
|
|
"github.com/ethereum/go-ethereum/websocket"
|
2014-10-31 13:30:08 +00:00
|
|
|
"github.com/ethereum/go-ethereum/xeth"
|
2014-09-30 21:26:16 +00:00
|
|
|
)
|
|
|
|
|
2014-12-23 14:46:46 +00:00
|
|
|
var wslogger = logger.NewLogger("WS")
|
|
|
|
|
2014-09-30 21:26:16 +00:00
|
|
|
func args(v ...interface{}) []interface{} {
|
|
|
|
return v
|
|
|
|
}
|
|
|
|
|
|
|
|
type WebSocketServer struct {
|
|
|
|
ethereum *eth.Ethereum
|
|
|
|
filterCallbacks map[int][]int
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewWebSocketServer(eth *eth.Ethereum) *WebSocketServer {
|
|
|
|
return &WebSocketServer{eth, make(map[int][]int)}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *WebSocketServer) Serv() {
|
2014-10-31 13:30:08 +00:00
|
|
|
pipe := xeth.NewJSXEth(self.ethereum)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
|
|
|
wsServ := websocket.NewServer("/eth", ":40404")
|
|
|
|
wsServ.MessageFunc(func(c *websocket.Client, msg *websocket.Message) {
|
|
|
|
switch msg.Call {
|
|
|
|
case "compile":
|
|
|
|
data := ethutil.NewValue(msg.Args)
|
|
|
|
bcode, err := ethutil.Compile(data.Get(0).Str(), false)
|
|
|
|
if err != nil {
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(args(nil, err.Error()), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
code := ethutil.Bytes2Hex(bcode)
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(args(code, nil), msg.Id)
|
|
|
|
case "eth_blockByNumber":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
|
|
|
block := pipe.BlockByNumber(int32(args.Get(0).Uint()))
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(block, msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_blockByHash":
|
|
|
|
args := msg.Arguments()
|
|
|
|
|
|
|
|
c.Write(pipe.BlockByHash(args.Get(0).Str()), msg.Id)
|
|
|
|
|
|
|
|
case "eth_transact":
|
2014-09-30 21:26:16 +00:00
|
|
|
if mp, ok := msg.Args[0].(map[string]interface{}); ok {
|
|
|
|
object := mapToTxParams(mp)
|
|
|
|
c.Write(
|
2015-01-06 10:44:22 +00:00
|
|
|
args(pipe.Transact(pipe.Key().PrivateKey, object["to"], object["value"], object["gas"], object["gasPrice"], object["data"])),
|
|
|
|
msg.Id,
|
2014-09-30 21:26:16 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
}
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_gasPrice":
|
|
|
|
c.Write("10000000000000", msg.Id)
|
|
|
|
case "eth_coinbase":
|
|
|
|
c.Write(pipe.CoinBase(), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_listening":
|
|
|
|
c.Write(pipe.IsListening(), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_mining":
|
|
|
|
c.Write(pipe.IsMining(), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_peerCount":
|
|
|
|
c.Write(pipe.PeerCount(), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_countAt":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(pipe.TxCountAt(args.Get(0).Str()), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_codeAt":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(len(pipe.CodeAt(args.Get(0).Str())), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_storageAt":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(pipe.StorageAt(args.Get(0).Str(), args.Get(1).Str()), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_balanceAt":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(pipe.BalanceAt(args.Get(0).Str()), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_secretToAddress":
|
2014-09-30 21:26:16 +00:00
|
|
|
args := msg.Arguments()
|
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
c.Write(pipe.SecretToAddress(args.Get(0).Str()), msg.Id)
|
2014-09-30 21:26:16 +00:00
|
|
|
|
2015-01-06 10:44:22 +00:00
|
|
|
case "eth_newFilter":
|
|
|
|
case "eth_newFilterString":
|
|
|
|
case "eth_messages":
|
2014-09-30 21:26:16 +00:00
|
|
|
// TODO
|
|
|
|
}
|
|
|
|
|
|
|
|
})
|
|
|
|
|
|
|
|
wsServ.Listen()
|
|
|
|
}
|
|
|
|
|
|
|
|
func StartWebSockets(eth *eth.Ethereum) {
|
2014-12-23 14:46:46 +00:00
|
|
|
wslogger.Infoln("Starting WebSockets")
|
|
|
|
|
2014-09-30 21:26:16 +00:00
|
|
|
sock := NewWebSocketServer(eth)
|
|
|
|
go sock.Serv()
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO This is starting to become a generic method. Move to utils
|
|
|
|
func mapToTxParams(object map[string]interface{}) map[string]string {
|
|
|
|
// Default values
|
|
|
|
if object["from"] == nil {
|
|
|
|
object["from"] = ""
|
|
|
|
}
|
|
|
|
if object["to"] == nil {
|
|
|
|
object["to"] = ""
|
|
|
|
}
|
|
|
|
if object["value"] == nil {
|
|
|
|
object["value"] = ""
|
|
|
|
}
|
|
|
|
if object["gas"] == nil {
|
|
|
|
object["gas"] = ""
|
|
|
|
}
|
|
|
|
if object["gasPrice"] == nil {
|
|
|
|
object["gasPrice"] = ""
|
|
|
|
}
|
|
|
|
|
|
|
|
var dataStr string
|
|
|
|
var data []string
|
|
|
|
if str, ok := object["data"].(string); ok {
|
|
|
|
data = []string{str}
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, str := range data {
|
|
|
|
if ethutil.IsHex(str) {
|
|
|
|
str = str[2:]
|
|
|
|
|
|
|
|
if len(str) != 64 {
|
|
|
|
str = ethutil.LeftPadString(str, 64)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
str = ethutil.Bytes2Hex(ethutil.LeftPadBytes(ethutil.Big(str).Bytes(), 32))
|
|
|
|
}
|
|
|
|
|
|
|
|
dataStr += str
|
|
|
|
}
|
|
|
|
object["data"] = dataStr
|
|
|
|
|
|
|
|
conv := make(map[string]string)
|
|
|
|
for key, value := range object {
|
|
|
|
if v, ok := value.(string); ok {
|
|
|
|
conv[key] = v
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return conv
|
|
|
|
}
|