forked from mirror/go-ethereum
Bas van Kervel
9 years ago
committed by
Jeffrey Wilcke
parent
f2ab351e8d
commit
19b2640e89
@ -1,27 +0,0 @@ |
||||
Copyright (c) 2012 The Go Authors. All rights reserved. |
||||
|
||||
Redistribution and use in source and binary forms, with or without |
||||
modification, are permitted provided that the following conditions are |
||||
met: |
||||
|
||||
* Redistributions of source code must retain the above copyright |
||||
notice, this list of conditions and the following disclaimer. |
||||
* Redistributions in binary form must reproduce the above |
||||
copyright notice, this list of conditions and the following disclaimer |
||||
in the documentation and/or other materials provided with the |
||||
distribution. |
||||
* Neither the name of Google Inc. nor the names of its |
||||
contributors may be used to endorse or promote products derived from |
||||
this software without specific prior written permission. |
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
@ -1,14 +0,0 @@ |
||||
### Extensions to the "os" package. |
||||
|
||||
## Find the current Executable and ExecutableFolder. |
||||
|
||||
There is sometimes utility in finding the current executable file |
||||
that is running. This can be used for upgrading the current executable |
||||
or finding resources located relative to the executable file. |
||||
|
||||
Multi-platform and supports: |
||||
* Linux |
||||
* OS X |
||||
* Windows |
||||
* Plan 9 |
||||
* BSDs. |
||||
@ -1,27 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// Extensions to the standard "os" package.
|
||||
package osext |
||||
|
||||
import "path/filepath" |
||||
|
||||
// Executable returns an absolute path that can be used to
|
||||
// re-invoke the current program.
|
||||
// It may not be valid after the current program exits.
|
||||
func Executable() (string, error) { |
||||
p, err := executable() |
||||
return filepath.Clean(p), err |
||||
} |
||||
|
||||
// Returns same path as Executable, returns just the folder
|
||||
// path. Excludes the executable name.
|
||||
func ExecutableFolder() (string, error) { |
||||
p, err := Executable() |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
folder, _ := filepath.Split(p) |
||||
return folder, nil |
||||
} |
||||
@ -1,20 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package osext |
||||
|
||||
import ( |
||||
"os" |
||||
"strconv" |
||||
"syscall" |
||||
) |
||||
|
||||
func executable() (string, error) { |
||||
f, err := os.Open("/proc/" + strconv.Itoa(os.Getpid()) + "/text") |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
defer f.Close() |
||||
return syscall.Fd2path(int(f.Fd())) |
||||
} |
||||
@ -1,28 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build linux netbsd openbsd solaris dragonfly
|
||||
|
||||
package osext |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
"os" |
||||
"runtime" |
||||
) |
||||
|
||||
func executable() (string, error) { |
||||
switch runtime.GOOS { |
||||
case "linux": |
||||
return os.Readlink("/proc/self/exe") |
||||
case "netbsd": |
||||
return os.Readlink("/proc/curproc/exe") |
||||
case "openbsd", "dragonfly": |
||||
return os.Readlink("/proc/curproc/file") |
||||
case "solaris": |
||||
return os.Readlink(fmt.Sprintf("/proc/%d/path/a.out", os.Getpid())) |
||||
} |
||||
return "", errors.New("ExecPath not implemented for " + runtime.GOOS) |
||||
} |
||||
@ -1,79 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build darwin freebsd
|
||||
|
||||
package osext |
||||
|
||||
import ( |
||||
"os" |
||||
"path/filepath" |
||||
"runtime" |
||||
"syscall" |
||||
"unsafe" |
||||
) |
||||
|
||||
var initCwd, initCwdErr = os.Getwd() |
||||
|
||||
func executable() (string, error) { |
||||
var mib [4]int32 |
||||
switch runtime.GOOS { |
||||
case "freebsd": |
||||
mib = [4]int32{1 /* CTL_KERN */, 14 /* KERN_PROC */, 12 /* KERN_PROC_PATHNAME */, -1} |
||||
case "darwin": |
||||
mib = [4]int32{1 /* CTL_KERN */, 38 /* KERN_PROCARGS */, int32(os.Getpid()), -1} |
||||
} |
||||
|
||||
n := uintptr(0) |
||||
// Get length.
|
||||
_, _, errNum := syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, 0, uintptr(unsafe.Pointer(&n)), 0, 0) |
||||
if errNum != 0 { |
||||
return "", errNum |
||||
} |
||||
if n == 0 { // This shouldn't happen.
|
||||
return "", nil |
||||
} |
||||
buf := make([]byte, n) |
||||
_, _, errNum = syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, uintptr(unsafe.Pointer(&buf[0])), uintptr(unsafe.Pointer(&n)), 0, 0) |
||||
if errNum != 0 { |
||||
return "", errNum |
||||
} |
||||
if n == 0 { // This shouldn't happen.
|
||||
return "", nil |
||||
} |
||||
for i, v := range buf { |
||||
if v == 0 { |
||||
buf = buf[:i] |
||||
break |
||||
} |
||||
} |
||||
var err error |
||||
execPath := string(buf) |
||||
// execPath will not be empty due to above checks.
|
||||
// Try to get the absolute path if the execPath is not rooted.
|
||||
if execPath[0] != '/' { |
||||
execPath, err = getAbs(execPath) |
||||
if err != nil { |
||||
return execPath, err |
||||
} |
||||
} |
||||
// For darwin KERN_PROCARGS may return the path to a symlink rather than the
|
||||
// actual executable.
|
||||
if runtime.GOOS == "darwin" { |
||||
if execPath, err = filepath.EvalSymlinks(execPath); err != nil { |
||||
return execPath, err |
||||
} |
||||
} |
||||
return execPath, nil |
||||
} |
||||
|
||||
func getAbs(execPath string) (string, error) { |
||||
if initCwdErr != nil { |
||||
return execPath, initCwdErr |
||||
} |
||||
// The execPath may begin with a "../" or a "./" so clean it first.
|
||||
// Join the two paths, trailing and starting slashes undetermined, so use
|
||||
// the generic Join function.
|
||||
return filepath.Join(initCwd, filepath.Clean(execPath)), nil |
||||
} |
||||
@ -1,79 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build darwin linux freebsd netbsd windows
|
||||
|
||||
package osext |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
oexec "os/exec" |
||||
"path/filepath" |
||||
"runtime" |
||||
"testing" |
||||
) |
||||
|
||||
const execPath_EnvVar = "OSTEST_OUTPUT_EXECPATH" |
||||
|
||||
func TestExecPath(t *testing.T) { |
||||
ep, err := Executable() |
||||
if err != nil { |
||||
t.Fatalf("ExecPath failed: %v", err) |
||||
} |
||||
// we want fn to be of the form "dir/prog"
|
||||
dir := filepath.Dir(filepath.Dir(ep)) |
||||
fn, err := filepath.Rel(dir, ep) |
||||
if err != nil { |
||||
t.Fatalf("filepath.Rel: %v", err) |
||||
} |
||||
cmd := &oexec.Cmd{} |
||||
// make child start with a relative program path
|
||||
cmd.Dir = dir |
||||
cmd.Path = fn |
||||
// forge argv[0] for child, so that we can verify we could correctly
|
||||
// get real path of the executable without influenced by argv[0].
|
||||
cmd.Args = []string{"-", "-test.run=XXXX"} |
||||
cmd.Env = []string{fmt.Sprintf("%s=1", execPath_EnvVar)} |
||||
out, err := cmd.CombinedOutput() |
||||
if err != nil { |
||||
t.Fatalf("exec(self) failed: %v", err) |
||||
} |
||||
outs := string(out) |
||||
if !filepath.IsAbs(outs) { |
||||
t.Fatalf("Child returned %q, want an absolute path", out) |
||||
} |
||||
if !sameFile(outs, ep) { |
||||
t.Fatalf("Child returned %q, not the same file as %q", out, ep) |
||||
} |
||||
} |
||||
|
||||
func sameFile(fn1, fn2 string) bool { |
||||
fi1, err := os.Stat(fn1) |
||||
if err != nil { |
||||
return false |
||||
} |
||||
fi2, err := os.Stat(fn2) |
||||
if err != nil { |
||||
return false |
||||
} |
||||
return os.SameFile(fi1, fi2) |
||||
} |
||||
|
||||
func init() { |
||||
if e := os.Getenv(execPath_EnvVar); e != "" { |
||||
// first chdir to another path
|
||||
dir := "/" |
||||
if runtime.GOOS == "windows" { |
||||
dir = filepath.VolumeName(".") |
||||
} |
||||
os.Chdir(dir) |
||||
if ep, err := Executable(); err != nil { |
||||
fmt.Fprint(os.Stderr, "ERROR: ", err) |
||||
} else { |
||||
fmt.Fprint(os.Stderr, ep) |
||||
} |
||||
os.Exit(0) |
||||
} |
||||
} |
||||
@ -1,34 +0,0 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package osext |
||||
|
||||
import ( |
||||
"syscall" |
||||
"unicode/utf16" |
||||
"unsafe" |
||||
) |
||||
|
||||
var ( |
||||
kernel = syscall.MustLoadDLL("kernel32.dll") |
||||
getModuleFileNameProc = kernel.MustFindProc("GetModuleFileNameW") |
||||
) |
||||
|
||||
// GetModuleFileName() with hModule = NULL
|
||||
func executable() (exePath string, err error) { |
||||
return getModuleFileName() |
||||
} |
||||
|
||||
func getModuleFileName() (string, error) { |
||||
var n uint32 |
||||
b := make([]uint16, syscall.MAX_PATH) |
||||
size := uint32(len(b)) |
||||
|
||||
r0, _, e1 := getModuleFileNameProc.Call(0, uintptr(unsafe.Pointer(&b[0])), uintptr(size)) |
||||
n = uint32(r0) |
||||
if n == 0 { |
||||
return "", e1 |
||||
} |
||||
return string(utf16.Decode(b[0:n])), nil |
||||
} |
||||
@ -1,4 +0,0 @@ |
||||
language: go |
||||
go: |
||||
- 1.3 |
||||
- 1.4 |
||||
@ -1,19 +0,0 @@ |
||||
Copyright (c) 2014 Olivier Poitrey <rs@dailymotion.com> |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is furnished |
||||
to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
@ -1,84 +0,0 @@ |
||||
# Go CORS handler [](https://godoc.org/github.com/rs/cors) [](https://raw.githubusercontent.com/rs/cors/master/LICENSE) [](https://travis-ci.org/rs/cors) |
||||
|
||||
CORS is a `net/http` handler implementing [Cross Origin Resource Sharing W3 specification](http://www.w3.org/TR/cors/) in Golang. |
||||
|
||||
## Getting Started |
||||
|
||||
After installing Go and setting up your [GOPATH](http://golang.org/doc/code.html#GOPATH), create your first `.go` file. We'll call it `server.go`. |
||||
|
||||
```go |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
// cors.Default() setup the middleware with default options being |
||||
// all origins accepted with simple methods (GET, POST). See |
||||
// documentation below for more options. |
||||
handler = cors.Default().Handler(h) |
||||
http.ListenAndServe(":8080", handler) |
||||
} |
||||
``` |
||||
|
||||
Install `cors`: |
||||
|
||||
go get github.com/rs/cors |
||||
|
||||
Then run your server: |
||||
|
||||
go run server.go |
||||
|
||||
The server now runs on `localhost:8080`: |
||||
|
||||
$ curl -D - -H 'Origin: http://foo.com' http://localhost:8080/ |
||||
HTTP/1.1 200 OK |
||||
Access-Control-Allow-Origin: foo.com |
||||
Content-Type: application/json |
||||
Date: Sat, 25 Oct 2014 03:43:57 GMT |
||||
Content-Length: 18 |
||||
|
||||
{"hello": "world"} |
||||
|
||||
### More Examples |
||||
|
||||
* `net/http`: [examples/nethttp/server.go](https://github.com/rs/cors/blob/master/examples/nethttp/server.go) |
||||
* [Goji](https://goji.io): [examples/goji/server.go](https://github.com/rs/cors/blob/master/examples/goji/server.go) |
||||
* [Martini](http://martini.codegangsta.io): [examples/martini/server.go](https://github.com/rs/cors/blob/master/examples/martini/server.go) |
||||
* [Negroni](https://github.com/codegangsta/negroni): [examples/negroni/server.go](https://github.com/rs/cors/blob/master/examples/negroni/server.go) |
||||
* [Alice](https://github.com/justinas/alice): [examples/alice/server.go](https://github.com/rs/cors/blob/master/examples/alice/server.go) |
||||
|
||||
## Parameters |
||||
|
||||
Parameters are passed to the middleware thru the `cors.New` method as follow: |
||||
|
||||
```go |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
AllowCredentials: true, |
||||
}) |
||||
|
||||
// Insert the middleware |
||||
handler = c.Handler(handler) |
||||
``` |
||||
|
||||
* **AllowedOrigins** `[]string`: A list of origins a cross-domain request can be executed from. If the special `*` value is present in the list, all origins will be allowed. The default value is `*`. |
||||
* **AllowedMethods** `[]string`: A list of methods the client is allowed to use with cross-domain requests. |
||||
* **AllowedHeaders** `[]string`: A list of non simple headers the client is allowed to use with cross-domain requests. Default value is simple methods (`GET` and `POST`) |
||||
* **ExposedHeaders** `[]string`: Indicates which headers are safe to expose to the API of a CORS API specification |
||||
* **AllowCredentials** `bool`: Indicates whether the request can include user credentials like cookies, HTTP authentication or client side SSL certificates. The default is `false`. |
||||
* **MaxAge** `int`: Indicates how long (in seconds) the results of a preflight request can be cached. The default is `0` which stands for no max age. |
||||
|
||||
See [API documentation](http://godoc.org/github.com/rs/cors) for more info. |
||||
|
||||
## Licenses |
||||
|
||||
All source code is licensed under the [MIT License](https://raw.github.com/rs/cors/master/LICENSE). |
||||
@ -1,37 +0,0 @@ |
||||
package cors |
||||
|
||||
import ( |
||||
"net/http" |
||||
"net/http/httptest" |
||||
"testing" |
||||
) |
||||
|
||||
func BenchmarkWithout(b *testing.B) { |
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
|
||||
for i := 0; i < b.N; i++ { |
||||
testHandler.ServeHTTP(res, req) |
||||
} |
||||
} |
||||
|
||||
func BenchmarkDefault(b *testing.B) { |
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
handler := Default() |
||||
|
||||
for i := 0; i < b.N; i++ { |
||||
handler.Handler(testHandler).ServeHTTP(res, req) |
||||
} |
||||
} |
||||
|
||||
func BenchmarkPreflight(b *testing.B) { |
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
handler := Default() |
||||
|
||||
for i := 0; i < b.N; i++ { |
||||
handler.Handler(testHandler).ServeHTTP(res, req) |
||||
} |
||||
} |
||||
@ -1,308 +0,0 @@ |
||||
/* |
||||
Package cors is net/http handler to handle CORS related requests |
||||
as defined by http://www.w3.org/TR/cors/
|
||||
|
||||
You can configure it by passing an option struct to cors.New: |
||||
|
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"foo.com"}, |
||||
AllowedMethods: []string{"GET", "POST", "DELETE"}, |
||||
AllowCredentials: true, |
||||
}) |
||||
|
||||
Then insert the handler in the chain: |
||||
|
||||
handler = c.Handler(handler) |
||||
|
||||
See Options documentation for more options. |
||||
|
||||
The resulting handler is a standard net/http handler. |
||||
*/ |
||||
package cors |
||||
|
||||
import ( |
||||
"log" |
||||
"net/http" |
||||
"os" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
// Options is a configuration container to setup the CORS middleware.
|
||||
type Options struct { |
||||
// AllowedOrigins is a list of origins a cross-domain request can be executed from.
|
||||
// If the special "*" value is present in the list, all origins will be allowed.
|
||||
// Default value is ["*"]
|
||||
AllowedOrigins []string |
||||
// AllowedMethods is a list of methods the client is allowed to use with
|
||||
// cross-domain requests. Default value is simple methods (GET and POST)
|
||||
AllowedMethods []string |
||||
// AllowedHeaders is list of non simple headers the client is allowed to use with
|
||||
// cross-domain requests.
|
||||
// If the special "*" value is present in the list, all headers will be allowed.
|
||||
// Default value is [] but "Origin" is always appended to the list.
|
||||
AllowedHeaders []string |
||||
// ExposedHeaders indicates which headers are safe to expose to the API of a CORS
|
||||
// API specification
|
||||
ExposedHeaders []string |
||||
// AllowCredentials indicates whether the request can include user credentials like
|
||||
// cookies, HTTP authentication or client side SSL certificates.
|
||||
AllowCredentials bool |
||||
// MaxAge indicates how long (in seconds) the results of a preflight request
|
||||
// can be cached
|
||||
MaxAge int |
||||
// Debugging flag adds additional output to debug server side CORS issues
|
||||
Debug bool |
||||
// log object to use when debugging
|
||||
log *log.Logger |
||||
} |
||||
|
||||
type Cors struct { |
||||
// The CORS Options
|
||||
options Options |
||||
} |
||||
|
||||
// New creates a new Cors handler with the provided options.
|
||||
func New(options Options) *Cors { |
||||
// Normalize options
|
||||
// Note: for origins and methods matching, the spec requires a case-sensitive matching.
|
||||
// As it may error prone, we chose to ignore the spec here.
|
||||
normOptions := Options{ |
||||
AllowedOrigins: convert(options.AllowedOrigins, strings.ToLower), |
||||
AllowedMethods: convert(options.AllowedMethods, strings.ToUpper), |
||||
// Origin is always appended as some browsers will always request
|
||||
// for this header at preflight
|
||||
AllowedHeaders: convert(append(options.AllowedHeaders, "Origin"), http.CanonicalHeaderKey), |
||||
ExposedHeaders: convert(options.ExposedHeaders, http.CanonicalHeaderKey), |
||||
AllowCredentials: options.AllowCredentials, |
||||
MaxAge: options.MaxAge, |
||||
Debug: options.Debug, |
||||
log: log.New(os.Stdout, "[cors] ", log.LstdFlags), |
||||
} |
||||
if len(normOptions.AllowedOrigins) == 0 { |
||||
// Default is all origins
|
||||
normOptions.AllowedOrigins = []string{"*"} |
||||
} |
||||
if len(normOptions.AllowedHeaders) == 1 { |
||||
// Add some sensible defaults
|
||||
normOptions.AllowedHeaders = []string{"Origin", "Accept", "Content-Type"} |
||||
} |
||||
if len(normOptions.AllowedMethods) == 0 { |
||||
// Default is simple methods
|
||||
normOptions.AllowedMethods = []string{"GET", "POST"} |
||||
} |
||||
|
||||
if normOptions.Debug { |
||||
normOptions.log.Printf("Options: %v", normOptions) |
||||
} |
||||
return &Cors{ |
||||
options: normOptions, |
||||
} |
||||
} |
||||
|
||||
// Default creates a new Cors handler with default options
|
||||
func Default() *Cors { |
||||
return New(Options{}) |
||||
} |
||||
|
||||
// Handler apply the CORS specification on the request, and add relevant CORS headers
|
||||
// as necessary.
|
||||
func (cors *Cors) Handler(h http.Handler) http.Handler { |
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
if r.Method == "OPTIONS" { |
||||
cors.logf("Handler: Preflight request") |
||||
cors.handlePreflight(w, r) |
||||
// Preflight requests are standalone and should stop the chain as some other
|
||||
// middleware may not handle OPTIONS requests correctly. One typical example
|
||||
// is authentication middleware ; OPTIONS requests won't carry authentication
|
||||
// headers (see #1)
|
||||
} else { |
||||
cors.logf("Handler: Actual request") |
||||
cors.handleActualRequest(w, r) |
||||
h.ServeHTTP(w, r) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
// Martini compatible handler
|
||||
func (cors *Cors) HandlerFunc(w http.ResponseWriter, r *http.Request) { |
||||
if r.Method == "OPTIONS" { |
||||
cors.logf("HandlerFunc: Preflight request") |
||||
cors.handlePreflight(w, r) |
||||
} else { |
||||
cors.logf("HandlerFunc: Actual request") |
||||
cors.handleActualRequest(w, r) |
||||
} |
||||
} |
||||
|
||||
// Negroni compatible interface
|
||||
func (cors *Cors) ServeHTTP(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) { |
||||
if r.Method == "OPTIONS" { |
||||
cors.logf("ServeHTTP: Preflight request") |
||||
cors.handlePreflight(w, r) |
||||
// Preflight requests are standalone and should stop the chain as some other
|
||||
// middleware may not handle OPTIONS requests correctly. One typical example
|
||||
// is authentication middleware ; OPTIONS requests won't carry authentication
|
||||
// headers (see #1)
|
||||
} else { |
||||
cors.logf("ServeHTTP: Actual request") |
||||
cors.handleActualRequest(w, r) |
||||
next(w, r) |
||||
} |
||||
} |
||||
|
||||
// handlePreflight handles pre-flight CORS requests
|
||||
func (cors *Cors) handlePreflight(w http.ResponseWriter, r *http.Request) { |
||||
options := cors.options |
||||
headers := w.Header() |
||||
origin := r.Header.Get("Origin") |
||||
|
||||
if r.Method != "OPTIONS" { |
||||
cors.logf(" Preflight aborted: %s!=OPTIONS", r.Method) |
||||
return |
||||
} |
||||
if origin == "" { |
||||
cors.logf(" Preflight aborted: empty origin") |
||||
return |
||||
} |
||||
if !cors.isOriginAllowed(origin) { |
||||
cors.logf(" Preflight aborted: origin '%s' not allowed", origin) |
||||
return |
||||
} |
||||
|
||||
reqMethod := r.Header.Get("Access-Control-Request-Method") |
||||
if !cors.isMethodAllowed(reqMethod) { |
||||
cors.logf(" Preflight aborted: method '%s' not allowed", reqMethod) |
||||
return |
||||
} |
||||
reqHeaders := parseHeaderList(r.Header.Get("Access-Control-Request-Headers")) |
||||
if !cors.areHeadersAllowed(reqHeaders) { |
||||
cors.logf(" Preflight aborted: headers '%v' not allowed", reqHeaders) |
||||
return |
||||
} |
||||
headers.Set("Access-Control-Allow-Origin", origin) |
||||
headers.Add("Vary", "Origin") |
||||
// Spec says: Since the list of methods can be unbounded, simply returning the method indicated
|
||||
// by Access-Control-Request-Method (if supported) can be enough
|
||||
headers.Set("Access-Control-Allow-Methods", strings.ToUpper(reqMethod)) |
||||
if len(reqHeaders) > 0 { |
||||
|
||||
// Spec says: Since the list of headers can be unbounded, simply returning supported headers
|
||||
// from Access-Control-Request-Headers can be enough
|
||||
headers.Set("Access-Control-Allow-Headers", strings.Join(reqHeaders, ", ")) |
||||
} |
||||
if options.AllowCredentials { |
||||
headers.Set("Access-Control-Allow-Credentials", "true") |
||||
} |
||||
if options.MaxAge > 0 { |
||||
headers.Set("Access-Control-Max-Age", strconv.Itoa(options.MaxAge)) |
||||
} |
||||
cors.logf(" Preflight response headers: %v", headers) |
||||
} |
||||
|
||||
// handleActualRequest handles simple cross-origin requests, actual request or redirects
|
||||
func (cors *Cors) handleActualRequest(w http.ResponseWriter, r *http.Request) { |
||||
options := cors.options |
||||
headers := w.Header() |
||||
origin := r.Header.Get("Origin") |
||||
|
||||
if r.Method == "OPTIONS" { |
||||
cors.logf(" Actual request no headers added: method == %s", r.Method) |
||||
return |
||||
} |
||||
if origin == "" { |
||||
cors.logf(" Actual request no headers added: missing origin") |
||||
return |
||||
} |
||||
if !cors.isOriginAllowed(origin) { |
||||
cors.logf(" Actual request no headers added: origin '%s' not allowed", origin) |
||||
return |
||||
} |
||||
|
||||
// Note that spec does define a way to specifically disallow a simple method like GET or
|
||||
// POST. Access-Control-Allow-Methods is only used for pre-flight requests and the
|
||||
// spec doesn't instruct to check the allowed methods for simple cross-origin requests.
|
||||
// We think it's a nice feature to be able to have control on those methods though.
|
||||
if !cors.isMethodAllowed(r.Method) { |
||||
if cors.options.Debug { |
||||
cors.logf(" Actual request no headers added: method '%s' not allowed", |
||||
r.Method) |
||||
} |
||||
|
||||
return |
||||
} |
||||
headers.Set("Access-Control-Allow-Origin", origin) |
||||
headers.Add("Vary", "Origin") |
||||
if len(options.ExposedHeaders) > 0 { |
||||
headers.Set("Access-Control-Expose-Headers", strings.Join(options.ExposedHeaders, ", ")) |
||||
} |
||||
if options.AllowCredentials { |
||||
headers.Set("Access-Control-Allow-Credentials", "true") |
||||
} |
||||
cors.logf(" Actual response added headers: %v", headers) |
||||
} |
||||
|
||||
// convenience method. checks if debugging is turned on before printing
|
||||
func (cors *Cors) logf(format string, a ...interface{}) { |
||||
if cors.options.Debug { |
||||
cors.options.log.Printf(format, a...) |
||||
} |
||||
} |
||||
|
||||
// isOriginAllowed checks if a given origin is allowed to perform cross-domain requests
|
||||
// on the endpoint
|
||||
func (cors *Cors) isOriginAllowed(origin string) bool { |
||||
allowedOrigins := cors.options.AllowedOrigins |
||||
origin = strings.ToLower(origin) |
||||
for _, allowedOrigin := range allowedOrigins { |
||||
switch allowedOrigin { |
||||
case "*": |
||||
return true |
||||
case origin: |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
// isMethodAllowed checks if a given method can be used as part of a cross-domain request
|
||||
// on the endpoing
|
||||
func (cors *Cors) isMethodAllowed(method string) bool { |
||||
allowedMethods := cors.options.AllowedMethods |
||||
if len(allowedMethods) == 0 { |
||||
// If no method allowed, always return false, even for preflight request
|
||||
return false |
||||
} |
||||
method = strings.ToUpper(method) |
||||
if method == "OPTIONS" { |
||||
// Always allow preflight requests
|
||||
return true |
||||
} |
||||
for _, allowedMethod := range allowedMethods { |
||||
if allowedMethod == method { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
// areHeadersAllowed checks if a given list of headers are allowed to used within
|
||||
// a cross-domain request.
|
||||
func (cors *Cors) areHeadersAllowed(requestedHeaders []string) bool { |
||||
if len(requestedHeaders) == 0 { |
||||
return true |
||||
} |
||||
for _, header := range requestedHeaders { |
||||
found := false |
||||
for _, allowedHeader := range cors.options.AllowedHeaders { |
||||
if allowedHeader == "*" || allowedHeader == header { |
||||
found = true |
||||
break |
||||
} |
||||
} |
||||
if !found { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
@ -1,288 +0,0 @@ |
||||
package cors |
||||
|
||||
import ( |
||||
"net/http" |
||||
"net/http/httptest" |
||||
"testing" |
||||
) |
||||
|
||||
var testHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
w.Write([]byte("bar")) |
||||
}) |
||||
|
||||
func assertHeaders(t *testing.T, resHeaders http.Header, reqHeaders map[string]string) { |
||||
for name, value := range reqHeaders { |
||||
if resHeaders.Get(name) != value { |
||||
t.Errorf("Invalid header `%s', wanted `%s', got `%s'", name, value, resHeaders.Get(name)) |
||||
} |
||||
} |
||||
} |
||||
|
||||
func TestNoConfig(t *testing.T) { |
||||
s := New(Options{ |
||||
// Intentionally left blank.
|
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestWildcardOrigin(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"*"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestAllowedOrigin(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestDisallowedOrigin(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://barbaz.com") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestAllowedMethod(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowedMethods: []string{"PUT", "DELETE"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "PUT") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "PUT", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestDisallowedMethod(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowedMethods: []string{"PUT", "DELETE"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "PATCH") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestAllowedHeader(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowedHeaders: []string{"X-Header-1", "x-header-2"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
req.Header.Add("Access-Control-Request-Headers", "X-Header-2, X-HEADER-1") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "GET", |
||||
"Access-Control-Allow-Headers": "X-Header-2, X-Header-1", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestAllowedWildcardHeader(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowedHeaders: []string{"*"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
req.Header.Add("Access-Control-Request-Headers", "X-Header-2, X-HEADER-1") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "GET", |
||||
"Access-Control-Allow-Headers": "X-Header-2, X-Header-1", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestDisallowedHeader(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowedHeaders: []string{"X-Header-1", "x-header-2"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
req.Header.Add("Access-Control-Request-Headers", "X-Header-3, X-Header-1") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestOriginHeader(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
req.Header.Add("Access-Control-Request-Headers", "origin") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "GET", |
||||
"Access-Control-Allow-Headers": "Origin", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
|
||||
func TestExposedHeader(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
ExposedHeaders: []string{"X-Header-1", "x-header-2"}, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("GET", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "X-Header-1, X-Header-2", |
||||
}) |
||||
} |
||||
|
||||
func TestAllowedCredentials(t *testing.T) { |
||||
s := New(Options{ |
||||
AllowedOrigins: []string{"http://foobar.com"}, |
||||
AllowCredentials: true, |
||||
}) |
||||
|
||||
res := httptest.NewRecorder() |
||||
req, _ := http.NewRequest("OPTIONS", "http://example.com/foo", nil) |
||||
req.Header.Add("Origin", "http://foobar.com") |
||||
req.Header.Add("Access-Control-Request-Method", "GET") |
||||
|
||||
s.Handler(testHandler).ServeHTTP(res, req) |
||||
|
||||
assertHeaders(t, res.Header(), map[string]string{ |
||||
"Access-Control-Allow-Origin": "http://foobar.com", |
||||
"Access-Control-Allow-Methods": "GET", |
||||
"Access-Control-Allow-Headers": "", |
||||
"Access-Control-Allow-Credentials": "true", |
||||
"Access-Control-Max-Age": "", |
||||
"Access-Control-Expose-Headers": "", |
||||
}) |
||||
} |
||||
@ -1,24 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/justinas/alice" |
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
}) |
||||
|
||||
mux := http.NewServeMux() |
||||
|
||||
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
chain := alice.New(c.Handler).Then(mux) |
||||
http.ListenAndServe(":8080", chain) |
||||
} |
||||
@ -1,18 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
// Use default options
|
||||
handler := cors.Default().Handler(h) |
||||
http.ListenAndServe(":8080", handler) |
||||
} |
||||
@ -1,22 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/rs/cors" |
||||
"github.com/zenazn/goji" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
}) |
||||
goji.Use(c.Handler) |
||||
|
||||
goji.Get("/", func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
goji.Serve() |
||||
} |
||||
@ -1,23 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"github.com/go-martini/martini" |
||||
"github.com/martini-contrib/render" |
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
}) |
||||
|
||||
m := martini.Classic() |
||||
m.Use(render.Renderer()) |
||||
m.Use(c.HandlerFunc) |
||||
|
||||
m.Get("/", func(r render.Render) { |
||||
r.JSON(200, map[string]interface{}{"hello": "world"}) |
||||
}) |
||||
|
||||
m.Run() |
||||
} |
||||
@ -1,26 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/codegangsta/negroni" |
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
}) |
||||
|
||||
mux := http.NewServeMux() |
||||
|
||||
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
n := negroni.Classic() |
||||
n.Use(c) |
||||
n.UseHandler(mux) |
||||
n.Run(":3000") |
||||
} |
||||
@ -1,20 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"http://foo.com"}, |
||||
}) |
||||
|
||||
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
http.ListenAndServe(":8080", c.Handler(handler)) |
||||
} |
||||
@ -1,22 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
func main() { |
||||
c := cors.New(cors.Options{ |
||||
AllowedOrigins: []string{"*"}, |
||||
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE"}, |
||||
AllowCredentials: true, |
||||
}) |
||||
|
||||
h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
w.Write([]byte("{\"hello\": \"world\"}")) |
||||
}) |
||||
|
||||
http.ListenAndServe(":8080", c.Handler(h)) |
||||
} |
||||
@ -1,27 +0,0 @@ |
||||
package cors |
||||
|
||||
import ( |
||||
"net/http" |
||||
"strings" |
||||
) |
||||
|
||||
type converter func(string) string |
||||
|
||||
// convert converts a list of string using the passed converter function
|
||||
func convert(s []string, c converter) []string { |
||||
out := []string{} |
||||
for _, i := range s { |
||||
out = append(out, c(i)) |
||||
} |
||||
return out |
||||
} |
||||
|
||||
func parseHeaderList(headerList string) (headers []string) { |
||||
for _, header := range strings.Split(headerList, ",") { |
||||
header = http.CanonicalHeaderKey(strings.TrimSpace(header)) |
||||
if header != "" { |
||||
headers = append(headers, header) |
||||
} |
||||
} |
||||
return headers |
||||
} |
||||
@ -1,28 +0,0 @@ |
||||
package cors |
||||
|
||||
import ( |
||||
"strings" |
||||
"testing" |
||||
) |
||||
|
||||
func TestConvert(t *testing.T) { |
||||
s := convert([]string{"A", "b", "C"}, strings.ToLower) |
||||
e := []string{"a", "b", "c"} |
||||
if s[0] != e[0] || s[1] != e[1] || s[2] != e[2] { |
||||
t.Errorf("%v != %v", s, e) |
||||
} |
||||
} |
||||
|
||||
func TestParseHeaderList(t *testing.T) { |
||||
h := parseHeaderList("header, second-header, THIRD-HEADER") |
||||
e := []string{"Header", "Second-Header", "Third-Header"} |
||||
if h[0] != e[0] || h[1] != e[1] || h[2] != e[2] { |
||||
t.Errorf("%v != %v", h, e) |
||||
} |
||||
} |
||||
|
||||
func TestParseHeaderListEmpty(t *testing.T) { |
||||
if len(parseHeaderList("")) != 0 { |
||||
t.Error("should be empty sclice") |
||||
} |
||||
} |
||||
@ -0,0 +1,113 @@ |
||||
// Copyright 2009 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket |
||||
|
||||
import ( |
||||
"bufio" |
||||
"crypto/tls" |
||||
"io" |
||||
"net" |
||||
"net/http" |
||||
"net/url" |
||||
) |
||||
|
||||
// DialError is an error that occurs while dialling a websocket server.
|
||||
type DialError struct { |
||||
*Config |
||||
Err error |
||||
} |
||||
|
||||
func (e *DialError) Error() string { |
||||
return "websocket.Dial " + e.Config.Location.String() + ": " + e.Err.Error() |
||||
} |
||||
|
||||
// NewConfig creates a new WebSocket config for client connection.
|
||||
func NewConfig(server, origin string) (config *Config, err error) { |
||||
config = new(Config) |
||||
config.Version = ProtocolVersionHybi13 |
||||
config.Location, err = url.ParseRequestURI(server) |
||||
if err != nil { |
||||
return |
||||
} |
||||
config.Origin, err = url.ParseRequestURI(origin) |
||||
if err != nil { |
||||
return |
||||
} |
||||
config.Header = http.Header(make(map[string][]string)) |
||||
return |
||||
} |
||||
|
||||
// NewClient creates a new WebSocket client connection over rwc.
|
||||
func NewClient(config *Config, rwc io.ReadWriteCloser) (ws *Conn, err error) { |
||||
br := bufio.NewReader(rwc) |
||||
bw := bufio.NewWriter(rwc) |
||||
err = hybiClientHandshake(config, br, bw) |
||||
if err != nil { |
||||
return |
||||
} |
||||
buf := bufio.NewReadWriter(br, bw) |
||||
ws = newHybiClientConn(config, buf, rwc) |
||||
return |
||||
} |
||||
|
||||
// Dial opens a new client connection to a WebSocket.
|
||||
func Dial(url_, protocol, origin string) (ws *Conn, err error) { |
||||
config, err := NewConfig(url_, origin) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
if protocol != "" { |
||||
config.Protocol = []string{protocol} |
||||
} |
||||
return DialConfig(config) |
||||
} |
||||
|
||||
var portMap = map[string]string{ |
||||
"ws": "80", |
||||
"wss": "443", |
||||
} |
||||
|
||||
func parseAuthority(location *url.URL) string { |
||||
if _, ok := portMap[location.Scheme]; ok { |
||||
if _, _, err := net.SplitHostPort(location.Host); err != nil { |
||||
return net.JoinHostPort(location.Host, portMap[location.Scheme]) |
||||
} |
||||
} |
||||
return location.Host |
||||
} |
||||
|
||||
// DialConfig opens a new client connection to a WebSocket with a config.
|
||||
func DialConfig(config *Config) (ws *Conn, err error) { |
||||
var client net.Conn |
||||
if config.Location == nil { |
||||
return nil, &DialError{config, ErrBadWebSocketLocation} |
||||
} |
||||
if config.Origin == nil { |
||||
return nil, &DialError{config, ErrBadWebSocketOrigin} |
||||
} |
||||
switch config.Location.Scheme { |
||||
case "ws": |
||||
client, err = net.Dial("tcp", parseAuthority(config.Location)) |
||||
|
||||
case "wss": |
||||
client, err = tls.Dial("tcp", parseAuthority(config.Location), config.TlsConfig) |
||||
|
||||
default: |
||||
err = ErrBadScheme |
||||
} |
||||
if err != nil { |
||||
goto Error |
||||
} |
||||
|
||||
ws, err = NewClient(config, client) |
||||
if err != nil { |
||||
client.Close() |
||||
goto Error |
||||
} |
||||
return |
||||
|
||||
Error: |
||||
return nil, &DialError{config, err} |
||||
} |
||||
@ -0,0 +1,31 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket_test |
||||
|
||||
import ( |
||||
"fmt" |
||||
"log" |
||||
|
||||
"golang.org/x/net/websocket" |
||||
) |
||||
|
||||
// This example demonstrates a trivial client.
|
||||
func ExampleDial() { |
||||
origin := "http://localhost/" |
||||
url := "ws://localhost:12345/ws" |
||||
ws, err := websocket.Dial(url, "", origin) |
||||
if err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
if _, err := ws.Write([]byte("hello, world!\n")); err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
var msg = make([]byte, 512) |
||||
var n int |
||||
if n, err = ws.Read(msg); err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
fmt.Printf("Received: %s.\n", msg[:n]) |
||||
} |
||||
@ -0,0 +1,26 @@ |
||||
// Copyright 2012 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket_test |
||||
|
||||
import ( |
||||
"io" |
||||
"net/http" |
||||
|
||||
"golang.org/x/net/websocket" |
||||
) |
||||
|
||||
// Echo the data received on the WebSocket.
|
||||
func EchoServer(ws *websocket.Conn) { |
||||
io.Copy(ws, ws) |
||||
} |
||||
|
||||
// This example demonstrates a trivial echo server.
|
||||
func ExampleHandler() { |
||||
http.Handle("/echo", websocket.Handler(EchoServer)) |
||||
err := http.ListenAndServe(":12345", nil) |
||||
if err != nil { |
||||
panic("ListenAndServe: " + err.Error()) |
||||
} |
||||
} |
||||
@ -0,0 +1,564 @@ |
||||
// Copyright 2011 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket |
||||
|
||||
// This file implements a protocol of hybi draft.
|
||||
// http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-17
|
||||
|
||||
import ( |
||||
"bufio" |
||||
"bytes" |
||||
"crypto/rand" |
||||
"crypto/sha1" |
||||
"encoding/base64" |
||||
"encoding/binary" |
||||
"fmt" |
||||
"io" |
||||
"io/ioutil" |
||||
"net/http" |
||||
"net/url" |
||||
"strings" |
||||
) |
||||
|
||||
const ( |
||||
websocketGUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11" |
||||
|
||||
closeStatusNormal = 1000 |
||||
closeStatusGoingAway = 1001 |
||||
closeStatusProtocolError = 1002 |
||||
closeStatusUnsupportedData = 1003 |
||||
closeStatusFrameTooLarge = 1004 |
||||
closeStatusNoStatusRcvd = 1005 |
||||
closeStatusAbnormalClosure = 1006 |
||||
closeStatusBadMessageData = 1007 |
||||
closeStatusPolicyViolation = 1008 |
||||
closeStatusTooBigData = 1009 |
||||
closeStatusExtensionMismatch = 1010 |
||||
|
||||
maxControlFramePayloadLength = 125 |
||||
) |
||||
|
||||
var ( |
||||
ErrBadMaskingKey = &ProtocolError{"bad masking key"} |
||||
ErrBadPongMessage = &ProtocolError{"bad pong message"} |
||||
ErrBadClosingStatus = &ProtocolError{"bad closing status"} |
||||
ErrUnsupportedExtensions = &ProtocolError{"unsupported extensions"} |
||||
ErrNotImplemented = &ProtocolError{"not implemented"} |
||||
|
||||
handshakeHeader = map[string]bool{ |
||||
"Host": true, |
||||
"Upgrade": true, |
||||
"Connection": true, |
||||
"Sec-Websocket-Key": true, |
||||
"Sec-Websocket-Origin": true, |
||||
"Sec-Websocket-Version": true, |
||||
"Sec-Websocket-Protocol": true, |
||||
"Sec-Websocket-Accept": true, |
||||
} |
||||
) |
||||
|
||||
// A hybiFrameHeader is a frame header as defined in hybi draft.
|
||||
type hybiFrameHeader struct { |
||||
Fin bool |
||||
Rsv [3]bool |
||||
OpCode byte |
||||
Length int64 |
||||
MaskingKey []byte |
||||
|
||||
data *bytes.Buffer |
||||
} |
||||
|
||||
// A hybiFrameReader is a reader for hybi frame.
|
||||
type hybiFrameReader struct { |
||||
reader io.Reader |
||||
|
||||
header hybiFrameHeader |
||||
pos int64 |
||||
length int |
||||
} |
||||
|
||||
func (frame *hybiFrameReader) Read(msg []byte) (n int, err error) { |
||||
n, err = frame.reader.Read(msg) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
if frame.header.MaskingKey != nil { |
||||
for i := 0; i < n; i++ { |
||||
msg[i] = msg[i] ^ frame.header.MaskingKey[frame.pos%4] |
||||
frame.pos++ |
||||
} |
||||
} |
||||
return n, err |
||||
} |
||||
|
||||
func (frame *hybiFrameReader) PayloadType() byte { return frame.header.OpCode } |
||||
|
||||
func (frame *hybiFrameReader) HeaderReader() io.Reader { |
||||
if frame.header.data == nil { |
||||
return nil |
||||
} |
||||
if frame.header.data.Len() == 0 { |
||||
return nil |
||||
} |
||||
return frame.header.data |
||||
} |
||||
|
||||
func (frame *hybiFrameReader) TrailerReader() io.Reader { return nil } |
||||
|
||||
func (frame *hybiFrameReader) Len() (n int) { return frame.length } |
||||
|
||||
// A hybiFrameReaderFactory creates new frame reader based on its frame type.
|
||||
type hybiFrameReaderFactory struct { |
||||
*bufio.Reader |
||||
} |
||||
|
||||
// NewFrameReader reads a frame header from the connection, and creates new reader for the frame.
|
||||
// See Section 5.2 Base Framing protocol for detail.
|
||||
// http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-17#section-5.2
|
||||
func (buf hybiFrameReaderFactory) NewFrameReader() (frame frameReader, err error) { |
||||
hybiFrame := new(hybiFrameReader) |
||||
frame = hybiFrame |
||||
var header []byte |
||||
var b byte |
||||
// First byte. FIN/RSV1/RSV2/RSV3/OpCode(4bits)
|
||||
b, err = buf.ReadByte() |
||||
if err != nil { |
||||
return |
||||
} |
||||
header = append(header, b) |
||||
hybiFrame.header.Fin = ((header[0] >> 7) & 1) != 0 |
||||
for i := 0; i < 3; i++ { |
||||
j := uint(6 - i) |
||||
hybiFrame.header.Rsv[i] = ((header[0] >> j) & 1) != 0 |
||||
} |
||||
hybiFrame.header.OpCode = header[0] & 0x0f |
||||
|
||||
// Second byte. Mask/Payload len(7bits)
|
||||
b, err = buf.ReadByte() |
||||
if err != nil { |
||||
return |
||||
} |
||||
header = append(header, b) |
||||
mask := (b & 0x80) != 0 |
||||
b &= 0x7f |
||||
lengthFields := 0 |
||||
switch { |
||||
case b <= 125: // Payload length 7bits.
|
||||
hybiFrame.header.Length = int64(b) |
||||
case b == 126: // Payload length 7+16bits
|
||||
lengthFields = 2 |
||||
case b == 127: // Payload length 7+64bits
|
||||
lengthFields = 8 |
||||
} |
||||
for i := 0; i < lengthFields; i++ { |
||||
b, err = buf.ReadByte() |
||||
if err != nil { |
||||
return |
||||
} |
||||
header = append(header, b) |
||||
hybiFrame.header.Length = hybiFrame.header.Length*256 + int64(b) |
||||
} |
||||
if mask { |
||||
// Masking key. 4 bytes.
|
||||
for i := 0; i < 4; i++ { |
||||
b, err = buf.ReadByte() |
||||
if err != nil { |
||||
return |
||||
} |
||||
header = append(header, b) |
||||
hybiFrame.header.MaskingKey = append(hybiFrame.header.MaskingKey, b) |
||||
} |
||||
} |
||||
hybiFrame.reader = io.LimitReader(buf.Reader, hybiFrame.header.Length) |
||||
hybiFrame.header.data = bytes.NewBuffer(header) |
||||
hybiFrame.length = len(header) + int(hybiFrame.header.Length) |
||||
return |
||||
} |
||||
|
||||
// A HybiFrameWriter is a writer for hybi frame.
|
||||
type hybiFrameWriter struct { |
||||
writer *bufio.Writer |
||||
|
||||
header *hybiFrameHeader |
||||
} |
||||
|
||||
func (frame *hybiFrameWriter) Write(msg []byte) (n int, err error) { |
||||
var header []byte |
||||
var b byte |
||||
if frame.header.Fin { |
||||
b |= 0x80 |
||||
} |
||||
for i := 0; i < 3; i++ { |
||||
if frame.header.Rsv[i] { |
||||
j := uint(6 - i) |
||||
b |= 1 << j |
||||
} |
||||
} |
||||
b |= frame.header.OpCode |
||||
header = append(header, b) |
||||
if frame.header.MaskingKey != nil { |
||||
b = 0x80 |
||||
} else { |
||||
b = 0 |
||||
} |
||||
lengthFields := 0 |
||||
length := len(msg) |
||||
switch { |
||||
case length <= 125: |
||||
b |= byte(length) |
||||
case length < 65536: |
||||
b |= 126 |
||||
lengthFields = 2 |
||||
default: |
||||
b |= 127 |
||||
lengthFields = 8 |
||||
} |
||||
header = append(header, b) |
||||
for i := 0; i < lengthFields; i++ { |
||||
j := uint((lengthFields - i - 1) * 8) |
||||
b = byte((length >> j) & 0xff) |
||||
header = append(header, b) |
||||
} |
||||
if frame.header.MaskingKey != nil { |
||||
if len(frame.header.MaskingKey) != 4 { |
||||
return 0, ErrBadMaskingKey |
||||
} |
||||
header = append(header, frame.header.MaskingKey...) |
||||
frame.writer.Write(header) |
||||
data := make([]byte, length) |
||||
for i := range data { |
||||
data[i] = msg[i] ^ frame.header.MaskingKey[i%4] |
||||
} |
||||
frame.writer.Write(data) |
||||
err = frame.writer.Flush() |
||||
return length, err |
||||
} |
||||
frame.writer.Write(header) |
||||
frame.writer.Write(msg) |
||||
err = frame.writer.Flush() |
||||
return length, err |
||||
} |
||||
|
||||
func (frame *hybiFrameWriter) Close() error { return nil } |
||||
|
||||
type hybiFrameWriterFactory struct { |
||||
*bufio.Writer |
||||
needMaskingKey bool |
||||
} |
||||
|
||||
func (buf hybiFrameWriterFactory) NewFrameWriter(payloadType byte) (frame frameWriter, err error) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: payloadType} |
||||
if buf.needMaskingKey { |
||||
frameHeader.MaskingKey, err = generateMaskingKey() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return &hybiFrameWriter{writer: buf.Writer, header: frameHeader}, nil |
||||
} |
||||
|
||||
type hybiFrameHandler struct { |
||||
conn *Conn |
||||
payloadType byte |
||||
} |
||||
|
||||
func (handler *hybiFrameHandler) HandleFrame(frame frameReader) (r frameReader, err error) { |
||||
if handler.conn.IsServerConn() { |
||||
// The client MUST mask all frames sent to the server.
|
||||
if frame.(*hybiFrameReader).header.MaskingKey == nil { |
||||
handler.WriteClose(closeStatusProtocolError) |
||||
return nil, io.EOF |
||||
} |
||||
} else { |
||||
// The server MUST NOT mask all frames.
|
||||
if frame.(*hybiFrameReader).header.MaskingKey != nil { |
||||
handler.WriteClose(closeStatusProtocolError) |
||||
return nil, io.EOF |
||||
} |
||||
} |
||||
if header := frame.HeaderReader(); header != nil { |
||||
io.Copy(ioutil.Discard, header) |
||||
} |
||||
switch frame.PayloadType() { |
||||
case ContinuationFrame: |
||||
frame.(*hybiFrameReader).header.OpCode = handler.payloadType |
||||
case TextFrame, BinaryFrame: |
||||
handler.payloadType = frame.PayloadType() |
||||
case CloseFrame: |
||||
return nil, io.EOF |
||||
case PingFrame: |
||||
pingMsg := make([]byte, maxControlFramePayloadLength) |
||||
n, err := io.ReadFull(frame, pingMsg) |
||||
if err != nil && err != io.ErrUnexpectedEOF { |
||||
return nil, err |
||||
} |
||||
io.Copy(ioutil.Discard, frame) |
||||
n, err = handler.WritePong(pingMsg[:n]) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return nil, nil |
||||
case PongFrame: |
||||
return nil, ErrNotImplemented |
||||
} |
||||
return frame, nil |
||||
} |
||||
|
||||
func (handler *hybiFrameHandler) WriteClose(status int) (err error) { |
||||
handler.conn.wio.Lock() |
||||
defer handler.conn.wio.Unlock() |
||||
w, err := handler.conn.frameWriterFactory.NewFrameWriter(CloseFrame) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
msg := make([]byte, 2) |
||||
binary.BigEndian.PutUint16(msg, uint16(status)) |
||||
_, err = w.Write(msg) |
||||
w.Close() |
||||
return err |
||||
} |
||||
|
||||
func (handler *hybiFrameHandler) WritePong(msg []byte) (n int, err error) { |
||||
handler.conn.wio.Lock() |
||||
defer handler.conn.wio.Unlock() |
||||
w, err := handler.conn.frameWriterFactory.NewFrameWriter(PongFrame) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
n, err = w.Write(msg) |
||||
w.Close() |
||||
return n, err |
||||
} |
||||
|
||||
// newHybiConn creates a new WebSocket connection speaking hybi draft protocol.
|
||||
func newHybiConn(config *Config, buf *bufio.ReadWriter, rwc io.ReadWriteCloser, request *http.Request) *Conn { |
||||
if buf == nil { |
||||
br := bufio.NewReader(rwc) |
||||
bw := bufio.NewWriter(rwc) |
||||
buf = bufio.NewReadWriter(br, bw) |
||||
} |
||||
ws := &Conn{config: config, request: request, buf: buf, rwc: rwc, |
||||
frameReaderFactory: hybiFrameReaderFactory{buf.Reader}, |
||||
frameWriterFactory: hybiFrameWriterFactory{ |
||||
buf.Writer, request == nil}, |
||||
PayloadType: TextFrame, |
||||
defaultCloseStatus: closeStatusNormal} |
||||
ws.frameHandler = &hybiFrameHandler{conn: ws} |
||||
return ws |
||||
} |
||||
|
||||
// generateMaskingKey generates a masking key for a frame.
|
||||
func generateMaskingKey() (maskingKey []byte, err error) { |
||||
maskingKey = make([]byte, 4) |
||||
if _, err = io.ReadFull(rand.Reader, maskingKey); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
// generateNonce generates a nonce consisting of a randomly selected 16-byte
|
||||
// value that has been base64-encoded.
|
||||
func generateNonce() (nonce []byte) { |
||||
key := make([]byte, 16) |
||||
if _, err := io.ReadFull(rand.Reader, key); err != nil { |
||||
panic(err) |
||||
} |
||||
nonce = make([]byte, 24) |
||||
base64.StdEncoding.Encode(nonce, key) |
||||
return |
||||
} |
||||
|
||||
// getNonceAccept computes the base64-encoded SHA-1 of the concatenation of
|
||||
// the nonce ("Sec-WebSocket-Key" value) with the websocket GUID string.
|
||||
func getNonceAccept(nonce []byte) (expected []byte, err error) { |
||||
h := sha1.New() |
||||
if _, err = h.Write(nonce); err != nil { |
||||
return |
||||
} |
||||
if _, err = h.Write([]byte(websocketGUID)); err != nil { |
||||
return |
||||
} |
||||
expected = make([]byte, 28) |
||||
base64.StdEncoding.Encode(expected, h.Sum(nil)) |
||||
return |
||||
} |
||||
|
||||
// Client handshake described in draft-ietf-hybi-thewebsocket-protocol-17
|
||||
func hybiClientHandshake(config *Config, br *bufio.Reader, bw *bufio.Writer) (err error) { |
||||
bw.WriteString("GET " + config.Location.RequestURI() + " HTTP/1.1\r\n") |
||||
|
||||
bw.WriteString("Host: " + config.Location.Host + "\r\n") |
||||
bw.WriteString("Upgrade: websocket\r\n") |
||||
bw.WriteString("Connection: Upgrade\r\n") |
||||
nonce := generateNonce() |
||||
if config.handshakeData != nil { |
||||
nonce = []byte(config.handshakeData["key"]) |
||||
} |
||||
bw.WriteString("Sec-WebSocket-Key: " + string(nonce) + "\r\n") |
||||
bw.WriteString("Origin: " + strings.ToLower(config.Origin.String()) + "\r\n") |
||||
|
||||
if config.Version != ProtocolVersionHybi13 { |
||||
return ErrBadProtocolVersion |
||||
} |
||||
|
||||
bw.WriteString("Sec-WebSocket-Version: " + fmt.Sprintf("%d", config.Version) + "\r\n") |
||||
if len(config.Protocol) > 0 { |
||||
bw.WriteString("Sec-WebSocket-Protocol: " + strings.Join(config.Protocol, ", ") + "\r\n") |
||||
} |
||||
// TODO(ukai): send Sec-WebSocket-Extensions.
|
||||
err = config.Header.WriteSubset(bw, handshakeHeader) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
bw.WriteString("\r\n") |
||||
if err = bw.Flush(); err != nil { |
||||
return err |
||||
} |
||||
|
||||
resp, err := http.ReadResponse(br, &http.Request{Method: "GET"}) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
if resp.StatusCode != 101 { |
||||
return ErrBadStatus |
||||
} |
||||
if strings.ToLower(resp.Header.Get("Upgrade")) != "websocket" || |
||||
strings.ToLower(resp.Header.Get("Connection")) != "upgrade" { |
||||
return ErrBadUpgrade |
||||
} |
||||
expectedAccept, err := getNonceAccept(nonce) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
if resp.Header.Get("Sec-WebSocket-Accept") != string(expectedAccept) { |
||||
return ErrChallengeResponse |
||||
} |
||||
if resp.Header.Get("Sec-WebSocket-Extensions") != "" { |
||||
return ErrUnsupportedExtensions |
||||
} |
||||
offeredProtocol := resp.Header.Get("Sec-WebSocket-Protocol") |
||||
if offeredProtocol != "" { |
||||
protocolMatched := false |
||||
for i := 0; i < len(config.Protocol); i++ { |
||||
if config.Protocol[i] == offeredProtocol { |
||||
protocolMatched = true |
||||
break |
||||
} |
||||
} |
||||
if !protocolMatched { |
||||
return ErrBadWebSocketProtocol |
||||
} |
||||
config.Protocol = []string{offeredProtocol} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// newHybiClientConn creates a client WebSocket connection after handshake.
|
||||
func newHybiClientConn(config *Config, buf *bufio.ReadWriter, rwc io.ReadWriteCloser) *Conn { |
||||
return newHybiConn(config, buf, rwc, nil) |
||||
} |
||||
|
||||
// A HybiServerHandshaker performs a server handshake using hybi draft protocol.
|
||||
type hybiServerHandshaker struct { |
||||
*Config |
||||
accept []byte |
||||
} |
||||
|
||||
func (c *hybiServerHandshaker) ReadHandshake(buf *bufio.Reader, req *http.Request) (code int, err error) { |
||||
c.Version = ProtocolVersionHybi13 |
||||
if req.Method != "GET" { |
||||
return http.StatusMethodNotAllowed, ErrBadRequestMethod |
||||
} |
||||
// HTTP version can be safely ignored.
|
||||
|
||||
if strings.ToLower(req.Header.Get("Upgrade")) != "websocket" || |
||||
!strings.Contains(strings.ToLower(req.Header.Get("Connection")), "upgrade") { |
||||
return http.StatusBadRequest, ErrNotWebSocket |
||||
} |
||||
|
||||
key := req.Header.Get("Sec-Websocket-Key") |
||||
if key == "" { |
||||
return http.StatusBadRequest, ErrChallengeResponse |
||||
} |
||||
version := req.Header.Get("Sec-Websocket-Version") |
||||
switch version { |
||||
case "13": |
||||
c.Version = ProtocolVersionHybi13 |
||||
default: |
||||
return http.StatusBadRequest, ErrBadWebSocketVersion |
||||
} |
||||
var scheme string |
||||
if req.TLS != nil { |
||||
scheme = "wss" |
||||
} else { |
||||
scheme = "ws" |
||||
} |
||||
c.Location, err = url.ParseRequestURI(scheme + "://" + req.Host + req.URL.RequestURI()) |
||||
if err != nil { |
||||
return http.StatusBadRequest, err |
||||
} |
||||
protocol := strings.TrimSpace(req.Header.Get("Sec-Websocket-Protocol")) |
||||
if protocol != "" { |
||||
protocols := strings.Split(protocol, ",") |
||||
for i := 0; i < len(protocols); i++ { |
||||
c.Protocol = append(c.Protocol, strings.TrimSpace(protocols[i])) |
||||
} |
||||
} |
||||
c.accept, err = getNonceAccept([]byte(key)) |
||||
if err != nil { |
||||
return http.StatusInternalServerError, err |
||||
} |
||||
return http.StatusSwitchingProtocols, nil |
||||
} |
||||
|
||||
// Origin parses Origin header in "req".
|
||||
// If origin is "null", returns (nil, nil).
|
||||
func Origin(config *Config, req *http.Request) (*url.URL, error) { |
||||
var origin string |
||||
switch config.Version { |
||||
case ProtocolVersionHybi13: |
||||
origin = req.Header.Get("Origin") |
||||
} |
||||
if origin == "null" { |
||||
return nil, nil |
||||
} |
||||
return url.ParseRequestURI(origin) |
||||
} |
||||
|
||||
func (c *hybiServerHandshaker) AcceptHandshake(buf *bufio.Writer) (err error) { |
||||
if len(c.Protocol) > 0 { |
||||
if len(c.Protocol) != 1 { |
||||
// You need choose a Protocol in Handshake func in Server.
|
||||
return ErrBadWebSocketProtocol |
||||
} |
||||
} |
||||
buf.WriteString("HTTP/1.1 101 Switching Protocols\r\n") |
||||
buf.WriteString("Upgrade: websocket\r\n") |
||||
buf.WriteString("Connection: Upgrade\r\n") |
||||
buf.WriteString("Sec-WebSocket-Accept: " + string(c.accept) + "\r\n") |
||||
if len(c.Protocol) > 0 { |
||||
buf.WriteString("Sec-WebSocket-Protocol: " + c.Protocol[0] + "\r\n") |
||||
} |
||||
// TODO(ukai): send Sec-WebSocket-Extensions.
|
||||
if c.Header != nil { |
||||
err := c.Header.WriteSubset(buf, handshakeHeader) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
} |
||||
buf.WriteString("\r\n") |
||||
return buf.Flush() |
||||
} |
||||
|
||||
func (c *hybiServerHandshaker) NewServerConn(buf *bufio.ReadWriter, rwc io.ReadWriteCloser, request *http.Request) *Conn { |
||||
return newHybiServerConn(c.Config, buf, rwc, request) |
||||
} |
||||
|
||||
// newHybiServerConn returns a new WebSocket connection speaking hybi draft protocol.
|
||||
func newHybiServerConn(config *Config, buf *bufio.ReadWriter, rwc io.ReadWriteCloser, request *http.Request) *Conn { |
||||
return newHybiConn(config, buf, rwc, request) |
||||
} |
||||
@ -0,0 +1,590 @@ |
||||
// Copyright 2011 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket |
||||
|
||||
import ( |
||||
"bufio" |
||||
"bytes" |
||||
"fmt" |
||||
"io" |
||||
"net/http" |
||||
"net/url" |
||||
"strings" |
||||
"testing" |
||||
) |
||||
|
||||
// Test the getNonceAccept function with values in
|
||||
// http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-17
|
||||
func TestSecWebSocketAccept(t *testing.T) { |
||||
nonce := []byte("dGhlIHNhbXBsZSBub25jZQ==") |
||||
expected := []byte("s3pPLMBiTxaQ9kYGzzhZRbK+xOo=") |
||||
accept, err := getNonceAccept(nonce) |
||||
if err != nil { |
||||
t.Errorf("getNonceAccept: returned error %v", err) |
||||
return |
||||
} |
||||
if !bytes.Equal(expected, accept) { |
||||
t.Errorf("getNonceAccept: expected %q got %q", expected, accept) |
||||
} |
||||
} |
||||
|
||||
func TestHybiClientHandshake(t *testing.T) { |
||||
b := bytes.NewBuffer([]byte{}) |
||||
bw := bufio.NewWriter(b) |
||||
br := bufio.NewReader(strings.NewReader(`HTTP/1.1 101 Switching Protocols |
||||
Upgrade: websocket |
||||
Connection: Upgrade |
||||
Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo= |
||||
Sec-WebSocket-Protocol: chat |
||||
|
||||
`)) |
||||
var err error |
||||
config := new(Config) |
||||
config.Location, err = url.ParseRequestURI("ws://server.example.com/chat") |
||||
if err != nil { |
||||
t.Fatal("location url", err) |
||||
} |
||||
config.Origin, err = url.ParseRequestURI("http://example.com") |
||||
if err != nil { |
||||
t.Fatal("origin url", err) |
||||
} |
||||
config.Protocol = append(config.Protocol, "chat") |
||||
config.Protocol = append(config.Protocol, "superchat") |
||||
config.Version = ProtocolVersionHybi13 |
||||
|
||||
config.handshakeData = map[string]string{ |
||||
"key": "dGhlIHNhbXBsZSBub25jZQ==", |
||||
} |
||||
err = hybiClientHandshake(config, br, bw) |
||||
if err != nil { |
||||
t.Errorf("handshake failed: %v", err) |
||||
} |
||||
req, err := http.ReadRequest(bufio.NewReader(b)) |
||||
if err != nil { |
||||
t.Fatalf("read request: %v", err) |
||||
} |
||||
if req.Method != "GET" { |
||||
t.Errorf("request method expected GET, but got %q", req.Method) |
||||
} |
||||
if req.URL.Path != "/chat" { |
||||
t.Errorf("request path expected /chat, but got %q", req.URL.Path) |
||||
} |
||||
if req.Proto != "HTTP/1.1" { |
||||
t.Errorf("request proto expected HTTP/1.1, but got %q", req.Proto) |
||||
} |
||||
if req.Host != "server.example.com" { |
||||
t.Errorf("request Host expected server.example.com, but got %v", req.Host) |
||||
} |
||||
var expectedHeader = map[string]string{ |
||||
"Connection": "Upgrade", |
||||
"Upgrade": "websocket", |
||||
"Sec-Websocket-Key": config.handshakeData["key"], |
||||
"Origin": config.Origin.String(), |
||||
"Sec-Websocket-Protocol": "chat, superchat", |
||||
"Sec-Websocket-Version": fmt.Sprintf("%d", ProtocolVersionHybi13), |
||||
} |
||||
for k, v := range expectedHeader { |
||||
if req.Header.Get(k) != v { |
||||
t.Errorf(fmt.Sprintf("%s expected %q but got %q", k, v, req.Header.Get(k))) |
||||
} |
||||
} |
||||
} |
||||
|
||||
func TestHybiClientHandshakeWithHeader(t *testing.T) { |
||||
b := bytes.NewBuffer([]byte{}) |
||||
bw := bufio.NewWriter(b) |
||||
br := bufio.NewReader(strings.NewReader(`HTTP/1.1 101 Switching Protocols |
||||
Upgrade: websocket |
||||
Connection: Upgrade |
||||
Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo= |
||||
Sec-WebSocket-Protocol: chat |
||||
|
||||
`)) |
||||
var err error |
||||
config := new(Config) |
||||
config.Location, err = url.ParseRequestURI("ws://server.example.com/chat") |
||||
if err != nil { |
||||
t.Fatal("location url", err) |
||||
} |
||||
config.Origin, err = url.ParseRequestURI("http://example.com") |
||||
if err != nil { |
||||
t.Fatal("origin url", err) |
||||
} |
||||
config.Protocol = append(config.Protocol, "chat") |
||||
config.Protocol = append(config.Protocol, "superchat") |
||||
config.Version = ProtocolVersionHybi13 |
||||
config.Header = http.Header(make(map[string][]string)) |
||||
config.Header.Add("User-Agent", "test") |
||||
|
||||
config.handshakeData = map[string]string{ |
||||
"key": "dGhlIHNhbXBsZSBub25jZQ==", |
||||
} |
||||
err = hybiClientHandshake(config, br, bw) |
||||
if err != nil { |
||||
t.Errorf("handshake failed: %v", err) |
||||
} |
||||
req, err := http.ReadRequest(bufio.NewReader(b)) |
||||
if err != nil { |
||||
t.Fatalf("read request: %v", err) |
||||
} |
||||
if req.Method != "GET" { |
||||
t.Errorf("request method expected GET, but got %q", req.Method) |
||||
} |
||||
if req.URL.Path != "/chat" { |
||||
t.Errorf("request path expected /chat, but got %q", req.URL.Path) |
||||
} |
||||
if req.Proto != "HTTP/1.1" { |
||||
t.Errorf("request proto expected HTTP/1.1, but got %q", req.Proto) |
||||
} |
||||
if req.Host != "server.example.com" { |
||||
t.Errorf("request Host expected server.example.com, but got %v", req.Host) |
||||
} |
||||
var expectedHeader = map[string]string{ |
||||
"Connection": "Upgrade", |
||||
"Upgrade": "websocket", |
||||
"Sec-Websocket-Key": config.handshakeData["key"], |
||||
"Origin": config.Origin.String(), |
||||
"Sec-Websocket-Protocol": "chat, superchat", |
||||
"Sec-Websocket-Version": fmt.Sprintf("%d", ProtocolVersionHybi13), |
||||
"User-Agent": "test", |
||||
} |
||||
for k, v := range expectedHeader { |
||||
if req.Header.Get(k) != v { |
||||
t.Errorf(fmt.Sprintf("%s expected %q but got %q", k, v, req.Header.Get(k))) |
||||
} |
||||
} |
||||
} |
||||
|
||||
func TestHybiServerHandshake(t *testing.T) { |
||||
config := new(Config) |
||||
handshaker := &hybiServerHandshaker{Config: config} |
||||
br := bufio.NewReader(strings.NewReader(`GET /chat HTTP/1.1 |
||||
Host: server.example.com |
||||
Upgrade: websocket |
||||
Connection: Upgrade |
||||
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ== |
||||
Origin: http://example.com
|
||||
Sec-WebSocket-Protocol: chat, superchat |
||||
Sec-WebSocket-Version: 13 |
||||
|
||||
`)) |
||||
req, err := http.ReadRequest(br) |
||||
if err != nil { |
||||
t.Fatal("request", err) |
||||
} |
||||
code, err := handshaker.ReadHandshake(br, req) |
||||
if err != nil { |
||||
t.Errorf("handshake failed: %v", err) |
||||
} |
||||
if code != http.StatusSwitchingProtocols { |
||||
t.Errorf("status expected %q but got %q", http.StatusSwitchingProtocols, code) |
||||
} |
||||
expectedProtocols := []string{"chat", "superchat"} |
||||
if fmt.Sprintf("%v", config.Protocol) != fmt.Sprintf("%v", expectedProtocols) { |
||||
t.Errorf("protocol expected %q but got %q", expectedProtocols, config.Protocol) |
||||
} |
||||
b := bytes.NewBuffer([]byte{}) |
||||
bw := bufio.NewWriter(b) |
||||
|
||||
config.Protocol = config.Protocol[:1] |
||||
|
||||
err = handshaker.AcceptHandshake(bw) |
||||
if err != nil { |
||||
t.Errorf("handshake response failed: %v", err) |
||||
} |
||||
expectedResponse := strings.Join([]string{ |
||||
"HTTP/1.1 101 Switching Protocols", |
||||
"Upgrade: websocket", |
||||
"Connection: Upgrade", |
||||
"Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=", |
||||
"Sec-WebSocket-Protocol: chat", |
||||
"", ""}, "\r\n") |
||||
|
||||
if b.String() != expectedResponse { |
||||
t.Errorf("handshake expected %q but got %q", expectedResponse, b.String()) |
||||
} |
||||
} |
||||
|
||||
func TestHybiServerHandshakeNoSubProtocol(t *testing.T) { |
||||
config := new(Config) |
||||
handshaker := &hybiServerHandshaker{Config: config} |
||||
br := bufio.NewReader(strings.NewReader(`GET /chat HTTP/1.1 |
||||
Host: server.example.com |
||||
Upgrade: websocket |
||||
Connection: Upgrade |
||||
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ== |
||||
Origin: http://example.com
|
||||
Sec-WebSocket-Version: 13 |
||||
|
||||
`)) |
||||
req, err := http.ReadRequest(br) |
||||
if err != nil { |
||||
t.Fatal("request", err) |
||||
} |
||||
code, err := handshaker.ReadHandshake(br, req) |
||||
if err != nil { |
||||
t.Errorf("handshake failed: %v", err) |
||||
} |
||||
if code != http.StatusSwitchingProtocols { |
||||
t.Errorf("status expected %q but got %q", http.StatusSwitchingProtocols, code) |
||||
} |
||||
if len(config.Protocol) != 0 { |
||||
t.Errorf("len(config.Protocol) expected 0, but got %q", len(config.Protocol)) |
||||
} |
||||
b := bytes.NewBuffer([]byte{}) |
||||
bw := bufio.NewWriter(b) |
||||
|
||||
err = handshaker.AcceptHandshake(bw) |
||||
if err != nil { |
||||
t.Errorf("handshake response failed: %v", err) |
||||
} |
||||
expectedResponse := strings.Join([]string{ |
||||
"HTTP/1.1 101 Switching Protocols", |
||||
"Upgrade: websocket", |
||||
"Connection: Upgrade", |
||||
"Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=", |
||||
"", ""}, "\r\n") |
||||
|
||||
if b.String() != expectedResponse { |
||||
t.Errorf("handshake expected %q but got %q", expectedResponse, b.String()) |
||||
} |
||||
} |
||||
|
||||
func TestHybiServerHandshakeHybiBadVersion(t *testing.T) { |
||||
config := new(Config) |
||||
handshaker := &hybiServerHandshaker{Config: config} |
||||
br := bufio.NewReader(strings.NewReader(`GET /chat HTTP/1.1 |
||||
Host: server.example.com |
||||
Upgrade: websocket |
||||
Connection: Upgrade |
||||
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ== |
||||
Sec-WebSocket-Origin: http://example.com
|
||||
Sec-WebSocket-Protocol: chat, superchat |
||||
Sec-WebSocket-Version: 9 |
||||
|
||||
`)) |
||||
req, err := http.ReadRequest(br) |
||||
if err != nil { |
||||
t.Fatal("request", err) |
||||
} |
||||
code, err := handshaker.ReadHandshake(br, req) |
||||
if err != ErrBadWebSocketVersion { |
||||
t.Errorf("handshake expected err %q but got %q", ErrBadWebSocketVersion, err) |
||||
} |
||||
if code != http.StatusBadRequest { |
||||
t.Errorf("status expected %q but got %q", http.StatusBadRequest, code) |
||||
} |
||||
} |
||||
|
||||
func testHybiFrame(t *testing.T, testHeader, testPayload, testMaskedPayload []byte, frameHeader *hybiFrameHeader) { |
||||
b := bytes.NewBuffer([]byte{}) |
||||
frameWriterFactory := &hybiFrameWriterFactory{bufio.NewWriter(b), false} |
||||
w, _ := frameWriterFactory.NewFrameWriter(TextFrame) |
||||
w.(*hybiFrameWriter).header = frameHeader |
||||
_, err := w.Write(testPayload) |
||||
w.Close() |
||||
if err != nil { |
||||
t.Errorf("Write error %q", err) |
||||
} |
||||
var expectedFrame []byte |
||||
expectedFrame = append(expectedFrame, testHeader...) |
||||
expectedFrame = append(expectedFrame, testMaskedPayload...) |
||||
if !bytes.Equal(expectedFrame, b.Bytes()) { |
||||
t.Errorf("frame expected %q got %q", expectedFrame, b.Bytes()) |
||||
} |
||||
frameReaderFactory := &hybiFrameReaderFactory{bufio.NewReader(b)} |
||||
r, err := frameReaderFactory.NewFrameReader() |
||||
if err != nil { |
||||
t.Errorf("Read error %q", err) |
||||
} |
||||
if header := r.HeaderReader(); header == nil { |
||||
t.Errorf("no header") |
||||
} else { |
||||
actualHeader := make([]byte, r.Len()) |
||||
n, err := header.Read(actualHeader) |
||||
if err != nil { |
||||
t.Errorf("Read header error %q", err) |
||||
} else { |
||||
if n < len(testHeader) { |
||||
t.Errorf("header too short %q got %q", testHeader, actualHeader[:n]) |
||||
} |
||||
if !bytes.Equal(testHeader, actualHeader[:n]) { |
||||
t.Errorf("header expected %q got %q", testHeader, actualHeader[:n]) |
||||
} |
||||
} |
||||
} |
||||
if trailer := r.TrailerReader(); trailer != nil { |
||||
t.Errorf("unexpected trailer %q", trailer) |
||||
} |
||||
frame := r.(*hybiFrameReader) |
||||
if frameHeader.Fin != frame.header.Fin || |
||||
frameHeader.OpCode != frame.header.OpCode || |
||||
len(testPayload) != int(frame.header.Length) { |
||||
t.Errorf("mismatch %v (%d) vs %v", frameHeader, len(testPayload), frame) |
||||
} |
||||
payload := make([]byte, len(testPayload)) |
||||
_, err = r.Read(payload) |
||||
if err != nil { |
||||
t.Errorf("read %v", err) |
||||
} |
||||
if !bytes.Equal(testPayload, payload) { |
||||
t.Errorf("payload %q vs %q", testPayload, payload) |
||||
} |
||||
} |
||||
|
||||
func TestHybiShortTextFrame(t *testing.T) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: TextFrame} |
||||
payload := []byte("hello") |
||||
testHybiFrame(t, []byte{0x81, 0x05}, payload, payload, frameHeader) |
||||
|
||||
payload = make([]byte, 125) |
||||
testHybiFrame(t, []byte{0x81, 125}, payload, payload, frameHeader) |
||||
} |
||||
|
||||
func TestHybiShortMaskedTextFrame(t *testing.T) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: TextFrame, |
||||
MaskingKey: []byte{0xcc, 0x55, 0x80, 0x20}} |
||||
payload := []byte("hello") |
||||
maskedPayload := []byte{0xa4, 0x30, 0xec, 0x4c, 0xa3} |
||||
header := []byte{0x81, 0x85} |
||||
header = append(header, frameHeader.MaskingKey...) |
||||
testHybiFrame(t, header, payload, maskedPayload, frameHeader) |
||||
} |
||||
|
||||
func TestHybiShortBinaryFrame(t *testing.T) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: BinaryFrame} |
||||
payload := []byte("hello") |
||||
testHybiFrame(t, []byte{0x82, 0x05}, payload, payload, frameHeader) |
||||
|
||||
payload = make([]byte, 125) |
||||
testHybiFrame(t, []byte{0x82, 125}, payload, payload, frameHeader) |
||||
} |
||||
|
||||
func TestHybiControlFrame(t *testing.T) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: PingFrame} |
||||
payload := []byte("hello") |
||||
testHybiFrame(t, []byte{0x89, 0x05}, payload, payload, frameHeader) |
||||
|
||||
frameHeader = &hybiFrameHeader{Fin: true, OpCode: PongFrame} |
||||
testHybiFrame(t, []byte{0x8A, 0x05}, payload, payload, frameHeader) |
||||
|
||||
frameHeader = &hybiFrameHeader{Fin: true, OpCode: CloseFrame} |
||||
payload = []byte{0x03, 0xe8} // 1000
|
||||
testHybiFrame(t, []byte{0x88, 0x02}, payload, payload, frameHeader) |
||||
} |
||||
|
||||
func TestHybiLongFrame(t *testing.T) { |
||||
frameHeader := &hybiFrameHeader{Fin: true, OpCode: TextFrame} |
||||
payload := make([]byte, 126) |
||||
testHybiFrame(t, []byte{0x81, 126, 0x00, 126}, payload, payload, frameHeader) |
||||
|
||||
payload = make([]byte, 65535) |
||||
testHybiFrame(t, []byte{0x81, 126, 0xff, 0xff}, payload, payload, frameHeader) |
||||
|
||||
payload = make([]byte, 65536) |
||||
testHybiFrame(t, []byte{0x81, 127, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00}, payload, payload, frameHeader) |
||||
} |
||||
|
||||
func TestHybiClientRead(t *testing.T) { |
||||
wireData := []byte{0x81, 0x05, 'h', 'e', 'l', 'l', 'o', |
||||
0x89, 0x05, 'h', 'e', 'l', 'l', 'o', // ping
|
||||
0x81, 0x05, 'w', 'o', 'r', 'l', 'd'} |
||||
br := bufio.NewReader(bytes.NewBuffer(wireData)) |
||||
bw := bufio.NewWriter(bytes.NewBuffer([]byte{})) |
||||
conn := newHybiConn(newConfig(t, "/"), bufio.NewReadWriter(br, bw), nil, nil) |
||||
|
||||
msg := make([]byte, 512) |
||||
n, err := conn.Read(msg) |
||||
if err != nil { |
||||
t.Errorf("read 1st frame, error %q", err) |
||||
} |
||||
if n != 5 { |
||||
t.Errorf("read 1st frame, expect 5, got %d", n) |
||||
} |
||||
if !bytes.Equal(wireData[2:7], msg[:n]) { |
||||
t.Errorf("read 1st frame %v, got %v", wireData[2:7], msg[:n]) |
||||
} |
||||
n, err = conn.Read(msg) |
||||
if err != nil { |
||||
t.Errorf("read 2nd frame, error %q", err) |
||||
} |
||||
if n != 5 { |
||||
t.Errorf("read 2nd frame, expect 5, got %d", n) |
||||
} |
||||
if !bytes.Equal(wireData[16:21], msg[:n]) { |
||||
t.Errorf("read 2nd frame %v, got %v", wireData[16:21], msg[:n]) |
||||
} |
||||
n, err = conn.Read(msg) |
||||
if err == nil { |
||||
t.Errorf("read not EOF") |
||||
} |
||||
if n != 0 { |
||||
t.Errorf("expect read 0, got %d", n) |
||||
} |
||||
} |
||||
|
||||
func TestHybiShortRead(t *testing.T) { |
||||
wireData := []byte{0x81, 0x05, 'h', 'e', 'l', 'l', 'o', |
||||
0x89, 0x05, 'h', 'e', 'l', 'l', 'o', // ping
|
||||
0x81, 0x05, 'w', 'o', 'r', 'l', 'd'} |
||||
br := bufio.NewReader(bytes.NewBuffer(wireData)) |
||||
bw := bufio.NewWriter(bytes.NewBuffer([]byte{})) |
||||
conn := newHybiConn(newConfig(t, "/"), bufio.NewReadWriter(br, bw), nil, nil) |
||||
|
||||
step := 0 |
||||
pos := 0 |
||||
expectedPos := []int{2, 5, 16, 19} |
||||
expectedLen := []int{3, 2, 3, 2} |
||||
for { |
||||
msg := make([]byte, 3) |
||||
n, err := conn.Read(msg) |
||||
if step >= len(expectedPos) { |
||||
if err == nil { |
||||
t.Errorf("read not EOF") |
||||
} |
||||
if n != 0 { |
||||
t.Errorf("expect read 0, got %d", n) |
||||
} |
||||
return |
||||
} |
||||
pos = expectedPos[step] |
||||
endPos := pos + expectedLen[step] |
||||
if err != nil { |
||||
t.Errorf("read from %d, got error %q", pos, err) |
||||
return |
||||
} |
||||
if n != endPos-pos { |
||||
t.Errorf("read from %d, expect %d, got %d", pos, endPos-pos, n) |
||||
} |
||||
if !bytes.Equal(wireData[pos:endPos], msg[:n]) { |
||||
t.Errorf("read from %d, frame %v, got %v", pos, wireData[pos:endPos], msg[:n]) |
||||
} |
||||
step++ |
||||
} |
||||
} |
||||
|
||||
func TestHybiServerRead(t *testing.T) { |
||||
wireData := []byte{0x81, 0x85, 0xcc, 0x55, 0x80, 0x20, |
||||
0xa4, 0x30, 0xec, 0x4c, 0xa3, // hello
|
||||
0x89, 0x85, 0xcc, 0x55, 0x80, 0x20, |
||||
0xa4, 0x30, 0xec, 0x4c, 0xa3, // ping: hello
|
||||
0x81, 0x85, 0xed, 0x83, 0xb4, 0x24, |
||||
0x9a, 0xec, 0xc6, 0x48, 0x89, // world
|
||||
} |
||||
br := bufio.NewReader(bytes.NewBuffer(wireData)) |
||||
bw := bufio.NewWriter(bytes.NewBuffer([]byte{})) |
||||
conn := newHybiConn(newConfig(t, "/"), bufio.NewReadWriter(br, bw), nil, new(http.Request)) |
||||
|
||||
expected := [][]byte{[]byte("hello"), []byte("world")} |
||||
|
||||
msg := make([]byte, 512) |
||||
n, err := conn.Read(msg) |
||||
if err != nil { |
||||
t.Errorf("read 1st frame, error %q", err) |
||||
} |
||||
if n != 5 { |
||||
t.Errorf("read 1st frame, expect 5, got %d", n) |
||||
} |
||||
if !bytes.Equal(expected[0], msg[:n]) { |
||||
t.Errorf("read 1st frame %q, got %q", expected[0], msg[:n]) |
||||
} |
||||
|
||||
n, err = conn.Read(msg) |
||||
if err != nil { |
||||
t.Errorf("read 2nd frame, error %q", err) |
||||
} |
||||
if n != 5 { |
||||
t.Errorf("read 2nd frame, expect 5, got %d", n) |
||||
} |
||||
if !bytes.Equal(expected[1], msg[:n]) { |
||||
t.Errorf("read 2nd frame %q, got %q", expected[1], msg[:n]) |
||||
} |
||||
|
||||
n, err = conn.Read(msg) |
||||
if err == nil { |
||||
t.Errorf("read not EOF") |
||||
} |
||||
if n != 0 { |
||||
t.Errorf("expect read 0, got %d", n) |
||||
} |
||||
} |
||||
|
||||
func TestHybiServerReadWithoutMasking(t *testing.T) { |
||||
wireData := []byte{0x81, 0x05, 'h', 'e', 'l', 'l', 'o'} |
||||
br := bufio.NewReader(bytes.NewBuffer(wireData)) |
||||
bw := bufio.NewWriter(bytes.NewBuffer([]byte{})) |
||||
conn := newHybiConn(newConfig(t, "/"), bufio.NewReadWriter(br, bw), nil, new(http.Request)) |
||||
// server MUST close the connection upon receiving a non-masked frame.
|
||||
msg := make([]byte, 512) |
||||
_, err := conn.Read(msg) |
||||
if err != io.EOF { |
||||
t.Errorf("read 1st frame, expect %q, but got %q", io.EOF, err) |
||||
} |
||||
} |
||||
|
||||
func TestHybiClientReadWithMasking(t *testing.T) { |
||||
wireData := []byte{0x81, 0x85, 0xcc, 0x55, 0x80, 0x20, |
||||
0xa4, 0x30, 0xec, 0x4c, 0xa3, // hello
|
||||
} |
||||
br := bufio.NewReader(bytes.NewBuffer(wireData)) |
||||
bw := bufio.NewWriter(bytes.NewBuffer([]byte{})) |
||||
conn := newHybiConn(newConfig(t, "/"), bufio.NewReadWriter(br, bw), nil, nil) |
||||
|
||||
// client MUST close the connection upon receiving a masked frame.
|
||||
msg := make([]byte, 512) |
||||
_, err := conn.Read(msg) |
||||
if err != io.EOF { |
||||
t.Errorf("read 1st frame, expect %q, but got %q", io.EOF, err) |
||||
} |
||||
} |
||||
|
||||
// Test the hybiServerHandshaker supports firefox implementation and
|
||||
// checks Connection request header include (but it's not necessary
|
||||
// equal to) "upgrade"
|
||||
func TestHybiServerFirefoxHandshake(t *testing.T) { |
||||
config := new(Config) |
||||
handshaker := &hybiServerHandshaker{Config: config} |
||||
br := bufio.NewReader(strings.NewReader(`GET /chat HTTP/1.1 |
||||
Host: server.example.com |
||||
Upgrade: websocket |
||||
Connection: keep-alive, upgrade |
||||
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ== |
||||
Origin: http://example.com
|
||||
Sec-WebSocket-Protocol: chat, superchat |
||||
Sec-WebSocket-Version: 13 |
||||
|
||||
`)) |
||||
req, err := http.ReadRequest(br) |
||||
if err != nil { |
||||
t.Fatal("request", err) |
||||
} |
||||
code, err := handshaker.ReadHandshake(br, req) |
||||
if err != nil { |
||||
t.Errorf("handshake failed: %v", err) |
||||
} |
||||
if code != http.StatusSwitchingProtocols { |
||||
t.Errorf("status expected %q but got %q", http.StatusSwitchingProtocols, code) |
||||
} |
||||
b := bytes.NewBuffer([]byte{}) |
||||
bw := bufio.NewWriter(b) |
||||
|
||||
config.Protocol = []string{"chat"} |
||||
|
||||
err = handshaker.AcceptHandshake(bw) |
||||
if err != nil { |
||||
t.Errorf("handshake response failed: %v", err) |
||||
} |
||||
expectedResponse := strings.Join([]string{ |
||||
"HTTP/1.1 101 Switching Protocols", |
||||
"Upgrade: websocket", |
||||
"Connection: Upgrade", |
||||
"Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=", |
||||
"Sec-WebSocket-Protocol: chat", |
||||
"", ""}, "\r\n") |
||||
|
||||
if b.String() != expectedResponse { |
||||
t.Errorf("handshake expected %q but got %q", expectedResponse, b.String()) |
||||
} |
||||
} |
||||
@ -0,0 +1,114 @@ |
||||
// Copyright 2009 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket |
||||
|
||||
import ( |
||||
"bufio" |
||||
"fmt" |
||||
"io" |
||||
"net/http" |
||||
) |
||||
|
||||
func newServerConn(rwc io.ReadWriteCloser, buf *bufio.ReadWriter, req *http.Request, config *Config, handshake func(*Config, *http.Request) error) (conn *Conn, err error) { |
||||
var hs serverHandshaker = &hybiServerHandshaker{Config: config} |
||||
code, err := hs.ReadHandshake(buf.Reader, req) |
||||
if err == ErrBadWebSocketVersion { |
||||
fmt.Fprintf(buf, "HTTP/1.1 %03d %s\r\n", code, http.StatusText(code)) |
||||
fmt.Fprintf(buf, "Sec-WebSocket-Version: %s\r\n", SupportedProtocolVersion) |
||||
buf.WriteString("\r\n") |
||||
buf.WriteString(err.Error()) |
||||
buf.Flush() |
||||
return |
||||
} |
||||
if err != nil { |
||||
fmt.Fprintf(buf, "HTTP/1.1 %03d %s\r\n", code, http.StatusText(code)) |
||||
buf.WriteString("\r\n") |
||||
buf.WriteString(err.Error()) |
||||
buf.Flush() |
||||
return |
||||
} |
||||
if handshake != nil { |
||||
err = handshake(config, req) |
||||
if err != nil { |
||||
code = http.StatusForbidden |
||||
fmt.Fprintf(buf, "HTTP/1.1 %03d %s\r\n", code, http.StatusText(code)) |
||||
buf.WriteString("\r\n") |
||||
buf.Flush() |
||||
return |
||||
} |
||||
} |
||||
err = hs.AcceptHandshake(buf.Writer) |
||||
if err != nil { |
||||
code = http.StatusBadRequest |
||||
fmt.Fprintf(buf, "HTTP/1.1 %03d %s\r\n", code, http.StatusText(code)) |
||||
buf.WriteString("\r\n") |
||||
buf.Flush() |
||||
return |
||||
} |
||||
conn = hs.NewServerConn(buf, rwc, req) |
||||
return |
||||
} |
||||
|
||||
// Server represents a server of a WebSocket.
|
||||
type Server struct { |
||||
// Config is a WebSocket configuration for new WebSocket connection.
|
||||
Config |
||||
|
||||
// Handshake is an optional function in WebSocket handshake.
|
||||
// For example, you can check, or don't check Origin header.
|
||||
// Another example, you can select config.Protocol.
|
||||
Handshake func(*Config, *http.Request) error |
||||
|
||||
// Handler handles a WebSocket connection.
|
||||
Handler |
||||
} |
||||
|
||||
// ServeHTTP implements the http.Handler interface for a WebSocket
|
||||
func (s Server) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
||||
s.serveWebSocket(w, req) |
||||
} |
||||
|
||||
func (s Server) serveWebSocket(w http.ResponseWriter, req *http.Request) { |
||||
rwc, buf, err := w.(http.Hijacker).Hijack() |
||||
if err != nil { |
||||
panic("Hijack failed: " + err.Error()) |
||||
return |
||||
} |
||||
// The server should abort the WebSocket connection if it finds
|
||||
// the client did not send a handshake that matches with protocol
|
||||
// specification.
|
||||
defer rwc.Close() |
||||
conn, err := newServerConn(rwc, buf, req, &s.Config, s.Handshake) |
||||
if err != nil { |
||||
return |
||||
} |
||||
if conn == nil { |
||||
panic("unexpected nil conn") |
||||
} |
||||
s.Handler(conn) |
||||
} |
||||
|
||||
// Handler is a simple interface to a WebSocket browser client.
|
||||
// It checks if Origin header is valid URL by default.
|
||||
// You might want to verify websocket.Conn.Config().Origin in the func.
|
||||
// If you use Server instead of Handler, you could call websocket.Origin and
|
||||
// check the origin in your Handshake func. So, if you want to accept
|
||||
// non-browser client, which doesn't send Origin header, you could use Server
|
||||
//. that doesn't check origin in its Handshake.
|
||||
type Handler func(*Conn) |
||||
|
||||
func checkOrigin(config *Config, req *http.Request) (err error) { |
||||
config.Origin, err = Origin(config, req) |
||||
if err == nil && config.Origin == nil { |
||||
return fmt.Errorf("null origin") |
||||
} |
||||
return err |
||||
} |
||||
|
||||
// ServeHTTP implements the http.Handler interface for a WebSocket
|
||||
func (h Handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
||||
s := Server{Handler: h, Handshake: checkOrigin} |
||||
s.serveWebSocket(w, req) |
||||
} |
||||
@ -0,0 +1,411 @@ |
||||
// Copyright 2009 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// Package websocket implements a client and server for the WebSocket protocol
|
||||
// as specified in RFC 6455.
|
||||
package websocket |
||||
|
||||
import ( |
||||
"bufio" |
||||
"crypto/tls" |
||||
"encoding/json" |
||||
"errors" |
||||
"io" |
||||
"io/ioutil" |
||||
"net" |
||||
"net/http" |
||||
"net/url" |
||||
"sync" |
||||
"time" |
||||
) |
||||
|
||||
const ( |
||||
ProtocolVersionHybi13 = 13 |
||||
ProtocolVersionHybi = ProtocolVersionHybi13 |
||||
SupportedProtocolVersion = "13" |
||||
|
||||
ContinuationFrame = 0 |
||||
TextFrame = 1 |
||||
BinaryFrame = 2 |
||||
CloseFrame = 8 |
||||
PingFrame = 9 |
||||
PongFrame = 10 |
||||
UnknownFrame = 255 |
||||
) |
||||
|
||||
// ProtocolError represents WebSocket protocol errors.
|
||||
type ProtocolError struct { |
||||
ErrorString string |
||||
} |
||||
|
||||
func (err *ProtocolError) Error() string { return err.ErrorString } |
||||
|
||||
var ( |
||||
ErrBadProtocolVersion = &ProtocolError{"bad protocol version"} |
||||
ErrBadScheme = &ProtocolError{"bad scheme"} |
||||
ErrBadStatus = &ProtocolError{"bad status"} |
||||
ErrBadUpgrade = &ProtocolError{"missing or bad upgrade"} |
||||
ErrBadWebSocketOrigin = &ProtocolError{"missing or bad WebSocket-Origin"} |
||||
ErrBadWebSocketLocation = &ProtocolError{"missing or bad WebSocket-Location"} |
||||
ErrBadWebSocketProtocol = &ProtocolError{"missing or bad WebSocket-Protocol"} |
||||
ErrBadWebSocketVersion = &ProtocolError{"missing or bad WebSocket Version"} |
||||
ErrChallengeResponse = &ProtocolError{"mismatch challenge/response"} |
||||
ErrBadFrame = &ProtocolError{"bad frame"} |
||||
ErrBadFrameBoundary = &ProtocolError{"not on frame boundary"} |
||||
ErrNotWebSocket = &ProtocolError{"not websocket protocol"} |
||||
ErrBadRequestMethod = &ProtocolError{"bad method"} |
||||
ErrNotSupported = &ProtocolError{"not supported"} |
||||
) |
||||
|
||||
// Addr is an implementation of net.Addr for WebSocket.
|
||||
type Addr struct { |
||||
*url.URL |
||||
} |
||||
|
||||
// Network returns the network type for a WebSocket, "websocket".
|
||||
func (addr *Addr) Network() string { return "websocket" } |
||||
|
||||
// Config is a WebSocket configuration
|
||||
type Config struct { |
||||
// A WebSocket server address.
|
||||
Location *url.URL |
||||
|
||||
// A Websocket client origin.
|
||||
Origin *url.URL |
||||
|
||||
// WebSocket subprotocols.
|
||||
Protocol []string |
||||
|
||||
// WebSocket protocol version.
|
||||
Version int |
||||
|
||||
// TLS config for secure WebSocket (wss).
|
||||
TlsConfig *tls.Config |
||||
|
||||
// Additional header fields to be sent in WebSocket opening handshake.
|
||||
Header http.Header |
||||
|
||||
handshakeData map[string]string |
||||
} |
||||
|
||||
// serverHandshaker is an interface to handle WebSocket server side handshake.
|
||||
type serverHandshaker interface { |
||||
// ReadHandshake reads handshake request message from client.
|
||||
// Returns http response code and error if any.
|
||||
ReadHandshake(buf *bufio.Reader, req *http.Request) (code int, err error) |
||||
|
||||
// AcceptHandshake accepts the client handshake request and sends
|
||||
// handshake response back to client.
|
||||
AcceptHandshake(buf *bufio.Writer) (err error) |
||||
|
||||
// NewServerConn creates a new WebSocket connection.
|
||||
NewServerConn(buf *bufio.ReadWriter, rwc io.ReadWriteCloser, request *http.Request) (conn *Conn) |
||||
} |
||||
|
||||
// frameReader is an interface to read a WebSocket frame.
|
||||
type frameReader interface { |
||||
// Reader is to read payload of the frame.
|
||||
io.Reader |
||||
|
||||
// PayloadType returns payload type.
|
||||
PayloadType() byte |
||||
|
||||
// HeaderReader returns a reader to read header of the frame.
|
||||
HeaderReader() io.Reader |
||||
|
||||
// TrailerReader returns a reader to read trailer of the frame.
|
||||
// If it returns nil, there is no trailer in the frame.
|
||||
TrailerReader() io.Reader |
||||
|
||||
// Len returns total length of the frame, including header and trailer.
|
||||
Len() int |
||||
} |
||||
|
||||
// frameReaderFactory is an interface to creates new frame reader.
|
||||
type frameReaderFactory interface { |
||||
NewFrameReader() (r frameReader, err error) |
||||
} |
||||
|
||||
// frameWriter is an interface to write a WebSocket frame.
|
||||
type frameWriter interface { |
||||
// Writer is to write payload of the frame.
|
||||
io.WriteCloser |
||||
} |
||||
|
||||
// frameWriterFactory is an interface to create new frame writer.
|
||||
type frameWriterFactory interface { |
||||
NewFrameWriter(payloadType byte) (w frameWriter, err error) |
||||
} |
||||
|
||||
type frameHandler interface { |
||||
HandleFrame(frame frameReader) (r frameReader, err error) |
||||
WriteClose(status int) (err error) |
||||
} |
||||
|
||||
// Conn represents a WebSocket connection.
|
||||
type Conn struct { |
||||
config *Config |
||||
request *http.Request |
||||
|
||||
buf *bufio.ReadWriter |
||||
rwc io.ReadWriteCloser |
||||
|
||||
rio sync.Mutex |
||||
frameReaderFactory |
||||
frameReader |
||||
|
||||
wio sync.Mutex |
||||
frameWriterFactory |
||||
|
||||
frameHandler |
||||
PayloadType byte |
||||
defaultCloseStatus int |
||||
} |
||||
|
||||
// Read implements the io.Reader interface:
|
||||
// it reads data of a frame from the WebSocket connection.
|
||||
// if msg is not large enough for the frame data, it fills the msg and next Read
|
||||
// will read the rest of the frame data.
|
||||
// it reads Text frame or Binary frame.
|
||||
func (ws *Conn) Read(msg []byte) (n int, err error) { |
||||
ws.rio.Lock() |
||||
defer ws.rio.Unlock() |
||||
again: |
||||
if ws.frameReader == nil { |
||||
frame, err := ws.frameReaderFactory.NewFrameReader() |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
ws.frameReader, err = ws.frameHandler.HandleFrame(frame) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
if ws.frameReader == nil { |
||||
goto again |
||||
} |
||||
} |
||||
n, err = ws.frameReader.Read(msg) |
||||
if err == io.EOF { |
||||
if trailer := ws.frameReader.TrailerReader(); trailer != nil { |
||||
io.Copy(ioutil.Discard, trailer) |
||||
} |
||||
ws.frameReader = nil |
||||
goto again |
||||
} |
||||
return n, err |
||||
} |
||||
|
||||
// Write implements the io.Writer interface:
|
||||
// it writes data as a frame to the WebSocket connection.
|
||||
func (ws *Conn) Write(msg []byte) (n int, err error) { |
||||
ws.wio.Lock() |
||||
defer ws.wio.Unlock() |
||||
w, err := ws.frameWriterFactory.NewFrameWriter(ws.PayloadType) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
n, err = w.Write(msg) |
||||
w.Close() |
||||
if err != nil { |
||||
return n, err |
||||
} |
||||
return n, err |
||||
} |
||||
|
||||
// Close implements the io.Closer interface.
|
||||
func (ws *Conn) Close() error { |
||||
err := ws.frameHandler.WriteClose(ws.defaultCloseStatus) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
return ws.rwc.Close() |
||||
} |
||||
|
||||
func (ws *Conn) IsClientConn() bool { return ws.request == nil } |
||||
func (ws *Conn) IsServerConn() bool { return ws.request != nil } |
||||
|
||||
// LocalAddr returns the WebSocket Origin for the connection for client, or
|
||||
// the WebSocket location for server.
|
||||
func (ws *Conn) LocalAddr() net.Addr { |
||||
if ws.IsClientConn() { |
||||
return &Addr{ws.config.Origin} |
||||
} |
||||
return &Addr{ws.config.Location} |
||||
} |
||||
|
||||
// RemoteAddr returns the WebSocket location for the connection for client, or
|
||||
// the Websocket Origin for server.
|
||||
func (ws *Conn) RemoteAddr() net.Addr { |
||||
if ws.IsClientConn() { |
||||
return &Addr{ws.config.Location} |
||||
} |
||||
return &Addr{ws.config.Origin} |
||||
} |
||||
|
||||
var errSetDeadline = errors.New("websocket: cannot set deadline: not using a net.Conn") |
||||
|
||||
// SetDeadline sets the connection's network read & write deadlines.
|
||||
func (ws *Conn) SetDeadline(t time.Time) error { |
||||
if conn, ok := ws.rwc.(net.Conn); ok { |
||||
return conn.SetDeadline(t) |
||||
} |
||||
return errSetDeadline |
||||
} |
||||
|
||||
// SetReadDeadline sets the connection's network read deadline.
|
||||
func (ws *Conn) SetReadDeadline(t time.Time) error { |
||||
if conn, ok := ws.rwc.(net.Conn); ok { |
||||
return conn.SetReadDeadline(t) |
||||
} |
||||
return errSetDeadline |
||||
} |
||||
|
||||
// SetWriteDeadline sets the connection's network write deadline.
|
||||
func (ws *Conn) SetWriteDeadline(t time.Time) error { |
||||
if conn, ok := ws.rwc.(net.Conn); ok { |
||||
return conn.SetWriteDeadline(t) |
||||
} |
||||
return errSetDeadline |
||||
} |
||||
|
||||
// Config returns the WebSocket config.
|
||||
func (ws *Conn) Config() *Config { return ws.config } |
||||
|
||||
// Request returns the http request upgraded to the WebSocket.
|
||||
// It is nil for client side.
|
||||
func (ws *Conn) Request() *http.Request { return ws.request } |
||||
|
||||
// Codec represents a symmetric pair of functions that implement a codec.
|
||||
type Codec struct { |
||||
Marshal func(v interface{}) (data []byte, payloadType byte, err error) |
||||
Unmarshal func(data []byte, payloadType byte, v interface{}) (err error) |
||||
} |
||||
|
||||
// Send sends v marshaled by cd.Marshal as single frame to ws.
|
||||
func (cd Codec) Send(ws *Conn, v interface{}) (err error) { |
||||
data, payloadType, err := cd.Marshal(v) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
ws.wio.Lock() |
||||
defer ws.wio.Unlock() |
||||
w, err := ws.frameWriterFactory.NewFrameWriter(payloadType) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
_, err = w.Write(data) |
||||
w.Close() |
||||
return err |
||||
} |
||||
|
||||
// Receive receives single frame from ws, unmarshaled by cd.Unmarshal and stores in v.
|
||||
func (cd Codec) Receive(ws *Conn, v interface{}) (err error) { |
||||
ws.rio.Lock() |
||||
defer ws.rio.Unlock() |
||||
if ws.frameReader != nil { |
||||
_, err = io.Copy(ioutil.Discard, ws.frameReader) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
ws.frameReader = nil |
||||
} |
||||
again: |
||||
frame, err := ws.frameReaderFactory.NewFrameReader() |
||||
if err != nil { |
||||
return err |
||||
} |
||||
frame, err = ws.frameHandler.HandleFrame(frame) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
if frame == nil { |
||||
goto again |
||||
} |
||||
payloadType := frame.PayloadType() |
||||
data, err := ioutil.ReadAll(frame) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
return cd.Unmarshal(data, payloadType, v) |
||||
} |
||||
|
||||
func marshal(v interface{}) (msg []byte, payloadType byte, err error) { |
||||
switch data := v.(type) { |
||||
case string: |
||||
return []byte(data), TextFrame, nil |
||||
case []byte: |
||||
return data, BinaryFrame, nil |
||||
} |
||||
return nil, UnknownFrame, ErrNotSupported |
||||
} |
||||
|
||||
func unmarshal(msg []byte, payloadType byte, v interface{}) (err error) { |
||||
switch data := v.(type) { |
||||
case *string: |
||||
*data = string(msg) |
||||
return nil |
||||
case *[]byte: |
||||
*data = msg |
||||
return nil |
||||
} |
||||
return ErrNotSupported |
||||
} |
||||
|
||||
/* |
||||
Message is a codec to send/receive text/binary data in a frame on WebSocket connection. |
||||
To send/receive text frame, use string type. |
||||
To send/receive binary frame, use []byte type. |
||||
|
||||
Trivial usage: |
||||
|
||||
import "websocket" |
||||
|
||||
// receive text frame
|
||||
var message string |
||||
websocket.Message.Receive(ws, &message) |
||||
|
||||
// send text frame
|
||||
message = "hello" |
||||
websocket.Message.Send(ws, message) |
||||
|
||||
// receive binary frame
|
||||
var data []byte |
||||
websocket.Message.Receive(ws, &data) |
||||
|
||||
// send binary frame
|
||||
data = []byte{0, 1, 2} |
||||
websocket.Message.Send(ws, data) |
||||
|
||||
*/ |
||||
var Message = Codec{marshal, unmarshal} |
||||
|
||||
func jsonMarshal(v interface{}) (msg []byte, payloadType byte, err error) { |
||||
msg, err = json.Marshal(v) |
||||
return msg, TextFrame, err |
||||
} |
||||
|
||||
func jsonUnmarshal(msg []byte, payloadType byte, v interface{}) (err error) { |
||||
return json.Unmarshal(msg, v) |
||||
} |
||||
|
||||
/* |
||||
JSON is a codec to send/receive JSON data in a frame from a WebSocket connection. |
||||
|
||||
Trivial usage: |
||||
|
||||
import "websocket" |
||||
|
||||
type T struct { |
||||
Msg string |
||||
Count int |
||||
} |
||||
|
||||
// receive JSON type T
|
||||
var data T |
||||
websocket.JSON.Receive(ws, &data) |
||||
|
||||
// send JSON type T
|
||||
websocket.JSON.Send(ws, data) |
||||
*/ |
||||
var JSON = Codec{jsonMarshal, jsonUnmarshal} |
||||
@ -0,0 +1,414 @@ |
||||
// Copyright 2009 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package websocket |
||||
|
||||
import ( |
||||
"bytes" |
||||
"fmt" |
||||
"io" |
||||
"log" |
||||
"net" |
||||
"net/http" |
||||
"net/http/httptest" |
||||
"net/url" |
||||
"strings" |
||||
"sync" |
||||
"testing" |
||||
) |
||||
|
||||
var serverAddr string |
||||
var once sync.Once |
||||
|
||||
func echoServer(ws *Conn) { io.Copy(ws, ws) } |
||||
|
||||
type Count struct { |
||||
S string |
||||
N int |
||||
} |
||||
|
||||
func countServer(ws *Conn) { |
||||
for { |
||||
var count Count |
||||
err := JSON.Receive(ws, &count) |
||||
if err != nil { |
||||
return |
||||
} |
||||
count.N++ |
||||
count.S = strings.Repeat(count.S, count.N) |
||||
err = JSON.Send(ws, count) |
||||
if err != nil { |
||||
return |
||||
} |
||||
} |
||||
} |
||||
|
||||
func subProtocolHandshake(config *Config, req *http.Request) error { |
||||
for _, proto := range config.Protocol { |
||||
if proto == "chat" { |
||||
config.Protocol = []string{proto} |
||||
return nil |
||||
} |
||||
} |
||||
return ErrBadWebSocketProtocol |
||||
} |
||||
|
||||
func subProtoServer(ws *Conn) { |
||||
for _, proto := range ws.Config().Protocol { |
||||
io.WriteString(ws, proto) |
||||
} |
||||
} |
||||
|
||||
func startServer() { |
||||
http.Handle("/echo", Handler(echoServer)) |
||||
http.Handle("/count", Handler(countServer)) |
||||
subproto := Server{ |
||||
Handshake: subProtocolHandshake, |
||||
Handler: Handler(subProtoServer), |
||||
} |
||||
http.Handle("/subproto", subproto) |
||||
server := httptest.NewServer(nil) |
||||
serverAddr = server.Listener.Addr().String() |
||||
log.Print("Test WebSocket server listening on ", serverAddr) |
||||
} |
||||
|
||||
func newConfig(t *testing.T, path string) *Config { |
||||
config, _ := NewConfig(fmt.Sprintf("ws://%s%s", serverAddr, path), "http://localhost") |
||||
return config |
||||
} |
||||
|
||||
func TestEcho(t *testing.T) { |
||||
once.Do(startServer) |
||||
|
||||
// websocket.Dial()
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
conn, err := NewClient(newConfig(t, "/echo"), client) |
||||
if err != nil { |
||||
t.Errorf("WebSocket handshake error: %v", err) |
||||
return |
||||
} |
||||
|
||||
msg := []byte("hello, world\n") |
||||
if _, err := conn.Write(msg); err != nil { |
||||
t.Errorf("Write: %v", err) |
||||
} |
||||
var actual_msg = make([]byte, 512) |
||||
n, err := conn.Read(actual_msg) |
||||
if err != nil { |
||||
t.Errorf("Read: %v", err) |
||||
} |
||||
actual_msg = actual_msg[0:n] |
||||
if !bytes.Equal(msg, actual_msg) { |
||||
t.Errorf("Echo: expected %q got %q", msg, actual_msg) |
||||
} |
||||
conn.Close() |
||||
} |
||||
|
||||
func TestAddr(t *testing.T) { |
||||
once.Do(startServer) |
||||
|
||||
// websocket.Dial()
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
conn, err := NewClient(newConfig(t, "/echo"), client) |
||||
if err != nil { |
||||
t.Errorf("WebSocket handshake error: %v", err) |
||||
return |
||||
} |
||||
|
||||
ra := conn.RemoteAddr().String() |
||||
if !strings.HasPrefix(ra, "ws://") || !strings.HasSuffix(ra, "/echo") { |
||||
t.Errorf("Bad remote addr: %v", ra) |
||||
} |
||||
la := conn.LocalAddr().String() |
||||
if !strings.HasPrefix(la, "http://") { |
||||
t.Errorf("Bad local addr: %v", la) |
||||
} |
||||
conn.Close() |
||||
} |
||||
|
||||
func TestCount(t *testing.T) { |
||||
once.Do(startServer) |
||||
|
||||
// websocket.Dial()
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
conn, err := NewClient(newConfig(t, "/count"), client) |
||||
if err != nil { |
||||
t.Errorf("WebSocket handshake error: %v", err) |
||||
return |
||||
} |
||||
|
||||
var count Count |
||||
count.S = "hello" |
||||
if err := JSON.Send(conn, count); err != nil { |
||||
t.Errorf("Write: %v", err) |
||||
} |
||||
if err := JSON.Receive(conn, &count); err != nil { |
||||
t.Errorf("Read: %v", err) |
||||
} |
||||
if count.N != 1 { |
||||
t.Errorf("count: expected %d got %d", 1, count.N) |
||||
} |
||||
if count.S != "hello" { |
||||
t.Errorf("count: expected %q got %q", "hello", count.S) |
||||
} |
||||
if err := JSON.Send(conn, count); err != nil { |
||||
t.Errorf("Write: %v", err) |
||||
} |
||||
if err := JSON.Receive(conn, &count); err != nil { |
||||
t.Errorf("Read: %v", err) |
||||
} |
||||
if count.N != 2 { |
||||
t.Errorf("count: expected %d got %d", 2, count.N) |
||||
} |
||||
if count.S != "hellohello" { |
||||
t.Errorf("count: expected %q got %q", "hellohello", count.S) |
||||
} |
||||
conn.Close() |
||||
} |
||||
|
||||
func TestWithQuery(t *testing.T) { |
||||
once.Do(startServer) |
||||
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
|
||||
config := newConfig(t, "/echo") |
||||
config.Location, err = url.ParseRequestURI(fmt.Sprintf("ws://%s/echo?q=v", serverAddr)) |
||||
if err != nil { |
||||
t.Fatal("location url", err) |
||||
} |
||||
|
||||
ws, err := NewClient(config, client) |
||||
if err != nil { |
||||
t.Errorf("WebSocket handshake: %v", err) |
||||
return |
||||
} |
||||
ws.Close() |
||||
} |
||||
|
||||
func testWithProtocol(t *testing.T, subproto []string) (string, error) { |
||||
once.Do(startServer) |
||||
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
|
||||
config := newConfig(t, "/subproto") |
||||
config.Protocol = subproto |
||||
|
||||
ws, err := NewClient(config, client) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
msg := make([]byte, 16) |
||||
n, err := ws.Read(msg) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
ws.Close() |
||||
return string(msg[:n]), nil |
||||
} |
||||
|
||||
func TestWithProtocol(t *testing.T) { |
||||
proto, err := testWithProtocol(t, []string{"chat"}) |
||||
if err != nil { |
||||
t.Errorf("SubProto: unexpected error: %v", err) |
||||
} |
||||
if proto != "chat" { |
||||
t.Errorf("SubProto: expected %q, got %q", "chat", proto) |
||||
} |
||||
} |
||||
|
||||
func TestWithTwoProtocol(t *testing.T) { |
||||
proto, err := testWithProtocol(t, []string{"test", "chat"}) |
||||
if err != nil { |
||||
t.Errorf("SubProto: unexpected error: %v", err) |
||||
} |
||||
if proto != "chat" { |
||||
t.Errorf("SubProto: expected %q, got %q", "chat", proto) |
||||
} |
||||
} |
||||
|
||||
func TestWithBadProtocol(t *testing.T) { |
||||
_, err := testWithProtocol(t, []string{"test"}) |
||||
if err != ErrBadStatus { |
||||
t.Errorf("SubProto: expected %v, got %v", ErrBadStatus, err) |
||||
} |
||||
} |
||||
|
||||
func TestHTTP(t *testing.T) { |
||||
once.Do(startServer) |
||||
|
||||
// If the client did not send a handshake that matches the protocol
|
||||
// specification, the server MUST return an HTTP response with an
|
||||
// appropriate error code (such as 400 Bad Request)
|
||||
resp, err := http.Get(fmt.Sprintf("http://%s/echo", serverAddr)) |
||||
if err != nil { |
||||
t.Errorf("Get: error %#v", err) |
||||
return |
||||
} |
||||
if resp == nil { |
||||
t.Error("Get: resp is null") |
||||
return |
||||
} |
||||
if resp.StatusCode != http.StatusBadRequest { |
||||
t.Errorf("Get: expected %q got %q", http.StatusBadRequest, resp.StatusCode) |
||||
} |
||||
} |
||||
|
||||
func TestTrailingSpaces(t *testing.T) { |
||||
// http://code.google.com/p/go/issues/detail?id=955
|
||||
// The last runs of this create keys with trailing spaces that should not be
|
||||
// generated by the client.
|
||||
once.Do(startServer) |
||||
config := newConfig(t, "/echo") |
||||
for i := 0; i < 30; i++ { |
||||
// body
|
||||
ws, err := DialConfig(config) |
||||
if err != nil { |
||||
t.Errorf("Dial #%d failed: %v", i, err) |
||||
break |
||||
} |
||||
ws.Close() |
||||
} |
||||
} |
||||
|
||||
func TestDialConfigBadVersion(t *testing.T) { |
||||
once.Do(startServer) |
||||
config := newConfig(t, "/echo") |
||||
config.Version = 1234 |
||||
|
||||
_, err := DialConfig(config) |
||||
|
||||
if dialerr, ok := err.(*DialError); ok { |
||||
if dialerr.Err != ErrBadProtocolVersion { |
||||
t.Errorf("dial expected err %q but got %q", ErrBadProtocolVersion, dialerr.Err) |
||||
} |
||||
} |
||||
} |
||||
|
||||
func TestSmallBuffer(t *testing.T) { |
||||
// http://code.google.com/p/go/issues/detail?id=1145
|
||||
// Read should be able to handle reading a fragment of a frame.
|
||||
once.Do(startServer) |
||||
|
||||
// websocket.Dial()
|
||||
client, err := net.Dial("tcp", serverAddr) |
||||
if err != nil { |
||||
t.Fatal("dialing", err) |
||||
} |
||||
conn, err := NewClient(newConfig(t, "/echo"), client) |
||||
if err != nil { |
||||
t.Errorf("WebSocket handshake error: %v", err) |
||||
return |
||||
} |
||||
|
||||
msg := []byte("hello, world\n") |
||||
if _, err := conn.Write(msg); err != nil { |
||||
t.Errorf("Write: %v", err) |
||||
} |
||||
var small_msg = make([]byte, 8) |
||||
n, err := conn.Read(small_msg) |
||||
if err != nil { |
||||
t.Errorf("Read: %v", err) |
||||
} |
||||
if !bytes.Equal(msg[:len(small_msg)], small_msg) { |
||||
t.Errorf("Echo: expected %q got %q", msg[:len(small_msg)], small_msg) |
||||
} |
||||
var second_msg = make([]byte, len(msg)) |
||||
n, err = conn.Read(second_msg) |
||||
if err != nil { |
||||
t.Errorf("Read: %v", err) |
||||
} |
||||
second_msg = second_msg[0:n] |
||||
if !bytes.Equal(msg[len(small_msg):], second_msg) { |
||||
t.Errorf("Echo: expected %q got %q", msg[len(small_msg):], second_msg) |
||||
} |
||||
conn.Close() |
||||
} |
||||
|
||||
var parseAuthorityTests = []struct { |
||||
in *url.URL |
||||
out string |
||||
}{ |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "ws", |
||||
Host: "www.google.com", |
||||
}, |
||||
"www.google.com:80", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "wss", |
||||
Host: "www.google.com", |
||||
}, |
||||
"www.google.com:443", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "ws", |
||||
Host: "www.google.com:80", |
||||
}, |
||||
"www.google.com:80", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "wss", |
||||
Host: "www.google.com:443", |
||||
}, |
||||
"www.google.com:443", |
||||
}, |
||||
// some invalid ones for parseAuthority. parseAuthority doesn't
|
||||
// concern itself with the scheme unless it actually knows about it
|
||||
{ |
||||
&url.URL{ |
||||
Scheme: "http", |
||||
Host: "www.google.com", |
||||
}, |
||||
"www.google.com", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "http", |
||||
Host: "www.google.com:80", |
||||
}, |
||||
"www.google.com:80", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "asdf", |
||||
Host: "127.0.0.1", |
||||
}, |
||||
"127.0.0.1", |
||||
}, |
||||
{ |
||||
&url.URL{ |
||||
Scheme: "asdf", |
||||
Host: "www.google.com", |
||||
}, |
||||
"www.google.com", |
||||
}, |
||||
} |
||||
|
||||
func TestParseAuthority(t *testing.T) { |
||||
for _, tt := range parseAuthorityTests { |
||||
out := parseAuthority(tt.in) |
||||
if out != tt.out { |
||||
t.Errorf("got %v; want %v", out, tt.out) |
||||
} |
||||
} |
||||
} |
||||
@ -1,74 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package utils |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/crypto" |
||||
"github.com/ethereum/go-ethereum/node" |
||||
"github.com/ethereum/go-ethereum/p2p" |
||||
rpc "github.com/ethereum/go-ethereum/rpc/v2" |
||||
) |
||||
|
||||
// PublicWeb3API offers helper utils
|
||||
type PublicWeb3API struct { |
||||
stack *node.Node |
||||
} |
||||
|
||||
// NewPublicWeb3API creates a new Web3Service instance
|
||||
func NewPublicWeb3API(stack *node.Node) *PublicWeb3API { |
||||
return &PublicWeb3API{stack} |
||||
} |
||||
|
||||
// ClientVersion returns the node name
|
||||
func (s *PublicWeb3API) ClientVersion() string { |
||||
return s.stack.Server().Name |
||||
} |
||||
|
||||
// Sha3 applies the ethereum sha3 implementation on the input.
|
||||
// It assumes the input is hex encoded.
|
||||
func (s *PublicWeb3API) Sha3(input string) string { |
||||
return common.ToHex(crypto.Sha3(common.FromHex(input))) |
||||
} |
||||
|
||||
// PublicNetAPI offers network related RPC methods
|
||||
type PublicNetAPI struct { |
||||
net *p2p.Server |
||||
networkVersion int |
||||
} |
||||
|
||||
// NewPublicNetAPI creates a new net api instance.
|
||||
func NewPublicNetAPI(net *p2p.Server, networkVersion int) *PublicNetAPI { |
||||
return &PublicNetAPI{net, networkVersion} |
||||
} |
||||
|
||||
// Listening returns an indication if the node is listening for network connections.
|
||||
func (s *PublicNetAPI) Listening() bool { |
||||
return true // always listening
|
||||
} |
||||
|
||||
// Peercount returns the number of connected peers
|
||||
func (s *PublicNetAPI) PeerCount() *rpc.HexNumber { |
||||
return rpc.NewHexNumber(s.net.PeerCount()) |
||||
} |
||||
|
||||
// ProtocolVersion returns the current ethereum protocol version.
|
||||
func (s *PublicNetAPI) Version() string { |
||||
return fmt.Sprintf("%d", s.networkVersion) |
||||
} |
||||
@ -0,0 +1,176 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package utils |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
|
||||
"strings" |
||||
|
||||
"github.com/codegangsta/cli" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/node" |
||||
"github.com/ethereum/go-ethereum/rpc" |
||||
) |
||||
|
||||
// NewInProcRPCClient will start a new RPC server for the given node and returns a client to interact with it.
|
||||
func NewInProcRPCClient(stack *node.Node) *inProcClient { |
||||
server := rpc.NewServer() |
||||
|
||||
offered := stack.APIs() |
||||
for _, api := range offered { |
||||
server.RegisterName(api.Namespace, api.Service) |
||||
} |
||||
|
||||
web3 := node.NewPublicWeb3API(stack) |
||||
server.RegisterName("web3", web3) |
||||
|
||||
var ethereum *eth.Ethereum |
||||
if err := stack.Service(ðereum); err == nil { |
||||
net := eth.NewPublicNetAPI(stack.Server(), ethereum.NetVersion()) |
||||
server.RegisterName("net", net) |
||||
} else { |
||||
glog.V(logger.Warn).Infof("%v\n", err) |
||||
} |
||||
|
||||
buf := &buf{ |
||||
requests: make(chan []byte), |
||||
responses: make(chan []byte), |
||||
} |
||||
client := &inProcClient{ |
||||
server: server, |
||||
buf: buf, |
||||
} |
||||
|
||||
go func() { |
||||
server.ServeCodec(rpc.NewJSONCodec(client.buf)) |
||||
}() |
||||
|
||||
return client |
||||
} |
||||
|
||||
// buf represents the connection between the RPC server and console
|
||||
type buf struct { |
||||
readBuf []byte // store remaining request bytes after a partial read
|
||||
requests chan []byte // list with raw serialized requests
|
||||
responses chan []byte // list with raw serialized responses
|
||||
} |
||||
|
||||
// will read the next request in json format
|
||||
func (b *buf) Read(p []byte) (int, error) { |
||||
// last read didn't read entire request, return remaining bytes
|
||||
if len(b.readBuf) > 0 { |
||||
n := copy(p, b.readBuf) |
||||
if n < len(b.readBuf) { |
||||
b.readBuf = b.readBuf[:n] |
||||
} else { |
||||
b.readBuf = b.readBuf[:0] |
||||
} |
||||
return n, nil |
||||
} |
||||
|
||||
// read next request
|
||||
req := <-b.requests |
||||
n := copy(p, req) |
||||
if n < len(req) { |
||||
// buf too small, store remaining chunk for next read
|
||||
b.readBuf = req[n:] |
||||
} |
||||
|
||||
return n, nil |
||||
} |
||||
|
||||
// Write send the given buffer to the backend
|
||||
func (b *buf) Write(p []byte) (n int, err error) { |
||||
b.responses <- p |
||||
return len(p), nil |
||||
} |
||||
|
||||
// Close cleans up obtained resources.
|
||||
func (b *buf) Close() error { |
||||
close(b.requests) |
||||
close(b.responses) |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// inProcClient starts a RPC server and uses buf to communicate with it.
|
||||
type inProcClient struct { |
||||
server *rpc.Server |
||||
buf *buf |
||||
} |
||||
|
||||
// Close will stop the RPC server
|
||||
func (c *inProcClient) Close() { |
||||
c.server.Stop() |
||||
} |
||||
|
||||
// Send a msg to the endpoint
|
||||
func (c *inProcClient) Send(msg interface{}) error { |
||||
d, err := json.Marshal(msg) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
c.buf.requests <- d |
||||
return nil |
||||
} |
||||
|
||||
// Recv reads a message and tries to parse it into the given msg
|
||||
func (c *inProcClient) Recv(msg interface{}) error { |
||||
data := <-c.buf.responses |
||||
return json.Unmarshal(data, &msg) |
||||
} |
||||
|
||||
// Returns the collection of modules the RPC server offers.
|
||||
func (c *inProcClient) SupportedModules() (map[string]string, error) { |
||||
return rpc.SupportedModules(c) |
||||
} |
||||
|
||||
// NewRemoteRPCClient returns a RPC client which connects to a running geth instance.
|
||||
// Depending on the given context this can either be a IPC or a HTTP client.
|
||||
func NewRemoteRPCClient(ctx *cli.Context) (rpc.Client, error) { |
||||
if ctx.Args().Present() { |
||||
endpoint := ctx.Args().First() |
||||
return NewRemoteRPCClientFromString(endpoint) |
||||
} |
||||
|
||||
// use IPC by default
|
||||
endpoint := IPCSocketPath(ctx) |
||||
return rpc.NewIPCClient(endpoint) |
||||
} |
||||
|
||||
// NewRemoteRPCClientFromString returns a RPC client which connects to the given
|
||||
// endpoint. It must start with either `ipc:` or `rpc:` (HTTP).
|
||||
func NewRemoteRPCClientFromString(endpoint string) (rpc.Client, error) { |
||||
if strings.HasPrefix(endpoint, "ipc:") { |
||||
return rpc.NewIPCClient(endpoint[4:]) |
||||
} |
||||
if strings.HasPrefix(endpoint, "rpc:") { |
||||
return rpc.NewHTTPClient(endpoint[4:]) |
||||
} |
||||
if strings.HasPrefix(endpoint, "http://") { |
||||
return rpc.NewHTTPClient(endpoint) |
||||
} |
||||
if strings.HasPrefix(endpoint, "ws:") { |
||||
return rpc.NewWSClient(endpoint) |
||||
} |
||||
|
||||
return nil, fmt.Errorf("invalid endpoint") |
||||
} |
||||
@ -0,0 +1,265 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package ethreg |
||||
|
||||
import ( |
||||
"errors" |
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/accounts" |
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/common/compiler" |
||||
"github.com/ethereum/go-ethereum/core" |
||||
"github.com/ethereum/go-ethereum/core/state" |
||||
"github.com/ethereum/go-ethereum/core/types" |
||||
"github.com/ethereum/go-ethereum/crypto" |
||||
"github.com/ethereum/go-ethereum/ethdb" |
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/common/registrar" |
||||
) |
||||
|
||||
// registryAPIBackend is a backend for an Ethereum Registry.
|
||||
type registryAPIBackend struct { |
||||
bc *core.BlockChain |
||||
chainDb ethdb.Database |
||||
txPool *core.TxPool |
||||
am *accounts.Manager |
||||
} |
||||
|
||||
// PrivateRegistarAPI offers various functions to access the Ethereum registry.
|
||||
type PrivateRegistarAPI struct { |
||||
be *registryAPIBackend |
||||
} |
||||
|
||||
// NewPrivateRegistarAPI creates a new PrivateRegistarAPI instance.
|
||||
func NewPrivateRegistarAPI(bc *core.BlockChain, chainDb ethdb.Database, txPool *core.TxPool, am *accounts.Manager) *PrivateRegistarAPI { |
||||
return &PrivateRegistarAPI{®istryAPIBackend{bc, chainDb, txPool, am}} |
||||
} |
||||
|
||||
// SetGlobalRegistrar allows clients to set the global registry for the node.
|
||||
// This method can be used to deploy a new registry. First zero out the current
|
||||
// address by calling the method with namereg = '0x0' and then call this method
|
||||
// again with '' as namereg. This will submit a transaction to the network which
|
||||
// will deploy a new registry on execution. The TX hash is returned. When called
|
||||
// with namereg '' and the current address is not zero the current global is
|
||||
// address is returned..
|
||||
func (api *PrivateRegistarAPI) SetGlobalRegistrar(namereg string, from common.Address) (string, error) { |
||||
return registrar.New(api.be).SetGlobalRegistrar(namereg, from) |
||||
} |
||||
|
||||
// SetHashReg queries the registry for a hash.
|
||||
func (api *PrivateRegistarAPI) SetHashReg(hashreg string, from common.Address) (string, error) { |
||||
return registrar.New(api.be).SetHashReg(hashreg, from) |
||||
} |
||||
|
||||
// SetUrlHint queries the registry for an url.
|
||||
func (api *PrivateRegistarAPI) SetUrlHint(hashreg string, from common.Address) (string, error) { |
||||
return registrar.New(api.be).SetUrlHint(hashreg, from) |
||||
} |
||||
|
||||
// SaveInfo stores contract information on the local file system.
|
||||
func (api *PrivateRegistarAPI) SaveInfo(info *compiler.ContractInfo, filename string) (contenthash common.Hash, err error) { |
||||
return compiler.SaveInfo(info, filename) |
||||
} |
||||
|
||||
// Register registers a new content hash in the registry.
|
||||
func (api *PrivateRegistarAPI) Register(sender common.Address, addr common.Address, contentHashHex string) (bool, error) { |
||||
block := api.be.bc.CurrentBlock() |
||||
state, err := state.New(block.Root(), api.be.chainDb) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
codeb := state.GetCode(addr) |
||||
codeHash := common.BytesToHash(crypto.Sha3(codeb)) |
||||
contentHash := common.HexToHash(contentHashHex) |
||||
|
||||
_, err = registrar.New(api.be).SetHashToHash(sender, codeHash, contentHash) |
||||
return err == nil, err |
||||
} |
||||
|
||||
// RegisterUrl registers a new url in the registry.
|
||||
func (api *PrivateRegistarAPI) RegisterUrl(sender common.Address, contentHashHex string, url string) (bool, error) { |
||||
_, err := registrar.New(api.be).SetUrlToHash(sender, common.HexToHash(contentHashHex), url) |
||||
return err == nil, err |
||||
} |
||||
|
||||
// callmsg is the message type used for call transations.
|
||||
type callmsg struct { |
||||
from *state.StateObject |
||||
to *common.Address |
||||
gas, gasPrice *big.Int |
||||
value *big.Int |
||||
data []byte |
||||
} |
||||
|
||||
// accessor boilerplate to implement core.Message
|
||||
func (m callmsg) From() (common.Address, error) { |
||||
return m.from.Address(), nil |
||||
} |
||||
func (m callmsg) Nonce() uint64 { |
||||
return m.from.Nonce() |
||||
} |
||||
func (m callmsg) To() *common.Address { |
||||
return m.to |
||||
} |
||||
func (m callmsg) GasPrice() *big.Int { |
||||
return m.gasPrice |
||||
} |
||||
func (m callmsg) Gas() *big.Int { |
||||
return m.gas |
||||
} |
||||
func (m callmsg) Value() *big.Int { |
||||
return m.value |
||||
} |
||||
func (m callmsg) Data() []byte { |
||||
return m.data |
||||
} |
||||
|
||||
// Call forms a transaction from the given arguments and tries to execute it on
|
||||
// a private VM with a copy of the state. Any changes are therefore only temporary
|
||||
// and not part of the actual state. This allows for local execution/queries.
|
||||
func (be *registryAPIBackend) Call(fromStr, toStr, valueStr, gasStr, gasPriceStr, dataStr string) (string, string, error) { |
||||
block := be.bc.CurrentBlock() |
||||
statedb, err := state.New(block.Root(), be.chainDb) |
||||
if err != nil { |
||||
return "", "", err |
||||
} |
||||
|
||||
var from *state.StateObject |
||||
if len(fromStr) == 0 { |
||||
accounts, err := be.am.Accounts() |
||||
if err != nil || len(accounts) == 0 { |
||||
from = statedb.GetOrNewStateObject(common.Address{}) |
||||
} else { |
||||
from = statedb.GetOrNewStateObject(accounts[0].Address) |
||||
} |
||||
} else { |
||||
from = statedb.GetOrNewStateObject(common.HexToAddress(fromStr)) |
||||
} |
||||
|
||||
from.SetBalance(common.MaxBig) |
||||
|
||||
msg := callmsg{ |
||||
from: from, |
||||
gas: common.Big(gasStr), |
||||
gasPrice: common.Big(gasPriceStr), |
||||
value: common.Big(valueStr), |
||||
data: common.FromHex(dataStr), |
||||
} |
||||
if len(toStr) > 0 { |
||||
addr := common.HexToAddress(toStr) |
||||
msg.to = &addr |
||||
} |
||||
|
||||
if msg.gas.Cmp(big.NewInt(0)) == 0 { |
||||
msg.gas = big.NewInt(50000000) |
||||
} |
||||
|
||||
if msg.gasPrice.Cmp(big.NewInt(0)) == 0 { |
||||
msg.gasPrice = new(big.Int).Mul(big.NewInt(50), common.Shannon) |
||||
} |
||||
|
||||
header := be.bc.CurrentBlock().Header() |
||||
vmenv := core.NewEnv(statedb, be.bc, msg, header) |
||||
gp := new(core.GasPool).AddGas(common.MaxBig) |
||||
res, gas, err := core.ApplyMessage(vmenv, msg, gp) |
||||
|
||||
return common.ToHex(res), gas.String(), err |
||||
} |
||||
|
||||
// StorageAt returns the data stores in the state for the given address and location.
|
||||
func (be *registryAPIBackend) StorageAt(addr string, storageAddr string) string { |
||||
block := be.bc.CurrentBlock() |
||||
state, err := state.New(block.Root(), be.chainDb) |
||||
if err != nil { |
||||
return "" |
||||
} |
||||
return state.GetState(common.HexToAddress(addr), common.HexToHash(storageAddr)).Hex() |
||||
} |
||||
|
||||
// Transact forms a transaction from the given arguments and submits it to the
|
||||
// transactio pool for execution.
|
||||
func (be *registryAPIBackend) Transact(fromStr, toStr, nonceStr, valueStr, gasStr, gasPriceStr, codeStr string) (string, error) { |
||||
if len(toStr) > 0 && toStr != "0x" && !common.IsHexAddress(toStr) { |
||||
return "", errors.New("invalid address") |
||||
} |
||||
|
||||
var ( |
||||
from = common.HexToAddress(fromStr) |
||||
to = common.HexToAddress(toStr) |
||||
value = common.Big(valueStr) |
||||
gas *big.Int |
||||
price *big.Int |
||||
data []byte |
||||
contractCreation bool |
||||
) |
||||
|
||||
if len(gasStr) == 0 { |
||||
gas = big.NewInt(90000) |
||||
} else { |
||||
gas = common.Big(gasStr) |
||||
} |
||||
|
||||
if len(gasPriceStr) == 0 { |
||||
price = big.NewInt(10000000000000) |
||||
} else { |
||||
price = common.Big(gasPriceStr) |
||||
} |
||||
|
||||
data = common.FromHex(codeStr) |
||||
if len(toStr) == 0 { |
||||
contractCreation = true |
||||
} |
||||
|
||||
nonce := be.txPool.State().GetNonce(from) |
||||
if len(nonceStr) != 0 { |
||||
nonce = common.Big(nonceStr).Uint64() |
||||
} |
||||
|
||||
var tx *types.Transaction |
||||
if contractCreation { |
||||
tx = types.NewContractCreation(nonce, value, gas, price, data) |
||||
} else { |
||||
tx = types.NewTransaction(nonce, to, value, gas, price, data) |
||||
} |
||||
|
||||
acc := accounts.Account{from} |
||||
signature, err := be.am.Sign(acc, tx.SigHash().Bytes()) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
signedTx, err := tx.WithSignature(signature) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
|
||||
be.txPool.SetLocal(signedTx) |
||||
if err := be.txPool.Add(signedTx); err != nil { |
||||
return "", nil |
||||
} |
||||
|
||||
if contractCreation { |
||||
addr := crypto.CreateAddress(from, nonce) |
||||
glog.V(logger.Info).Infof("Tx(%s) created: %s\n", signedTx.Hash().Hex(), addr.Hex()) |
||||
} else { |
||||
glog.V(logger.Info).Infof("Tx(%s) to: %s\n", signedTx.Hash().Hex(), tx.To().Hex()) |
||||
} |
||||
|
||||
return signedTx.Hash().Hex(), nil |
||||
} |
||||
@ -1,48 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package ethreg |
||||
|
||||
import ( |
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/common/registrar" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
// implements a versioned Registrar on an archiving full node
|
||||
type EthReg struct { |
||||
backend *xeth.XEth |
||||
registry *registrar.Registrar |
||||
} |
||||
|
||||
func New(xe *xeth.XEth) (self *EthReg) { |
||||
self = &EthReg{backend: xe} |
||||
self.registry = registrar.New(xe) |
||||
return |
||||
} |
||||
|
||||
func (self *EthReg) Registry() *registrar.Registrar { |
||||
return self.registry |
||||
} |
||||
|
||||
func (self *EthReg) Resolver(n *big.Int) *registrar.Registrar { |
||||
xe := self.backend |
||||
if n != nil { |
||||
xe = self.backend.AtStateNum(n.Int64()) |
||||
} |
||||
return registrar.New(xe) |
||||
} |
||||
@ -1,465 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"fmt" |
||||
"io" |
||||
"math/big" |
||||
"os" |
||||
"time" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/common/compiler" |
||||
"github.com/ethereum/go-ethereum/common/natspec" |
||||
"github.com/ethereum/go-ethereum/common/registrar" |
||||
"github.com/ethereum/go-ethereum/core" |
||||
"github.com/ethereum/go-ethereum/core/types" |
||||
"github.com/ethereum/go-ethereum/crypto" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/node" |
||||
"github.com/ethereum/go-ethereum/p2p/discover" |
||||
"github.com/ethereum/go-ethereum/rlp" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/comms" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/rpc/useragent" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
AdminApiversion = "1.0" |
||||
importBatchSize = 2500 |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
AdminMapping = map[string]adminhandler{ |
||||
"admin_addPeer": (*adminApi).AddPeer, |
||||
"admin_peers": (*adminApi).Peers, |
||||
"admin_nodeInfo": (*adminApi).NodeInfo, |
||||
"admin_exportChain": (*adminApi).ExportChain, |
||||
"admin_importChain": (*adminApi).ImportChain, |
||||
"admin_setSolc": (*adminApi).SetSolc, |
||||
"admin_datadir": (*adminApi).DataDir, |
||||
"admin_startRPC": (*adminApi).StartRPC, |
||||
"admin_stopRPC": (*adminApi).StopRPC, |
||||
"admin_setGlobalRegistrar": (*adminApi).SetGlobalRegistrar, |
||||
"admin_setHashReg": (*adminApi).SetHashReg, |
||||
"admin_setUrlHint": (*adminApi).SetUrlHint, |
||||
"admin_saveInfo": (*adminApi).SaveInfo, |
||||
"admin_register": (*adminApi).Register, |
||||
"admin_registerUrl": (*adminApi).RegisterUrl, |
||||
"admin_startNatSpec": (*adminApi).StartNatSpec, |
||||
"admin_stopNatSpec": (*adminApi).StopNatSpec, |
||||
"admin_getContractInfo": (*adminApi).GetContractInfo, |
||||
"admin_httpGet": (*adminApi).HttpGet, |
||||
"admin_sleepBlocks": (*adminApi).SleepBlocks, |
||||
"admin_sleep": (*adminApi).Sleep, |
||||
"admin_enableUserAgent": (*adminApi).EnableUserAgent, |
||||
} |
||||
) |
||||
|
||||
// admin callback handler
|
||||
type adminhandler func(*adminApi, *shared.Request) (interface{}, error) |
||||
|
||||
// admin api provider
|
||||
type adminApi struct { |
||||
xeth *xeth.XEth |
||||
stack *node.Node |
||||
ethereum *eth.Ethereum |
||||
codec codec.Codec |
||||
coder codec.ApiCoder |
||||
} |
||||
|
||||
// create a new admin api instance
|
||||
func NewAdminApi(xeth *xeth.XEth, stack *node.Node, codec codec.Codec) *adminApi { |
||||
api := &adminApi{ |
||||
xeth: xeth, |
||||
stack: stack, |
||||
codec: codec, |
||||
coder: codec.New(nil), |
||||
} |
||||
if stack != nil { |
||||
stack.Service(&api.ethereum) |
||||
} |
||||
return api |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *adminApi) Methods() []string { |
||||
methods := make([]string, len(AdminMapping)) |
||||
i := 0 |
||||
for k := range AdminMapping { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *adminApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := AdminMapping[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, &shared.NotImplementedError{req.Method} |
||||
} |
||||
|
||||
func (self *adminApi) Name() string { |
||||
return shared.AdminApiName |
||||
} |
||||
|
||||
func (self *adminApi) ApiVersion() string { |
||||
return AdminApiversion |
||||
} |
||||
|
||||
func (self *adminApi) AddPeer(req *shared.Request) (interface{}, error) { |
||||
args := new(AddPeerArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
node, err := discover.ParseNode(args.Url) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("invalid node URL: %v", err) |
||||
} |
||||
self.stack.Server().AddPeer(node) |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) Peers(req *shared.Request) (interface{}, error) { |
||||
return self.stack.Server().PeersInfo(), nil |
||||
} |
||||
|
||||
func (self *adminApi) NodeInfo(req *shared.Request) (interface{}, error) { |
||||
return self.stack.Server().NodeInfo(), nil |
||||
} |
||||
|
||||
func (self *adminApi) DataDir(req *shared.Request) (interface{}, error) { |
||||
return self.stack.DataDir(), nil |
||||
} |
||||
|
||||
func hasAllBlocks(chain *core.BlockChain, bs []*types.Block) bool { |
||||
for _, b := range bs { |
||||
if !chain.HasBlock(b.Hash()) { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
|
||||
func (self *adminApi) ImportChain(req *shared.Request) (interface{}, error) { |
||||
args := new(ImportExportChainArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
fh, err := os.Open(args.Filename) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
defer fh.Close() |
||||
stream := rlp.NewStream(fh, 0) |
||||
|
||||
// Run actual the import.
|
||||
blocks := make(types.Blocks, importBatchSize) |
||||
n := 0 |
||||
for batch := 0; ; batch++ { |
||||
|
||||
i := 0 |
||||
for ; i < importBatchSize; i++ { |
||||
var b types.Block |
||||
if err := stream.Decode(&b); err == io.EOF { |
||||
break |
||||
} else if err != nil { |
||||
return false, fmt.Errorf("at block %d: %v", n, err) |
||||
} |
||||
blocks[i] = &b |
||||
n++ |
||||
} |
||||
if i == 0 { |
||||
break |
||||
} |
||||
// Import the batch.
|
||||
if hasAllBlocks(self.ethereum.BlockChain(), blocks[:i]) { |
||||
continue |
||||
} |
||||
if _, err := self.ethereum.BlockChain().InsertChain(blocks[:i]); err != nil { |
||||
return false, fmt.Errorf("invalid block %d: %v", n, err) |
||||
} |
||||
} |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) ExportChain(req *shared.Request) (interface{}, error) { |
||||
args := new(ImportExportChainArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
fh, err := os.OpenFile(args.Filename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.ModePerm) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
defer fh.Close() |
||||
if err := self.ethereum.BlockChain().Export(fh); err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) SetSolc(req *shared.Request) (interface{}, error) { |
||||
args := new(SetSolcArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
solc, err := self.xeth.SetSolc(args.Path) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return solc.Info(), nil |
||||
} |
||||
|
||||
func (self *adminApi) StartRPC(req *shared.Request) (interface{}, error) { |
||||
args := new(StartRPCArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
cfg := comms.HttpConfig{ |
||||
ListenAddress: args.ListenAddress, |
||||
ListenPort: args.ListenPort, |
||||
CorsDomain: args.CorsDomain, |
||||
} |
||||
|
||||
apis, err := ParseApiString(args.Apis, self.codec, self.xeth, self.stack) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
err = comms.StartHttp(cfg, self.codec, Merge(apis...)) |
||||
if err == nil { |
||||
return true, nil |
||||
} |
||||
return false, err |
||||
} |
||||
|
||||
func (self *adminApi) StopRPC(req *shared.Request) (interface{}, error) { |
||||
comms.StopHttp() |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) SleepBlocks(req *shared.Request) (interface{}, error) { |
||||
args := new(SleepBlocksArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
var timer <-chan time.Time |
||||
var height *big.Int |
||||
var err error |
||||
if args.Timeout > 0 { |
||||
timer = time.NewTimer(time.Duration(args.Timeout) * time.Second).C |
||||
} |
||||
|
||||
height = new(big.Int).Add(self.xeth.CurrentBlock().Number(), big.NewInt(args.N)) |
||||
height, err = sleepBlocks(self.xeth.UpdateState(), height, timer) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return height.Uint64(), nil |
||||
} |
||||
|
||||
func sleepBlocks(wait chan *big.Int, height *big.Int, timer <-chan time.Time) (newHeight *big.Int, err error) { |
||||
wait <- height |
||||
select { |
||||
case <-timer: |
||||
// if times out make sure the xeth loop does not block
|
||||
go func() { |
||||
select { |
||||
case wait <- nil: |
||||
case <-wait: |
||||
} |
||||
}() |
||||
return nil, fmt.Errorf("timeout") |
||||
case newHeight = <-wait: |
||||
} |
||||
return |
||||
} |
||||
|
||||
func (self *adminApi) Sleep(req *shared.Request) (interface{}, error) { |
||||
args := new(SleepArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
time.Sleep(time.Duration(args.S) * time.Second) |
||||
return nil, nil |
||||
} |
||||
|
||||
func (self *adminApi) SetGlobalRegistrar(req *shared.Request) (interface{}, error) { |
||||
args := new(SetGlobalRegistrarArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
sender := common.HexToAddress(args.ContractAddress) |
||||
|
||||
reg := registrar.New(self.xeth) |
||||
txhash, err := reg.SetGlobalRegistrar(args.NameReg, sender) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return txhash, nil |
||||
} |
||||
|
||||
func (self *adminApi) SetHashReg(req *shared.Request) (interface{}, error) { |
||||
args := new(SetHashRegArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
reg := registrar.New(self.xeth) |
||||
sender := common.HexToAddress(args.Sender) |
||||
txhash, err := reg.SetHashReg(args.HashReg, sender) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return txhash, nil |
||||
} |
||||
|
||||
func (self *adminApi) SetUrlHint(req *shared.Request) (interface{}, error) { |
||||
args := new(SetUrlHintArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
urlHint := args.UrlHint |
||||
sender := common.HexToAddress(args.Sender) |
||||
|
||||
reg := registrar.New(self.xeth) |
||||
txhash, err := reg.SetUrlHint(urlHint, sender) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return txhash, nil |
||||
} |
||||
|
||||
func (self *adminApi) SaveInfo(req *shared.Request) (interface{}, error) { |
||||
args := new(SaveInfoArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
contenthash, err := compiler.SaveInfo(&args.ContractInfo, args.Filename) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return contenthash.Hex(), nil |
||||
} |
||||
|
||||
func (self *adminApi) Register(req *shared.Request) (interface{}, error) { |
||||
args := new(RegisterArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
sender := common.HexToAddress(args.Sender) |
||||
// sender and contract address are passed as hex strings
|
||||
codeb := self.xeth.CodeAtBytes(args.Address) |
||||
codeHash := common.BytesToHash(crypto.Sha3(codeb)) |
||||
contentHash := common.HexToHash(args.ContentHashHex) |
||||
registry := registrar.New(self.xeth) |
||||
|
||||
_, err := registry.SetHashToHash(sender, codeHash, contentHash) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) RegisterUrl(req *shared.Request) (interface{}, error) { |
||||
args := new(RegisterUrlArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
sender := common.HexToAddress(args.Sender) |
||||
registry := registrar.New(self.xeth) |
||||
_, err := registry.SetUrlToHash(sender, common.HexToHash(args.ContentHash), args.Url) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) StartNatSpec(req *shared.Request) (interface{}, error) { |
||||
self.ethereum.NatSpec = true |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) StopNatSpec(req *shared.Request) (interface{}, error) { |
||||
self.ethereum.NatSpec = false |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *adminApi) GetContractInfo(req *shared.Request) (interface{}, error) { |
||||
args := new(GetContractInfoArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
infoDoc, err := natspec.FetchDocsForContract(args.Contract, self.xeth, self.ethereum.HTTPClient()) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
var info interface{} |
||||
err = self.coder.Decode(infoDoc, &info) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return info, nil |
||||
} |
||||
|
||||
func (self *adminApi) HttpGet(req *shared.Request) (interface{}, error) { |
||||
args := new(HttpGetArgs) |
||||
if err := self.coder.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
resp, err := self.ethereum.HTTPClient().Get(args.Uri, args.Path) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return string(resp), nil |
||||
} |
||||
|
||||
func (self *adminApi) EnableUserAgent(req *shared.Request) (interface{}, error) { |
||||
if fe, ok := self.xeth.Frontend().(*useragent.RemoteFrontend); ok { |
||||
fe.Enable() |
||||
} |
||||
return true, nil |
||||
} |
||||
@ -1,468 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"github.com/ethereum/go-ethereum/common/compiler" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type AddPeerArgs struct { |
||||
Url string |
||||
} |
||||
|
||||
func (args *AddPeerArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) != 1 { |
||||
return shared.NewDecodeParamError("Expected enode as argument") |
||||
} |
||||
|
||||
urlstr, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("url", "not a string") |
||||
} |
||||
args.Url = urlstr |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type ImportExportChainArgs struct { |
||||
Filename string |
||||
} |
||||
|
||||
func (args *ImportExportChainArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) != 1 { |
||||
return shared.NewDecodeParamError("Expected filename as argument") |
||||
} |
||||
|
||||
filename, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("filename", "not a string") |
||||
} |
||||
args.Filename = filename |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SetSolcArgs struct { |
||||
Path string |
||||
} |
||||
|
||||
func (args *SetSolcArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) != 1 { |
||||
return shared.NewDecodeParamError("Expected path as argument") |
||||
} |
||||
|
||||
if pathstr, ok := obj[0].(string); ok { |
||||
args.Path = pathstr |
||||
return nil |
||||
} |
||||
|
||||
return shared.NewInvalidTypeError("path", "not a string") |
||||
} |
||||
|
||||
type StartRPCArgs struct { |
||||
ListenAddress string |
||||
ListenPort uint |
||||
CorsDomain string |
||||
Apis string |
||||
} |
||||
|
||||
func (args *StartRPCArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
args.ListenAddress = "127.0.0.1" |
||||
args.ListenPort = 8545 |
||||
args.Apis = "net,eth,web3" |
||||
|
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if addr, ok := obj[0].(string); ok { |
||||
args.ListenAddress = addr |
||||
} else { |
||||
return shared.NewInvalidTypeError("listenAddress", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if port, ok := obj[1].(float64); ok && port >= 0 && port <= 64*1024 { |
||||
args.ListenPort = uint(port) |
||||
} else { |
||||
return shared.NewInvalidTypeError("listenPort", "not a valid port number") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 3 && obj[2] != nil { |
||||
if corsDomain, ok := obj[2].(string); ok { |
||||
args.CorsDomain = corsDomain |
||||
} else { |
||||
return shared.NewInvalidTypeError("corsDomain", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 4 && obj[3] != nil { |
||||
if apis, ok := obj[3].(string); ok { |
||||
args.Apis = apis |
||||
} else { |
||||
return shared.NewInvalidTypeError("apis", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SleepArgs struct { |
||||
S int |
||||
} |
||||
|
||||
func (args *SleepArgs) UnmarshalJSON(b []byte) (err error) { |
||||
|
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
if len(obj) >= 1 { |
||||
if obj[0] != nil { |
||||
if n, err := numString(obj[0]); err == nil { |
||||
args.S = int(n.Int64()) |
||||
} else { |
||||
return shared.NewInvalidTypeError("N", "not an integer: "+err.Error()) |
||||
} |
||||
} else { |
||||
return shared.NewInsufficientParamsError(0, 1) |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type SleepBlocksArgs struct { |
||||
N int64 |
||||
Timeout int64 |
||||
} |
||||
|
||||
func (args *SleepBlocksArgs) UnmarshalJSON(b []byte) (err error) { |
||||
|
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
args.N = 1 |
||||
args.Timeout = 0 |
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if n, err := numString(obj[0]); err == nil { |
||||
args.N = n.Int64() |
||||
} else { |
||||
return shared.NewInvalidTypeError("N", "not an integer: "+err.Error()) |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if n, err := numString(obj[1]); err == nil { |
||||
args.Timeout = n.Int64() |
||||
} else { |
||||
return shared.NewInvalidTypeError("Timeout", "not an integer: "+err.Error()) |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SetGlobalRegistrarArgs struct { |
||||
NameReg string |
||||
ContractAddress string |
||||
} |
||||
|
||||
func (args *SetGlobalRegistrarArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) == 0 { |
||||
return shared.NewDecodeParamError("Expected namereg address") |
||||
} |
||||
|
||||
if len(obj) >= 1 { |
||||
if namereg, ok := obj[0].(string); ok { |
||||
args.NameReg = namereg |
||||
} else { |
||||
return shared.NewInvalidTypeError("NameReg", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if addr, ok := obj[1].(string); ok { |
||||
args.ContractAddress = addr |
||||
} else { |
||||
return shared.NewInvalidTypeError("ContractAddress", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SetHashRegArgs struct { |
||||
HashReg string |
||||
Sender string |
||||
} |
||||
|
||||
func (args *SetHashRegArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if hashreg, ok := obj[0].(string); ok { |
||||
args.HashReg = hashreg |
||||
} else { |
||||
return shared.NewInvalidTypeError("HashReg", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if sender, ok := obj[1].(string); ok { |
||||
args.Sender = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("Sender", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SetUrlHintArgs struct { |
||||
UrlHint string |
||||
Sender string |
||||
} |
||||
|
||||
func (args *SetUrlHintArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if urlhint, ok := obj[0].(string); ok { |
||||
args.UrlHint = urlhint |
||||
} else { |
||||
return shared.NewInvalidTypeError("UrlHint", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if sender, ok := obj[1].(string); ok { |
||||
args.Sender = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("Sender", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type SaveInfoArgs struct { |
||||
ContractInfo compiler.ContractInfo |
||||
Filename string |
||||
} |
||||
|
||||
func (args *SaveInfoArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 2 { |
||||
return shared.NewInsufficientParamsError(len(obj), 2) |
||||
} |
||||
|
||||
if jsonraw, err := json.Marshal(obj[0]); err == nil { |
||||
if err = json.Unmarshal(jsonraw, &args.ContractInfo); err != nil { |
||||
return err |
||||
} |
||||
} else { |
||||
return err |
||||
} |
||||
|
||||
if filename, ok := obj[1].(string); ok { |
||||
args.Filename = filename |
||||
} else { |
||||
return shared.NewInvalidTypeError("Filename", "not a string") |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type RegisterArgs struct { |
||||
Sender string |
||||
Address string |
||||
ContentHashHex string |
||||
} |
||||
|
||||
func (args *RegisterArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 3 { |
||||
return shared.NewInsufficientParamsError(len(obj), 3) |
||||
} |
||||
|
||||
if len(obj) >= 1 { |
||||
if sender, ok := obj[0].(string); ok { |
||||
args.Sender = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("Sender", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 { |
||||
if address, ok := obj[1].(string); ok { |
||||
args.Address = address |
||||
} else { |
||||
return shared.NewInvalidTypeError("Address", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 3 { |
||||
if hex, ok := obj[2].(string); ok { |
||||
args.ContentHashHex = hex |
||||
} else { |
||||
return shared.NewInvalidTypeError("ContentHashHex", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type RegisterUrlArgs struct { |
||||
Sender string |
||||
ContentHash string |
||||
Url string |
||||
} |
||||
|
||||
func (args *RegisterUrlArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) >= 1 { |
||||
if sender, ok := obj[0].(string); ok { |
||||
args.Sender = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("Sender", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 { |
||||
if sender, ok := obj[1].(string); ok { |
||||
args.ContentHash = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("ContentHash", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 3 { |
||||
if sender, ok := obj[2].(string); ok { |
||||
args.Url = sender |
||||
} else { |
||||
return shared.NewInvalidTypeError("Url", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type GetContractInfoArgs struct { |
||||
Contract string |
||||
} |
||||
|
||||
func (args *GetContractInfoArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if len(obj) >= 1 { |
||||
if contract, ok := obj[0].(string); ok { |
||||
args.Contract = contract |
||||
} else { |
||||
return shared.NewInvalidTypeError("Contract", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type HttpGetArgs struct { |
||||
Uri string |
||||
Path string |
||||
} |
||||
|
||||
func (args *HttpGetArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if len(obj) >= 1 { |
||||
if uri, ok := obj[0].(string); ok { |
||||
args.Uri = uri |
||||
} else { |
||||
return shared.NewInvalidTypeError("Uri", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if path, ok := obj[1].(string); ok { |
||||
args.Path = path |
||||
} else { |
||||
return shared.NewInvalidTypeError("Path", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
@ -1,143 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Admin_JS = ` |
||||
web3._extend({ |
||||
property: 'admin', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'addPeer', |
||||
call: 'admin_addPeer', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'exportChain', |
||||
call: 'admin_exportChain', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'importChain', |
||||
call: 'admin_importChain', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'sleepBlocks', |
||||
call: 'admin_sleepBlocks', |
||||
params: 2, |
||||
inputFormatter: [null, null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setSolc', |
||||
call: 'admin_setSolc', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'startRPC', |
||||
call: 'admin_startRPC', |
||||
params: 4, |
||||
inputFormatter: [null, null, null, null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'stopRPC', |
||||
call: 'admin_stopRPC', |
||||
params: 0, |
||||
inputFormatter: [] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setGlobalRegistrar', |
||||
call: 'admin_setGlobalRegistrar', |
||||
params: 2, |
||||
inputFormatter: [null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setHashReg', |
||||
call: 'admin_setHashReg', |
||||
params: 2, |
||||
inputFormatter: [null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setUrlHint', |
||||
call: 'admin_setUrlHint', |
||||
params: 2, |
||||
inputFormatter: [null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'saveInfo', |
||||
call: 'admin_saveInfo', |
||||
params: 2, |
||||
inputFormatter: [null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'register', |
||||
call: 'admin_register', |
||||
params: 3, |
||||
inputFormatter: [null,null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'registerUrl', |
||||
call: 'admin_registerUrl', |
||||
params: 3, |
||||
inputFormatter: [null,null,null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'startNatSpec', |
||||
call: 'admin_startNatSpec', |
||||
params: 0, |
||||
inputFormatter: [] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'stopNatSpec', |
||||
call: 'admin_stopNatSpec', |
||||
params: 0, |
||||
inputFormatter: [] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'getContractInfo', |
||||
call: 'admin_getContractInfo', |
||||
params: 1, |
||||
inputFormatter: [null], |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'httpGet', |
||||
call: 'admin_httpGet', |
||||
params: 2, |
||||
inputFormatter: [null, null] |
||||
}) |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'nodeInfo', |
||||
getter: 'admin_nodeInfo' |
||||
}), |
||||
new web3._extend.Property({ |
||||
name: 'peers', |
||||
getter: 'admin_peers' |
||||
}), |
||||
new web3._extend.Property({ |
||||
name: 'datadir', |
||||
getter: 'admin_datadir' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,26 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
// Merge multiple API's to a single API instance
|
||||
func Merge(apis ...shared.EthereumApi) shared.EthereumApi { |
||||
return newMergedApi(apis...) |
||||
} |
||||
@ -1,170 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"encoding/json" |
||||
"strconv" |
||||
|
||||
"github.com/ethereum/go-ethereum/common/compiler" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
func TestParseApiString(t *testing.T) { |
||||
apis, err := ParseApiString("", codec.JSON, nil, nil) |
||||
if err == nil { |
||||
t.Errorf("Expected an err from parsing empty API string but got nil") |
||||
} |
||||
|
||||
if len(apis) != 0 { |
||||
t.Errorf("Expected 0 apis from empty API string") |
||||
} |
||||
|
||||
apis, err = ParseApiString("eth", codec.JSON, nil, nil) |
||||
if err != nil { |
||||
t.Errorf("Expected nil err from parsing empty API string but got %v", err) |
||||
} |
||||
|
||||
if len(apis) != 1 { |
||||
t.Errorf("Expected 1 apis but got %d - %v", apis, apis) |
||||
} |
||||
|
||||
apis, err = ParseApiString("eth,eth", codec.JSON, nil, nil) |
||||
if err != nil { |
||||
t.Errorf("Expected nil err from parsing empty API string but got \"%v\"", err) |
||||
} |
||||
|
||||
if len(apis) != 2 { |
||||
t.Errorf("Expected 2 apis but got %d - %v", apis, apis) |
||||
} |
||||
|
||||
apis, err = ParseApiString("eth,invalid", codec.JSON, nil, nil) |
||||
if err == nil { |
||||
t.Errorf("Expected an err but got no err") |
||||
} |
||||
|
||||
} |
||||
|
||||
const solcVersion = "0.9.23" |
||||
|
||||
func TestCompileSolidity(t *testing.T) { |
||||
|
||||
solc, err := compiler.New("") |
||||
if solc == nil { |
||||
t.Skip("no solc found: skip") |
||||
} else if solc.Version() != solcVersion { |
||||
t.Skip("WARNING: skipping test because of solc different version (%v, test written for %v, may need to update)", solc.Version(), solcVersion) |
||||
} |
||||
source := `contract test {\n` + |
||||
" /// @notice Will multiply `a` by 7." + `\n` + |
||||
` function multiply(uint a) returns(uint d) {\n` + |
||||
` return a * 7;\n` + |
||||
` }\n` + |
||||
`}\n` |
||||
|
||||
jsonstr := `{"jsonrpc":"2.0","method":"eth_compileSolidity","params":["` + source + `"],"id":64}` |
||||
|
||||
expCode := "0x605880600c6000396000f3006000357c010000000000000000000000000000000000000000000000000000000090048063c6888fa114602e57005b603d6004803590602001506047565b8060005260206000f35b60006007820290506053565b91905056" |
||||
expAbiDefinition := `[{"constant":false,"inputs":[{"name":"a","type":"uint256"}],"name":"multiply","outputs":[{"name":"d","type":"uint256"}],"type":"function"}]` |
||||
expUserDoc := `{"methods":{"multiply(uint256)":{"notice":"Will multiply ` + "`a`" + ` by 7."}}}` |
||||
expDeveloperDoc := `{"methods":{}}` |
||||
expCompilerVersion := solc.Version() |
||||
expLanguage := "Solidity" |
||||
expLanguageVersion := "0" |
||||
expSource := source |
||||
|
||||
eth := ð.Ethereum{} |
||||
xeth := xeth.NewTest(nil, nil) |
||||
api := NewEthApi(xeth, eth, codec.JSON) |
||||
|
||||
var rpcRequest shared.Request |
||||
json.Unmarshal([]byte(jsonstr), &rpcRequest) |
||||
|
||||
response, err := api.CompileSolidity(&rpcRequest) |
||||
if err != nil { |
||||
t.Errorf("Execution failed, %v", err) |
||||
} |
||||
|
||||
respjson, err := json.Marshal(response) |
||||
if err != nil { |
||||
t.Errorf("expected no error, got %v", err) |
||||
} |
||||
|
||||
var contracts = make(map[string]*compiler.Contract) |
||||
err = json.Unmarshal(respjson, &contracts) |
||||
if err != nil { |
||||
t.Errorf("expected no error, got %v", err) |
||||
} |
||||
|
||||
if len(contracts) != 1 { |
||||
t.Errorf("expected one contract, got %v", len(contracts)) |
||||
} |
||||
|
||||
contract := contracts["test"] |
||||
|
||||
if contract.Code != expCode { |
||||
t.Errorf("Expected \n%s got \n%s", expCode, contract.Code) |
||||
} |
||||
|
||||
if strconv.Quote(contract.Info.Source) != `"`+expSource+`"` { |
||||
t.Errorf("Expected \n'%s' got \n'%s'", expSource, strconv.Quote(contract.Info.Source)) |
||||
} |
||||
|
||||
if contract.Info.Language != expLanguage { |
||||
t.Errorf("Expected %s got %s", expLanguage, contract.Info.Language) |
||||
} |
||||
|
||||
if contract.Info.LanguageVersion != expLanguageVersion { |
||||
t.Errorf("Expected %s got %s", expLanguageVersion, contract.Info.LanguageVersion) |
||||
} |
||||
|
||||
if contract.Info.CompilerVersion != expCompilerVersion { |
||||
t.Errorf("Expected %s got %s", expCompilerVersion, contract.Info.CompilerVersion) |
||||
} |
||||
|
||||
userdoc, err := json.Marshal(contract.Info.UserDoc) |
||||
if err != nil { |
||||
t.Errorf("expected no error, got %v", err) |
||||
} |
||||
|
||||
devdoc, err := json.Marshal(contract.Info.DeveloperDoc) |
||||
if err != nil { |
||||
t.Errorf("expected no error, got %v", err) |
||||
} |
||||
|
||||
abidef, err := json.Marshal(contract.Info.AbiDefinition) |
||||
if err != nil { |
||||
t.Errorf("expected no error, got %v", err) |
||||
} |
||||
|
||||
if string(abidef) != expAbiDefinition { |
||||
t.Errorf("Expected \n'%s' got \n'%s'", expAbiDefinition, string(abidef)) |
||||
} |
||||
|
||||
if string(userdoc) != expUserDoc { |
||||
t.Errorf("Expected \n'%s' got \n'%s'", expUserDoc, string(userdoc)) |
||||
} |
||||
|
||||
if string(devdoc) != expDeveloperDoc { |
||||
t.Errorf("Expected %s got %s", expDeveloperDoc, string(devdoc)) |
||||
} |
||||
} |
||||
@ -1,74 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type CompileArgs struct { |
||||
Source string |
||||
} |
||||
|
||||
func (args *CompileArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
argstr, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("arg0", "is not a string") |
||||
} |
||||
args.Source = argstr |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type FilterStringArgs struct { |
||||
Word string |
||||
} |
||||
|
||||
func (args *FilterStringArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
var argstr string |
||||
argstr, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("filter", "not a string") |
||||
} |
||||
switch argstr { |
||||
case "latest", "pending": |
||||
break |
||||
default: |
||||
return shared.NewValidationError("Word", "Must be `latest` or `pending`") |
||||
} |
||||
args.Word = argstr |
||||
return nil |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -1,144 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
DbApiversion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
DbMapping = map[string]dbhandler{ |
||||
"db_getString": (*dbApi).GetString, |
||||
"db_putString": (*dbApi).PutString, |
||||
"db_getHex": (*dbApi).GetHex, |
||||
"db_putHex": (*dbApi).PutHex, |
||||
} |
||||
) |
||||
|
||||
// db callback handler
|
||||
type dbhandler func(*dbApi, *shared.Request) (interface{}, error) |
||||
|
||||
// db api provider
|
||||
type dbApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]dbhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new db api instance
|
||||
func NewDbApi(xeth *xeth.XEth, ethereum *eth.Ethereum, coder codec.Codec) *dbApi { |
||||
return &dbApi{ |
||||
xeth: xeth, |
||||
ethereum: ethereum, |
||||
methods: DbMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *dbApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *dbApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, &shared.NotImplementedError{req.Method} |
||||
} |
||||
|
||||
func (self *dbApi) Name() string { |
||||
return shared.DbApiName |
||||
} |
||||
|
||||
func (self *dbApi) ApiVersion() string { |
||||
return DbApiversion |
||||
} |
||||
|
||||
func (self *dbApi) GetString(req *shared.Request) (interface{}, error) { |
||||
args := new(DbArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if err := args.requirements(); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
ret, err := self.xeth.DbGet([]byte(args.Database + args.Key)) |
||||
return string(ret), err |
||||
} |
||||
|
||||
func (self *dbApi) PutString(req *shared.Request) (interface{}, error) { |
||||
args := new(DbArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if err := args.requirements(); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return self.xeth.DbPut([]byte(args.Database+args.Key), args.Value), nil |
||||
} |
||||
|
||||
func (self *dbApi) GetHex(req *shared.Request) (interface{}, error) { |
||||
args := new(DbHexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if err := args.requirements(); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if res, err := self.xeth.DbGet([]byte(args.Database + args.Key)); err == nil { |
||||
return newHexData(res), nil |
||||
} else { |
||||
return nil, err |
||||
} |
||||
} |
||||
|
||||
func (self *dbApi) PutHex(req *shared.Request) (interface{}, error) { |
||||
args := new(DbHexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if err := args.requirements(); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return self.xeth.DbPut([]byte(args.Database+args.Key), args.Value), nil |
||||
} |
||||
@ -1,126 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type DbArgs struct { |
||||
Database string |
||||
Key string |
||||
Value []byte |
||||
} |
||||
|
||||
func (args *DbArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 2 { |
||||
return shared.NewInsufficientParamsError(len(obj), 2) |
||||
} |
||||
|
||||
var objstr string |
||||
var ok bool |
||||
|
||||
if objstr, ok = obj[0].(string); !ok { |
||||
return shared.NewInvalidTypeError("database", "not a string") |
||||
} |
||||
args.Database = objstr |
||||
|
||||
if objstr, ok = obj[1].(string); !ok { |
||||
return shared.NewInvalidTypeError("key", "not a string") |
||||
} |
||||
args.Key = objstr |
||||
|
||||
if len(obj) > 2 { |
||||
objstr, ok = obj[2].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("value", "not a string") |
||||
} |
||||
|
||||
args.Value = []byte(objstr) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (a *DbArgs) requirements() error { |
||||
if len(a.Database) == 0 { |
||||
return shared.NewValidationError("Database", "cannot be blank") |
||||
} |
||||
if len(a.Key) == 0 { |
||||
return shared.NewValidationError("Key", "cannot be blank") |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type DbHexArgs struct { |
||||
Database string |
||||
Key string |
||||
Value []byte |
||||
} |
||||
|
||||
func (args *DbHexArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 2 { |
||||
return shared.NewInsufficientParamsError(len(obj), 2) |
||||
} |
||||
|
||||
var objstr string |
||||
var ok bool |
||||
|
||||
if objstr, ok = obj[0].(string); !ok { |
||||
return shared.NewInvalidTypeError("database", "not a string") |
||||
} |
||||
args.Database = objstr |
||||
|
||||
if objstr, ok = obj[1].(string); !ok { |
||||
return shared.NewInvalidTypeError("key", "not a string") |
||||
} |
||||
args.Key = objstr |
||||
|
||||
if len(obj) > 2 { |
||||
objstr, ok = obj[2].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("value", "not a string") |
||||
} |
||||
|
||||
args.Value = common.FromHex(objstr) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (a *DbHexArgs) requirements() error { |
||||
if len(a.Database) == 0 { |
||||
return shared.NewValidationError("Database", "cannot be blank") |
||||
} |
||||
if len(a.Key) == 0 { |
||||
return shared.NewValidationError("Key", "cannot be blank") |
||||
} |
||||
return nil |
||||
} |
||||
@ -1,29 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Db_JS = ` |
||||
web3._extend({ |
||||
property: 'db', |
||||
methods: |
||||
[ |
||||
], |
||||
properties: |
||||
[ |
||||
] |
||||
}); |
||||
` |
||||
@ -1,303 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strings" |
||||
"time" |
||||
|
||||
"github.com/ethereum/ethash" |
||||
"github.com/ethereum/go-ethereum/core" |
||||
"github.com/ethereum/go-ethereum/core/state" |
||||
"github.com/ethereum/go-ethereum/core/vm" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rlp" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
"github.com/rcrowley/go-metrics" |
||||
) |
||||
|
||||
const ( |
||||
DebugApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
DebugMapping = map[string]debughandler{ |
||||
"debug_dumpBlock": (*debugApi).DumpBlock, |
||||
"debug_getBlockRlp": (*debugApi).GetBlockRlp, |
||||
"debug_printBlock": (*debugApi).PrintBlock, |
||||
"debug_processBlock": (*debugApi).ProcessBlock, |
||||
"debug_seedHash": (*debugApi).SeedHash, |
||||
"debug_setHead": (*debugApi).SetHead, |
||||
"debug_metrics": (*debugApi).Metrics, |
||||
} |
||||
) |
||||
|
||||
// debug callback handler
|
||||
type debughandler func(*debugApi, *shared.Request) (interface{}, error) |
||||
|
||||
// admin api provider
|
||||
type debugApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]debughandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new debug api instance
|
||||
func NewDebugApi(xeth *xeth.XEth, ethereum *eth.Ethereum, coder codec.Codec) *debugApi { |
||||
return &debugApi{ |
||||
xeth: xeth, |
||||
ethereum: ethereum, |
||||
methods: DebugMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *debugApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *debugApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, &shared.NotImplementedError{req.Method} |
||||
} |
||||
|
||||
func (self *debugApi) Name() string { |
||||
return shared.DebugApiName |
||||
} |
||||
|
||||
func (self *debugApi) ApiVersion() string { |
||||
return DebugApiVersion |
||||
} |
||||
|
||||
func (self *debugApi) PrintBlock(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
return fmt.Sprintf("%s", block), nil |
||||
} |
||||
|
||||
func (self *debugApi) DumpBlock(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, fmt.Errorf("block #%d not found", args.BlockNumber) |
||||
} |
||||
|
||||
stateDb, err := state.New(block.Root(), self.ethereum.ChainDb()) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return stateDb.RawDump(), nil |
||||
} |
||||
|
||||
func (self *debugApi) GetBlockRlp(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, fmt.Errorf("block #%d not found", args.BlockNumber) |
||||
} |
||||
encoded, err := rlp.EncodeToBytes(block) |
||||
return fmt.Sprintf("%x", encoded), err |
||||
} |
||||
|
||||
func (self *debugApi) SetHead(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
self.ethereum.BlockChain().SetHead(uint64(args.BlockNumber)) |
||||
|
||||
return nil, nil |
||||
} |
||||
|
||||
func (self *debugApi) ProcessBlock(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, fmt.Errorf("block #%d not found", args.BlockNumber) |
||||
} |
||||
|
||||
old := vm.Debug |
||||
defer func() { vm.Debug = old }() |
||||
vm.Debug = true |
||||
|
||||
var ( |
||||
blockchain = self.ethereum.BlockChain() |
||||
validator = blockchain.Validator() |
||||
processor = blockchain.Processor() |
||||
) |
||||
|
||||
err := core.ValidateHeader(blockchain.AuxValidator(), block.Header(), blockchain.GetHeader(block.ParentHash()), true, false) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
statedb, err := state.New(blockchain.GetBlock(block.ParentHash()).Root(), self.ethereum.ChainDb()) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
receipts, _, usedGas, err := processor.Process(block, statedb) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
err = validator.ValidateState(block, blockchain.GetBlock(block.ParentHash()), statedb, receipts, usedGas) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *debugApi) SeedHash(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if hash, err := ethash.GetSeedHash(uint64(args.BlockNumber)); err == nil { |
||||
return fmt.Sprintf("0x%x", hash), nil |
||||
} else { |
||||
return nil, err |
||||
} |
||||
} |
||||
|
||||
func (self *debugApi) Metrics(req *shared.Request) (interface{}, error) { |
||||
args := new(MetricsArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
// Create a rate formatter
|
||||
units := []string{"", "K", "M", "G", "T", "E", "P"} |
||||
round := func(value float64, prec int) string { |
||||
unit := 0 |
||||
for value >= 1000 { |
||||
unit, value, prec = unit+1, value/1000, 2 |
||||
} |
||||
return fmt.Sprintf(fmt.Sprintf("%%.%df%s", prec, units[unit]), value) |
||||
} |
||||
format := func(total float64, rate float64) string { |
||||
return fmt.Sprintf("%s (%s/s)", round(total, 0), round(rate, 2)) |
||||
} |
||||
// Iterate over all the metrics, and just dump for now
|
||||
counters := make(map[string]interface{}) |
||||
metrics.DefaultRegistry.Each(func(name string, metric interface{}) { |
||||
// Create or retrieve the counter hierarchy for this metric
|
||||
root, parts := counters, strings.Split(name, "/") |
||||
for _, part := range parts[:len(parts)-1] { |
||||
if _, ok := root[part]; !ok { |
||||
root[part] = make(map[string]interface{}) |
||||
} |
||||
root = root[part].(map[string]interface{}) |
||||
} |
||||
name = parts[len(parts)-1] |
||||
|
||||
// Fill the counter with the metric details, formatting if requested
|
||||
if args.Raw { |
||||
switch metric := metric.(type) { |
||||
case metrics.Meter: |
||||
root[name] = map[string]interface{}{ |
||||
"AvgRate01Min": metric.Rate1(), |
||||
"AvgRate05Min": metric.Rate5(), |
||||
"AvgRate15Min": metric.Rate15(), |
||||
"MeanRate": metric.RateMean(), |
||||
"Overall": float64(metric.Count()), |
||||
} |
||||
|
||||
case metrics.Timer: |
||||
root[name] = map[string]interface{}{ |
||||
"AvgRate01Min": metric.Rate1(), |
||||
"AvgRate05Min": metric.Rate5(), |
||||
"AvgRate15Min": metric.Rate15(), |
||||
"MeanRate": metric.RateMean(), |
||||
"Overall": float64(metric.Count()), |
||||
"Percentiles": map[string]interface{}{ |
||||
"5": metric.Percentile(0.05), |
||||
"20": metric.Percentile(0.2), |
||||
"50": metric.Percentile(0.5), |
||||
"80": metric.Percentile(0.8), |
||||
"95": metric.Percentile(0.95), |
||||
}, |
||||
} |
||||
|
||||
default: |
||||
root[name] = "Unknown metric type" |
||||
} |
||||
} else { |
||||
switch metric := metric.(type) { |
||||
case metrics.Meter: |
||||
root[name] = map[string]interface{}{ |
||||
"Avg01Min": format(metric.Rate1()*60, metric.Rate1()), |
||||
"Avg05Min": format(metric.Rate5()*300, metric.Rate5()), |
||||
"Avg15Min": format(metric.Rate15()*900, metric.Rate15()), |
||||
"Overall": format(float64(metric.Count()), metric.RateMean()), |
||||
} |
||||
|
||||
case metrics.Timer: |
||||
root[name] = map[string]interface{}{ |
||||
"Avg01Min": format(metric.Rate1()*60, metric.Rate1()), |
||||
"Avg05Min": format(metric.Rate5()*300, metric.Rate5()), |
||||
"Avg15Min": format(metric.Rate15()*900, metric.Rate15()), |
||||
"Overall": format(float64(metric.Count()), metric.RateMean()), |
||||
"Maximum": time.Duration(metric.Max()).String(), |
||||
"Minimum": time.Duration(metric.Min()).String(), |
||||
"Percentiles": map[string]interface{}{ |
||||
"5": time.Duration(metric.Percentile(0.05)).String(), |
||||
"20": time.Duration(metric.Percentile(0.2)).String(), |
||||
"50": time.Duration(metric.Percentile(0.5)).String(), |
||||
"80": time.Duration(metric.Percentile(0.8)).String(), |
||||
"95": time.Duration(metric.Percentile(0.95)).String(), |
||||
}, |
||||
} |
||||
|
||||
default: |
||||
root[name] = "Unknown metric type" |
||||
} |
||||
} |
||||
}) |
||||
return counters, nil |
||||
} |
||||
@ -1,87 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
"math/big" |
||||
"reflect" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type WaitForBlockArgs struct { |
||||
MinHeight int |
||||
Timeout int // in seconds
|
||||
} |
||||
|
||||
func (args *WaitForBlockArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) > 2 { |
||||
return fmt.Errorf("waitForArgs needs 0, 1, 2 arguments") |
||||
} |
||||
|
||||
// default values when not provided
|
||||
args.MinHeight = -1 |
||||
args.Timeout = -1 |
||||
|
||||
if len(obj) >= 1 { |
||||
var minHeight *big.Int |
||||
if minHeight, err = numString(obj[0]); err != nil { |
||||
return err |
||||
} |
||||
args.MinHeight = int(minHeight.Int64()) |
||||
} |
||||
|
||||
if len(obj) >= 2 { |
||||
timeout, err := numString(obj[1]) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
args.Timeout = int(timeout.Int64()) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type MetricsArgs struct { |
||||
Raw bool |
||||
} |
||||
|
||||
func (args *MetricsArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
if len(obj) > 1 { |
||||
return fmt.Errorf("metricsArgs needs 0, 1 arguments") |
||||
} |
||||
// default values when not provided
|
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if value, ok := obj[0].(bool); !ok { |
||||
return fmt.Errorf("invalid argument %v", reflect.TypeOf(obj[0])) |
||||
} else { |
||||
args.Raw = value |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
@ -1,83 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Debug_JS = ` |
||||
web3._extend({ |
||||
property: 'debug', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'printBlock', |
||||
call: 'debug_printBlock', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'getBlockRlp', |
||||
call: 'debug_getBlockRlp', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setHead', |
||||
call: 'debug_setHead', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'processBlock', |
||||
call: 'debug_processBlock', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'seedHash', |
||||
call: 'debug_seedHash', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'dumpBlock', |
||||
call: 'debug_dumpBlock', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'metrics', |
||||
call: 'debug_metrics', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'verbosity', |
||||
call: 'debug_verbosity', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.utils.fromDecimal] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'vmodule', |
||||
call: 'debug_vmodule', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
], |
||||
properties: |
||||
[ |
||||
] |
||||
}); |
||||
` |
||||
@ -1,721 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/json" |
||||
"math/big" |
||||
|
||||
"fmt" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/common/natspec" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rlp" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
"gopkg.in/fatih/set.v0" |
||||
) |
||||
|
||||
const ( |
||||
EthApiVersion = "1.0" |
||||
) |
||||
|
||||
// eth api provider
|
||||
// See https://github.com/ethereum/wiki/wiki/JSON-RPC
|
||||
type ethApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]ethhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// eth callback handler
|
||||
type ethhandler func(*ethApi, *shared.Request) (interface{}, error) |
||||
|
||||
var ( |
||||
ethMapping = map[string]ethhandler{ |
||||
"eth_accounts": (*ethApi).Accounts, |
||||
"eth_blockNumber": (*ethApi).BlockNumber, |
||||
"eth_getBalance": (*ethApi).GetBalance, |
||||
"eth_protocolVersion": (*ethApi).ProtocolVersion, |
||||
"eth_coinbase": (*ethApi).Coinbase, |
||||
"eth_mining": (*ethApi).IsMining, |
||||
"eth_syncing": (*ethApi).IsSyncing, |
||||
"eth_gasPrice": (*ethApi).GasPrice, |
||||
"eth_getStorage": (*ethApi).GetStorage, |
||||
"eth_storageAt": (*ethApi).GetStorage, |
||||
"eth_getStorageAt": (*ethApi).GetStorageAt, |
||||
"eth_getTransactionCount": (*ethApi).GetTransactionCount, |
||||
"eth_getBlockTransactionCountByHash": (*ethApi).GetBlockTransactionCountByHash, |
||||
"eth_getBlockTransactionCountByNumber": (*ethApi).GetBlockTransactionCountByNumber, |
||||
"eth_getUncleCountByBlockHash": (*ethApi).GetUncleCountByBlockHash, |
||||
"eth_getUncleCountByBlockNumber": (*ethApi).GetUncleCountByBlockNumber, |
||||
"eth_getData": (*ethApi).GetData, |
||||
"eth_getCode": (*ethApi).GetData, |
||||
"eth_getNatSpec": (*ethApi).GetNatSpec, |
||||
"eth_sign": (*ethApi).Sign, |
||||
"eth_sendRawTransaction": (*ethApi).SubmitTransaction, |
||||
"eth_submitTransaction": (*ethApi).SubmitTransaction, |
||||
"eth_sendTransaction": (*ethApi).SendTransaction, |
||||
"eth_signTransaction": (*ethApi).SignTransaction, |
||||
"eth_transact": (*ethApi).SendTransaction, |
||||
"eth_estimateGas": (*ethApi).EstimateGas, |
||||
"eth_call": (*ethApi).Call, |
||||
"eth_flush": (*ethApi).Flush, |
||||
"eth_getBlockByHash": (*ethApi).GetBlockByHash, |
||||
"eth_getBlockByNumber": (*ethApi).GetBlockByNumber, |
||||
"eth_getTransactionByHash": (*ethApi).GetTransactionByHash, |
||||
"eth_getTransactionByBlockNumberAndIndex": (*ethApi).GetTransactionByBlockNumberAndIndex, |
||||
"eth_getTransactionByBlockHashAndIndex": (*ethApi).GetTransactionByBlockHashAndIndex, |
||||
"eth_getUncleByBlockHashAndIndex": (*ethApi).GetUncleByBlockHashAndIndex, |
||||
"eth_getUncleByBlockNumberAndIndex": (*ethApi).GetUncleByBlockNumberAndIndex, |
||||
"eth_getCompilers": (*ethApi).GetCompilers, |
||||
"eth_compileSolidity": (*ethApi).CompileSolidity, |
||||
"eth_newFilter": (*ethApi).NewFilter, |
||||
"eth_newBlockFilter": (*ethApi).NewBlockFilter, |
||||
"eth_newPendingTransactionFilter": (*ethApi).NewPendingTransactionFilter, |
||||
"eth_uninstallFilter": (*ethApi).UninstallFilter, |
||||
"eth_getFilterChanges": (*ethApi).GetFilterChanges, |
||||
"eth_getFilterLogs": (*ethApi).GetFilterLogs, |
||||
"eth_getLogs": (*ethApi).GetLogs, |
||||
"eth_hashrate": (*ethApi).Hashrate, |
||||
"eth_getWork": (*ethApi).GetWork, |
||||
"eth_submitWork": (*ethApi).SubmitWork, |
||||
"eth_submitHashrate": (*ethApi).SubmitHashrate, |
||||
"eth_resend": (*ethApi).Resend, |
||||
"eth_pendingTransactions": (*ethApi).PendingTransactions, |
||||
"eth_getTransactionReceipt": (*ethApi).GetTransactionReceipt, |
||||
} |
||||
) |
||||
|
||||
// create new ethApi instance
|
||||
func NewEthApi(xeth *xeth.XEth, eth *eth.Ethereum, codec codec.Codec) *ethApi { |
||||
return ðApi{xeth, eth, ethMapping, codec.New(nil)} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *ethApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *ethApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *ethApi) Name() string { |
||||
return shared.EthApiName |
||||
} |
||||
|
||||
func (self *ethApi) ApiVersion() string { |
||||
return EthApiVersion |
||||
} |
||||
|
||||
func (self *ethApi) Accounts(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.Accounts(), nil |
||||
} |
||||
|
||||
func (self *ethApi) Hashrate(req *shared.Request) (interface{}, error) { |
||||
return newHexNum(self.xeth.HashRate()), nil |
||||
} |
||||
|
||||
func (self *ethApi) BlockNumber(req *shared.Request) (interface{}, error) { |
||||
num := self.xeth.CurrentBlock().Number() |
||||
return newHexNum(num.Bytes()), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetBalance(req *shared.Request) (interface{}, error) { |
||||
args := new(GetBalanceArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
return self.xeth.AtStateNum(args.BlockNumber).BalanceAt(args.Address), nil |
||||
} |
||||
|
||||
func (self *ethApi) ProtocolVersion(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.EthVersion(), nil |
||||
} |
||||
|
||||
func (self *ethApi) Coinbase(req *shared.Request) (interface{}, error) { |
||||
return newHexData(self.xeth.Coinbase()), nil |
||||
} |
||||
|
||||
func (self *ethApi) IsMining(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.IsMining(), nil |
||||
} |
||||
|
||||
func (self *ethApi) IsSyncing(req *shared.Request) (interface{}, error) { |
||||
origin, current, height := self.ethereum.Downloader().Progress() |
||||
if current < height { |
||||
return map[string]interface{}{ |
||||
"startingBlock": newHexNum(big.NewInt(int64(origin)).Bytes()), |
||||
"currentBlock": newHexNum(big.NewInt(int64(current)).Bytes()), |
||||
"highestBlock": newHexNum(big.NewInt(int64(height)).Bytes()), |
||||
}, nil |
||||
} |
||||
return false, nil |
||||
} |
||||
|
||||
func (self *ethApi) GasPrice(req *shared.Request) (interface{}, error) { |
||||
return newHexNum(self.xeth.DefaultGasPrice().Bytes()), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetStorage(req *shared.Request) (interface{}, error) { |
||||
args := new(GetStorageArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
return self.xeth.AtStateNum(args.BlockNumber).State().SafeGet(args.Address).Storage(), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetStorageAt(req *shared.Request) (interface{}, error) { |
||||
args := new(GetStorageAtArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
return self.xeth.AtStateNum(args.BlockNumber).StorageAt(args.Address, args.Key), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetTransactionCount(req *shared.Request) (interface{}, error) { |
||||
args := new(GetTxCountArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
count := self.xeth.AtStateNum(args.BlockNumber).TxCountAt(args.Address) |
||||
return fmt.Sprintf("%#x", count), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetBlockTransactionCountByHash(req *shared.Request) (interface{}, error) { |
||||
args := new(HashArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
block := self.xeth.EthBlockByHash(args.Hash) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return fmt.Sprintf("%#x", len(block.Transactions())), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetBlockTransactionCountByNumber(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return fmt.Sprintf("%#x", len(block.Transactions())), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetUncleCountByBlockHash(req *shared.Request) (interface{}, error) { |
||||
args := new(HashArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByHash(args.Hash) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return fmt.Sprintf("%#x", len(block.Uncles())), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetUncleCountByBlockNumber(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumArg) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return fmt.Sprintf("%#x", len(block.Uncles())), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetData(req *shared.Request) (interface{}, error) { |
||||
args := new(GetDataArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
v := self.xeth.AtStateNum(args.BlockNumber).CodeAtBytes(args.Address) |
||||
return newHexData(v), nil |
||||
} |
||||
|
||||
func (self *ethApi) Sign(req *shared.Request) (interface{}, error) { |
||||
args := new(NewSigArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
v, err := self.xeth.Sign(args.From, args.Data, false) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
func (self *ethApi) SubmitTransaction(req *shared.Request) (interface{}, error) { |
||||
args := new(NewDataArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
v, err := self.xeth.PushTx(args.Data) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// JsonTransaction is returned as response by the JSON RPC. It contains the
|
||||
// signed RLP encoded transaction as Raw and the signed transaction object as Tx.
|
||||
type JsonTransaction struct { |
||||
Raw string `json:"raw"` |
||||
Tx *tx `json:"tx"` |
||||
} |
||||
|
||||
func (self *ethApi) SignTransaction(req *shared.Request) (interface{}, error) { |
||||
args := new(NewTxArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
// nonce may be nil ("guess" mode)
|
||||
var nonce string |
||||
if args.Nonce != nil { |
||||
nonce = args.Nonce.String() |
||||
} |
||||
|
||||
var gas, price string |
||||
if args.Gas != nil { |
||||
gas = args.Gas.String() |
||||
} |
||||
if args.GasPrice != nil { |
||||
price = args.GasPrice.String() |
||||
} |
||||
tx, err := self.xeth.SignTransaction(args.From, args.To, nonce, args.Value.String(), gas, price, args.Data) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
data, err := rlp.EncodeToBytes(tx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return JsonTransaction{"0x" + common.Bytes2Hex(data), newTx(tx)}, nil |
||||
} |
||||
|
||||
func (self *ethApi) SendTransaction(req *shared.Request) (interface{}, error) { |
||||
args := new(NewTxArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
// nonce may be nil ("guess" mode)
|
||||
var nonce string |
||||
if args.Nonce != nil { |
||||
nonce = args.Nonce.String() |
||||
} |
||||
|
||||
var gas, price string |
||||
if args.Gas != nil { |
||||
gas = args.Gas.String() |
||||
} |
||||
if args.GasPrice != nil { |
||||
price = args.GasPrice.String() |
||||
} |
||||
v, err := self.xeth.Transact(args.From, args.To, nonce, args.Value.String(), gas, price, args.Data) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
func (self *ethApi) GetNatSpec(req *shared.Request) (interface{}, error) { |
||||
args := new(NewTxArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
var jsontx = fmt.Sprintf(`{"params":[{"to":"%s","data": "%s"}]}`, args.To, args.Data) |
||||
notice := natspec.GetNotice(self.xeth, jsontx, self.ethereum.HTTPClient()) |
||||
|
||||
return notice, nil |
||||
} |
||||
|
||||
func (self *ethApi) EstimateGas(req *shared.Request) (interface{}, error) { |
||||
_, gas, err := self.doCall(req.Params) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
// TODO unwrap the parent method's ToHex call
|
||||
if len(gas) == 0 { |
||||
return newHexNum(0), nil |
||||
} else { |
||||
return newHexNum(common.String2Big(gas)), err |
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) Call(req *shared.Request) (interface{}, error) { |
||||
v, _, err := self.doCall(req.Params) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
// TODO unwrap the parent method's ToHex call
|
||||
if v == "0x0" { |
||||
return newHexData([]byte{}), nil |
||||
} else { |
||||
return newHexData(common.FromHex(v)), nil |
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) Flush(req *shared.Request) (interface{}, error) { |
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *ethApi) doCall(params json.RawMessage) (string, string, error) { |
||||
args := new(CallArgs) |
||||
if err := self.codec.Decode(params, &args); err != nil { |
||||
return "", "", err |
||||
} |
||||
|
||||
return self.xeth.AtStateNum(args.BlockNumber).Call(args.From, args.To, args.Value.String(), args.Gas.String(), args.GasPrice.String(), args.Data) |
||||
} |
||||
|
||||
func (self *ethApi) GetBlockByHash(req *shared.Request) (interface{}, error) { |
||||
args := new(GetBlockByHashArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
block := self.xeth.EthBlockByHash(args.BlockHash) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return NewBlockRes(block, self.xeth.Td(block.Hash()), args.IncludeTxs), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetBlockByNumber(req *shared.Request) (interface{}, error) { |
||||
args := new(GetBlockByNumberArgs) |
||||
if err := json.Unmarshal(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
block := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if block == nil { |
||||
return nil, nil |
||||
} |
||||
return NewBlockRes(block, self.xeth.Td(block.Hash()), args.IncludeTxs), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetTransactionByHash(req *shared.Request) (interface{}, error) { |
||||
args := new(HashArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
tx, bhash, bnum, txi := self.xeth.EthTransactionByHash(args.Hash) |
||||
if tx != nil { |
||||
v := NewTransactionRes(tx) |
||||
// if the blockhash is 0, assume this is a pending transaction
|
||||
if bytes.Compare(bhash.Bytes(), bytes.Repeat([]byte{0}, 32)) != 0 { |
||||
v.BlockHash = newHexData(bhash) |
||||
v.BlockNumber = newHexNum(bnum) |
||||
v.TxIndex = newHexNum(txi) |
||||
} |
||||
return v, nil |
||||
} |
||||
return nil, nil |
||||
} |
||||
|
||||
func (self *ethApi) GetTransactionByBlockHashAndIndex(req *shared.Request) (interface{}, error) { |
||||
args := new(HashIndexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
raw := self.xeth.EthBlockByHash(args.Hash) |
||||
if raw == nil { |
||||
return nil, nil |
||||
} |
||||
block := NewBlockRes(raw, self.xeth.Td(raw.Hash()), true) |
||||
if args.Index >= int64(len(block.Transactions)) || args.Index < 0 { |
||||
return nil, nil |
||||
} else { |
||||
return block.Transactions[args.Index], nil |
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) GetTransactionByBlockNumberAndIndex(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumIndexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
raw := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if raw == nil { |
||||
return nil, nil |
||||
} |
||||
block := NewBlockRes(raw, self.xeth.Td(raw.Hash()), true) |
||||
if args.Index >= int64(len(block.Transactions)) || args.Index < 0 { |
||||
// return NewValidationError("Index", "does not exist")
|
||||
return nil, nil |
||||
} |
||||
return block.Transactions[args.Index], nil |
||||
} |
||||
|
||||
func (self *ethApi) GetUncleByBlockHashAndIndex(req *shared.Request) (interface{}, error) { |
||||
args := new(HashIndexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
raw := self.xeth.EthBlockByHash(args.Hash) |
||||
if raw == nil { |
||||
return nil, nil |
||||
} |
||||
block := NewBlockRes(raw, self.xeth.Td(raw.Hash()), false) |
||||
if args.Index >= int64(len(block.Uncles)) || args.Index < 0 { |
||||
// return NewValidationError("Index", "does not exist")
|
||||
return nil, nil |
||||
} |
||||
return block.Uncles[args.Index], nil |
||||
} |
||||
|
||||
func (self *ethApi) GetUncleByBlockNumberAndIndex(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockNumIndexArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
raw := self.xeth.EthBlockByNumber(args.BlockNumber) |
||||
if raw == nil { |
||||
return nil, nil |
||||
} |
||||
block := NewBlockRes(raw, self.xeth.Td(raw.Hash()), true) |
||||
if args.Index >= int64(len(block.Uncles)) || args.Index < 0 { |
||||
return nil, nil |
||||
} else { |
||||
return block.Uncles[args.Index], nil |
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) GetCompilers(req *shared.Request) (interface{}, error) { |
||||
var lang string |
||||
if solc, _ := self.xeth.Solc(); solc != nil { |
||||
lang = "Solidity" |
||||
} |
||||
c := []string{lang} |
||||
return c, nil |
||||
} |
||||
|
||||
func (self *ethApi) CompileSolidity(req *shared.Request) (interface{}, error) { |
||||
solc, _ := self.xeth.Solc() |
||||
if solc == nil { |
||||
return nil, shared.NewNotAvailableError(req.Method, "solc (solidity compiler) not found") |
||||
} |
||||
|
||||
args := new(SourceArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
contracts, err := solc.Compile(args.Source) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return contracts, nil |
||||
} |
||||
|
||||
func (self *ethApi) NewFilter(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockFilterArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
id := self.xeth.NewLogFilter(args.Earliest, args.Latest, args.Skip, args.Max, args.Address, args.Topics) |
||||
return newHexNum(big.NewInt(int64(id)).Bytes()), nil |
||||
} |
||||
|
||||
func (self *ethApi) NewBlockFilter(req *shared.Request) (interface{}, error) { |
||||
return newHexNum(self.xeth.NewBlockFilter()), nil |
||||
} |
||||
|
||||
func (self *ethApi) NewPendingTransactionFilter(req *shared.Request) (interface{}, error) { |
||||
return newHexNum(self.xeth.NewTransactionFilter()), nil |
||||
} |
||||
|
||||
func (self *ethApi) UninstallFilter(req *shared.Request) (interface{}, error) { |
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
return self.xeth.UninstallFilter(args.Id), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetFilterChanges(req *shared.Request) (interface{}, error) { |
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
switch self.xeth.GetFilterType(args.Id) { |
||||
case xeth.BlockFilterTy: |
||||
return NewHashesRes(self.xeth.BlockFilterChanged(args.Id)), nil |
||||
case xeth.TransactionFilterTy: |
||||
return NewHashesRes(self.xeth.TransactionFilterChanged(args.Id)), nil |
||||
case xeth.LogFilterTy: |
||||
return NewLogsRes(self.xeth.LogFilterChanged(args.Id)), nil |
||||
default: |
||||
return []string{}, nil // reply empty string slice
|
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) GetFilterLogs(req *shared.Request) (interface{}, error) { |
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
return NewLogsRes(self.xeth.Logs(args.Id)), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetLogs(req *shared.Request) (interface{}, error) { |
||||
args := new(BlockFilterArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
return NewLogsRes(self.xeth.AllLogs(args.Earliest, args.Latest, args.Skip, args.Max, args.Address, args.Topics)), nil |
||||
} |
||||
|
||||
func (self *ethApi) GetWork(req *shared.Request) (interface{}, error) { |
||||
self.xeth.SetMining(true, 0) |
||||
ret, err := self.xeth.RemoteMining().GetWork() |
||||
if err != nil { |
||||
return nil, shared.NewNotReadyError("mining work") |
||||
} else { |
||||
return ret, nil |
||||
} |
||||
} |
||||
|
||||
func (self *ethApi) SubmitWork(req *shared.Request) (interface{}, error) { |
||||
args := new(SubmitWorkArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
return self.xeth.RemoteMining().SubmitWork(args.Nonce, common.HexToHash(args.Digest), common.HexToHash(args.Header)), nil |
||||
} |
||||
|
||||
func (self *ethApi) SubmitHashrate(req *shared.Request) (interface{}, error) { |
||||
args := new(SubmitHashRateArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return false, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
self.xeth.RemoteMining().SubmitHashrate(common.HexToHash(args.Id), args.Rate) |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *ethApi) Resend(req *shared.Request) (interface{}, error) { |
||||
args := new(ResendArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
from := common.HexToAddress(args.Tx.From) |
||||
|
||||
pending := self.ethereum.TxPool().GetTransactions() |
||||
for _, p := range pending { |
||||
if pFrom, err := p.From(); err == nil && pFrom == from && p.SigHash() == args.Tx.tx.SigHash() { |
||||
self.ethereum.TxPool().RemoveTx(common.HexToHash(args.Tx.Hash)) |
||||
return self.xeth.Transact(args.Tx.From, args.Tx.To, args.Tx.Nonce, args.Tx.Value, args.GasLimit, args.GasPrice, args.Tx.Data) |
||||
} |
||||
} |
||||
|
||||
return nil, fmt.Errorf("Transaction %s not found", args.Tx.Hash) |
||||
} |
||||
|
||||
func (self *ethApi) PendingTransactions(req *shared.Request) (interface{}, error) { |
||||
txs := self.ethereum.TxPool().GetTransactions() |
||||
|
||||
// grab the accounts from the account manager. This will help with determining which
|
||||
// transactions should be returned.
|
||||
accounts, err := self.ethereum.AccountManager().Accounts() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
// Add the accouns to a new set
|
||||
accountSet := set.New() |
||||
for _, account := range accounts { |
||||
accountSet.Add(account.Address) |
||||
} |
||||
|
||||
var ltxs []*tx |
||||
for _, tx := range txs { |
||||
if from, _ := tx.From(); accountSet.Has(from) { |
||||
ltxs = append(ltxs, newTx(tx)) |
||||
} |
||||
} |
||||
|
||||
return ltxs, nil |
||||
} |
||||
|
||||
func (self *ethApi) GetTransactionReceipt(req *shared.Request) (interface{}, error) { |
||||
args := new(HashArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
txhash := common.BytesToHash(common.FromHex(args.Hash)) |
||||
tx, bhash, bnum, txi := self.xeth.EthTransactionByHash(args.Hash) |
||||
rec := self.xeth.GetTxReceipt(txhash) |
||||
// We could have an error of "not found". Should disambiguate
|
||||
// if err != nil {
|
||||
// return err, nil
|
||||
// }
|
||||
if rec != nil && tx != nil { |
||||
v := NewReceiptRes(rec) |
||||
v.BlockHash = newHexData(bhash) |
||||
v.BlockNumber = newHexNum(bnum) |
||||
v.TransactionIndex = newHexNum(txi) |
||||
return v, nil |
||||
} |
||||
|
||||
return nil, nil |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
@ -1,66 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
// JS api provided by web3.js
|
||||
// eth_sign not standard
|
||||
|
||||
const Eth_JS = ` |
||||
web3._extend({ |
||||
property: 'eth', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'sign', |
||||
call: 'eth_sign', |
||||
params: 2, |
||||
inputFormatter: [web3._extend.utils.toAddress, null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'resend', |
||||
call: 'eth_resend', |
||||
params: 3, |
||||
inputFormatter: [web3._extend.formatters.inputTransactionFormatter, web3._extend.utils.fromDecimal, web3._extend.utils.fromDecimal] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'getNatSpec', |
||||
call: 'eth_getNatSpec', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.formatters.inputTransactionFormatter] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'signTransaction', |
||||
call: 'eth_signTransaction', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.formatters.inputTransactionFormatter] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'submitTransaction', |
||||
call: 'eth_submitTransaction', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.formatters.inputTransactionFormatter] |
||||
}) |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'pendingTransactions', |
||||
getter: 'eth_pendingTransactions' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,88 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
const ( |
||||
MergedApiVersion = "1.0" |
||||
) |
||||
|
||||
// combines multiple API's
|
||||
type MergedApi struct { |
||||
apis map[string]string |
||||
methods map[string]shared.EthereumApi |
||||
} |
||||
|
||||
// create new merged api instance
|
||||
func newMergedApi(apis ...shared.EthereumApi) *MergedApi { |
||||
mergedApi := new(MergedApi) |
||||
mergedApi.apis = make(map[string]string, len(apis)) |
||||
mergedApi.methods = make(map[string]shared.EthereumApi) |
||||
|
||||
for _, api := range apis { |
||||
if api != nil { |
||||
mergedApi.apis[api.Name()] = api.ApiVersion() |
||||
for _, method := range api.Methods() { |
||||
mergedApi.methods[method] = api |
||||
} |
||||
} |
||||
} |
||||
return mergedApi |
||||
} |
||||
|
||||
// Supported RPC methods
|
||||
func (self *MergedApi) Methods() []string { |
||||
all := make([]string, len(self.methods)) |
||||
for method, _ := range self.methods { |
||||
all = append(all, method) |
||||
} |
||||
return all |
||||
} |
||||
|
||||
// Call the correct API's Execute method for the given request
|
||||
func (self *MergedApi) Execute(req *shared.Request) (interface{}, error) { |
||||
glog.V(logger.Detail).Infof("%s %s", req.Method, req.Params) |
||||
|
||||
if res, _ := self.handle(req); res != nil { |
||||
return res, nil |
||||
} |
||||
if api, found := self.methods[req.Method]; found { |
||||
return api.Execute(req) |
||||
} |
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *MergedApi) Name() string { |
||||
return shared.MergedApiName |
||||
} |
||||
|
||||
func (self *MergedApi) ApiVersion() string { |
||||
return MergedApiVersion |
||||
} |
||||
|
||||
func (self *MergedApi) handle(req *shared.Request) (interface{}, error) { |
||||
if req.Method == "modules" { // provided API's
|
||||
return self.apis, nil |
||||
} |
||||
|
||||
return nil, nil |
||||
} |
||||
@ -1,177 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/ethash" |
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
const ( |
||||
MinerApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
MinerMapping = map[string]minerhandler{ |
||||
"miner_hashrate": (*minerApi).Hashrate, |
||||
"miner_makeDAG": (*minerApi).MakeDAG, |
||||
"miner_setExtra": (*minerApi).SetExtra, |
||||
"miner_setGasPrice": (*minerApi).SetGasPrice, |
||||
"miner_setEtherbase": (*minerApi).SetEtherbase, |
||||
"miner_startAutoDAG": (*minerApi).StartAutoDAG, |
||||
"miner_start": (*minerApi).StartMiner, |
||||
"miner_stopAutoDAG": (*minerApi).StopAutoDAG, |
||||
"miner_stop": (*minerApi).StopMiner, |
||||
} |
||||
) |
||||
|
||||
// miner callback handler
|
||||
type minerhandler func(*minerApi, *shared.Request) (interface{}, error) |
||||
|
||||
// miner api provider
|
||||
type minerApi struct { |
||||
ethereum *eth.Ethereum |
||||
methods map[string]minerhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new miner api instance
|
||||
func NewMinerApi(ethereum *eth.Ethereum, coder codec.Codec) *minerApi { |
||||
return &minerApi{ |
||||
ethereum: ethereum, |
||||
methods: MinerMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *minerApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, &shared.NotImplementedError{req.Method} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *minerApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
func (self *minerApi) Name() string { |
||||
return shared.MinerApiName |
||||
} |
||||
|
||||
func (self *minerApi) ApiVersion() string { |
||||
return MinerApiVersion |
||||
} |
||||
|
||||
func (self *minerApi) StartMiner(req *shared.Request) (interface{}, error) { |
||||
args := new(StartMinerArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
if args.Threads == -1 { // (not specified by user, use default)
|
||||
args.Threads = self.ethereum.MinerThreads |
||||
} |
||||
|
||||
self.ethereum.StartAutoDAG() |
||||
err := self.ethereum.StartMining(args.Threads, "") |
||||
if err == nil { |
||||
return true, nil |
||||
} |
||||
|
||||
return false, err |
||||
} |
||||
|
||||
func (self *minerApi) StopMiner(req *shared.Request) (interface{}, error) { |
||||
self.ethereum.StopMining() |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *minerApi) Hashrate(req *shared.Request) (interface{}, error) { |
||||
return self.ethereum.Miner().HashRate(), nil |
||||
} |
||||
|
||||
func (self *minerApi) SetExtra(req *shared.Request) (interface{}, error) { |
||||
args := new(SetExtraArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if err := self.ethereum.Miner().SetExtra([]byte(args.Data)); err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *minerApi) SetGasPrice(req *shared.Request) (interface{}, error) { |
||||
args := new(GasPriceArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
self.ethereum.Miner().SetGasPrice(common.String2Big(args.Price)) |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *minerApi) SetEtherbase(req *shared.Request) (interface{}, error) { |
||||
args := new(SetEtherbaseArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return false, err |
||||
} |
||||
self.ethereum.SetEtherbase(args.Etherbase) |
||||
return nil, nil |
||||
} |
||||
|
||||
func (self *minerApi) StartAutoDAG(req *shared.Request) (interface{}, error) { |
||||
self.ethereum.StartAutoDAG() |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *minerApi) StopAutoDAG(req *shared.Request) (interface{}, error) { |
||||
self.ethereum.StopAutoDAG() |
||||
return true, nil |
||||
} |
||||
|
||||
func (self *minerApi) MakeDAG(req *shared.Request) (interface{}, error) { |
||||
args := new(MakeDAGArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if args.BlockNumber < 0 { |
||||
return false, shared.NewValidationError("BlockNumber", "BlockNumber must be positive") |
||||
} |
||||
|
||||
err := ethash.MakeDAG(uint64(args.BlockNumber), "") |
||||
if err == nil { |
||||
return true, nil |
||||
} |
||||
return false, err |
||||
} |
||||
@ -1,142 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type StartMinerArgs struct { |
||||
Threads int |
||||
} |
||||
|
||||
func (args *StartMinerArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) == 0 || obj[0] == nil { |
||||
args.Threads = -1 |
||||
return nil |
||||
} |
||||
|
||||
var num *big.Int |
||||
if num, err = numString(obj[0]); err != nil { |
||||
return err |
||||
} |
||||
args.Threads = int(num.Int64()) |
||||
return nil |
||||
} |
||||
|
||||
type SetExtraArgs struct { |
||||
Data string |
||||
} |
||||
|
||||
func (args *SetExtraArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
extrastr, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("Price", "not a string") |
||||
} |
||||
args.Data = extrastr |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type GasPriceArgs struct { |
||||
Price string |
||||
} |
||||
|
||||
func (args *GasPriceArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if pricestr, ok := obj[0].(string); ok { |
||||
args.Price = pricestr |
||||
return nil |
||||
} |
||||
|
||||
return shared.NewInvalidTypeError("Price", "not a string") |
||||
} |
||||
|
||||
type SetEtherbaseArgs struct { |
||||
Etherbase common.Address |
||||
} |
||||
|
||||
func (args *SetEtherbaseArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if addr, ok := obj[0].(string); ok { |
||||
args.Etherbase = common.HexToAddress(addr) |
||||
if (args.Etherbase == common.Address{}) { |
||||
return shared.NewInvalidTypeError("Etherbase", "not a valid address") |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
return shared.NewInvalidTypeError("Etherbase", "not a string") |
||||
} |
||||
|
||||
type MakeDAGArgs struct { |
||||
BlockNumber int64 |
||||
} |
||||
|
||||
func (args *MakeDAGArgs) UnmarshalJSON(b []byte) (err error) { |
||||
args.BlockNumber = -1 |
||||
var obj []interface{} |
||||
|
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if err := blockHeight(obj[0], &args.BlockNumber); err != nil { |
||||
return err |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
@ -1,83 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Miner_JS = ` |
||||
web3._extend({ |
||||
property: 'miner', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'start', |
||||
call: 'miner_start', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'stop', |
||||
call: 'miner_stop', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setEtherbase', |
||||
call: 'miner_setEtherbase', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.formatters.formatInputInt], |
||||
outputFormatter: web3._extend.formatters.formatOutputBool |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setExtra', |
||||
call: 'miner_setExtra', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'setGasPrice', |
||||
call: 'miner_setGasPrice', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.utils.fromDecial] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'startAutoDAG', |
||||
call: 'miner_startAutoDAG', |
||||
params: 0, |
||||
inputFormatter: [] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'stopAutoDAG', |
||||
call: 'miner_stopAutoDAG', |
||||
params: 0, |
||||
inputFormatter: [] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'makeDAG', |
||||
call: 'miner_makeDAG', |
||||
params: 1, |
||||
inputFormatter: [web3._extend.formatters.inputDefaultBlockNumberFormatter] |
||||
}) |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'hashrate', |
||||
getter: 'miner_hashrate', |
||||
outputFormatter: web3._extend.utils.toDecimal |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,99 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
NetApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
netMapping = map[string]nethandler{ |
||||
"net_peerCount": (*netApi).PeerCount, |
||||
"net_listening": (*netApi).IsListening, |
||||
"net_version": (*netApi).Version, |
||||
} |
||||
) |
||||
|
||||
// net callback handler
|
||||
type nethandler func(*netApi, *shared.Request) (interface{}, error) |
||||
|
||||
// net api provider
|
||||
type netApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]nethandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new net api instance
|
||||
func NewNetApi(xeth *xeth.XEth, eth *eth.Ethereum, coder codec.Codec) *netApi { |
||||
return &netApi{ |
||||
xeth: xeth, |
||||
ethereum: eth, |
||||
methods: netMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *netApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *netApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *netApi) Name() string { |
||||
return shared.NetApiName |
||||
} |
||||
|
||||
func (self *netApi) ApiVersion() string { |
||||
return NetApiVersion |
||||
} |
||||
|
||||
// Number of connected peers
|
||||
func (self *netApi) PeerCount(req *shared.Request) (interface{}, error) { |
||||
return newHexNum(self.xeth.PeerCount()), nil |
||||
} |
||||
|
||||
func (self *netApi) IsListening(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.IsListening(), nil |
||||
} |
||||
|
||||
func (self *netApi) Version(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.NetworkVersion(), nil |
||||
} |
||||
@ -1,39 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Net_JS = ` |
||||
web3._extend({ |
||||
property: 'net', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'addPeer', |
||||
call: 'net_addPeer', |
||||
params: 1, |
||||
inputFormatter: [null] |
||||
}) |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'version', |
||||
getter: 'net_version' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,522 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/binary" |
||||
"encoding/hex" |
||||
"encoding/json" |
||||
"math/big" |
||||
"strings" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/core/types" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type hexdata struct { |
||||
data []byte |
||||
isNil bool |
||||
} |
||||
|
||||
func (d *hexdata) String() string { |
||||
return "0x" + common.Bytes2Hex(d.data) |
||||
} |
||||
|
||||
func (d *hexdata) MarshalJSON() ([]byte, error) { |
||||
if d.isNil { |
||||
return json.Marshal(nil) |
||||
} |
||||
return json.Marshal(d.String()) |
||||
} |
||||
|
||||
func newHexData(input interface{}) *hexdata { |
||||
d := new(hexdata) |
||||
|
||||
if input == nil { |
||||
d.isNil = true |
||||
return d |
||||
} |
||||
switch input := input.(type) { |
||||
case []byte: |
||||
d.data = input |
||||
case common.Hash: |
||||
d.data = input.Bytes() |
||||
case *common.Hash: |
||||
if input == nil { |
||||
d.isNil = true |
||||
} else { |
||||
d.data = input.Bytes() |
||||
} |
||||
case common.Address: |
||||
d.data = input.Bytes() |
||||
case *common.Address: |
||||
if input == nil { |
||||
d.isNil = true |
||||
} else { |
||||
d.data = input.Bytes() |
||||
} |
||||
case types.Bloom: |
||||
d.data = input.Bytes() |
||||
case *types.Bloom: |
||||
if input == nil { |
||||
d.isNil = true |
||||
} else { |
||||
d.data = input.Bytes() |
||||
} |
||||
case *big.Int: |
||||
if input == nil { |
||||
d.isNil = true |
||||
} else { |
||||
d.data = input.Bytes() |
||||
} |
||||
case int64: |
||||
d.data = big.NewInt(input).Bytes() |
||||
case uint64: |
||||
buff := make([]byte, 8) |
||||
binary.BigEndian.PutUint64(buff, input) |
||||
d.data = buff |
||||
case int: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case uint: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case int8: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case uint8: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case int16: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case uint16: |
||||
buff := make([]byte, 2) |
||||
binary.BigEndian.PutUint16(buff, input) |
||||
d.data = buff |
||||
case int32: |
||||
d.data = big.NewInt(int64(input)).Bytes() |
||||
case uint32: |
||||
buff := make([]byte, 4) |
||||
binary.BigEndian.PutUint32(buff, input) |
||||
d.data = buff |
||||
case string: // hexstring
|
||||
// aaargh ffs TODO: avoid back-and-forth hex encodings where unneeded
|
||||
bytes, err := hex.DecodeString(strings.TrimPrefix(input, "0x")) |
||||
if err != nil { |
||||
d.isNil = true |
||||
} else { |
||||
d.data = bytes |
||||
} |
||||
default: |
||||
d.isNil = true |
||||
} |
||||
|
||||
return d |
||||
} |
||||
|
||||
type hexnum struct { |
||||
data []byte |
||||
isNil bool |
||||
} |
||||
|
||||
func (d *hexnum) String() string { |
||||
// Get hex string from bytes
|
||||
out := common.Bytes2Hex(d.data) |
||||
// Trim leading 0s
|
||||
out = strings.TrimLeft(out, "0") |
||||
// Output "0x0" when value is 0
|
||||
if len(out) == 0 { |
||||
out = "0" |
||||
} |
||||
return "0x" + out |
||||
} |
||||
|
||||
func (d *hexnum) MarshalJSON() ([]byte, error) { |
||||
if d.isNil { |
||||
return json.Marshal(nil) |
||||
} |
||||
return json.Marshal(d.String()) |
||||
} |
||||
|
||||
func newHexNum(input interface{}) *hexnum { |
||||
d := new(hexnum) |
||||
|
||||
d.data = newHexData(input).data |
||||
|
||||
return d |
||||
} |
||||
|
||||
type BlockRes struct { |
||||
fullTx bool |
||||
|
||||
BlockNumber *hexnum `json:"number"` |
||||
BlockHash *hexdata `json:"hash"` |
||||
ParentHash *hexdata `json:"parentHash"` |
||||
Nonce *hexdata `json:"nonce"` |
||||
Sha3Uncles *hexdata `json:"sha3Uncles"` |
||||
LogsBloom *hexdata `json:"logsBloom"` |
||||
TransactionRoot *hexdata `json:"transactionsRoot"` |
||||
StateRoot *hexdata `json:"stateRoot"` |
||||
ReceiptRoot *hexdata `json:"receiptRoot"` |
||||
Miner *hexdata `json:"miner"` |
||||
Difficulty *hexnum `json:"difficulty"` |
||||
TotalDifficulty *hexnum `json:"totalDifficulty"` |
||||
Size *hexnum `json:"size"` |
||||
ExtraData *hexdata `json:"extraData"` |
||||
GasLimit *hexnum `json:"gasLimit"` |
||||
GasUsed *hexnum `json:"gasUsed"` |
||||
UnixTimestamp *hexnum `json:"timestamp"` |
||||
Transactions []*TransactionRes `json:"transactions"` |
||||
Uncles []*UncleRes `json:"uncles"` |
||||
} |
||||
|
||||
func (b *BlockRes) MarshalJSON() ([]byte, error) { |
||||
if b.fullTx { |
||||
var ext struct { |
||||
BlockNumber *hexnum `json:"number"` |
||||
BlockHash *hexdata `json:"hash"` |
||||
ParentHash *hexdata `json:"parentHash"` |
||||
Nonce *hexdata `json:"nonce"` |
||||
Sha3Uncles *hexdata `json:"sha3Uncles"` |
||||
LogsBloom *hexdata `json:"logsBloom"` |
||||
TransactionRoot *hexdata `json:"transactionsRoot"` |
||||
StateRoot *hexdata `json:"stateRoot"` |
||||
ReceiptRoot *hexdata `json:"receiptRoot"` |
||||
Miner *hexdata `json:"miner"` |
||||
Difficulty *hexnum `json:"difficulty"` |
||||
TotalDifficulty *hexnum `json:"totalDifficulty"` |
||||
Size *hexnum `json:"size"` |
||||
ExtraData *hexdata `json:"extraData"` |
||||
GasLimit *hexnum `json:"gasLimit"` |
||||
GasUsed *hexnum `json:"gasUsed"` |
||||
UnixTimestamp *hexnum `json:"timestamp"` |
||||
Transactions []*TransactionRes `json:"transactions"` |
||||
Uncles []*hexdata `json:"uncles"` |
||||
} |
||||
|
||||
ext.BlockNumber = b.BlockNumber |
||||
ext.BlockHash = b.BlockHash |
||||
ext.ParentHash = b.ParentHash |
||||
ext.Nonce = b.Nonce |
||||
ext.Sha3Uncles = b.Sha3Uncles |
||||
ext.LogsBloom = b.LogsBloom |
||||
ext.TransactionRoot = b.TransactionRoot |
||||
ext.StateRoot = b.StateRoot |
||||
ext.ReceiptRoot = b.ReceiptRoot |
||||
ext.Miner = b.Miner |
||||
ext.Difficulty = b.Difficulty |
||||
ext.TotalDifficulty = b.TotalDifficulty |
||||
ext.Size = b.Size |
||||
ext.ExtraData = b.ExtraData |
||||
ext.GasLimit = b.GasLimit |
||||
ext.GasUsed = b.GasUsed |
||||
ext.UnixTimestamp = b.UnixTimestamp |
||||
ext.Transactions = b.Transactions |
||||
ext.Uncles = make([]*hexdata, len(b.Uncles)) |
||||
for i, u := range b.Uncles { |
||||
ext.Uncles[i] = u.BlockHash |
||||
} |
||||
return json.Marshal(ext) |
||||
} else { |
||||
var ext struct { |
||||
BlockNumber *hexnum `json:"number"` |
||||
BlockHash *hexdata `json:"hash"` |
||||
ParentHash *hexdata `json:"parentHash"` |
||||
Nonce *hexdata `json:"nonce"` |
||||
Sha3Uncles *hexdata `json:"sha3Uncles"` |
||||
LogsBloom *hexdata `json:"logsBloom"` |
||||
TransactionRoot *hexdata `json:"transactionsRoot"` |
||||
StateRoot *hexdata `json:"stateRoot"` |
||||
ReceiptRoot *hexdata `json:"receiptRoot"` |
||||
Miner *hexdata `json:"miner"` |
||||
Difficulty *hexnum `json:"difficulty"` |
||||
TotalDifficulty *hexnum `json:"totalDifficulty"` |
||||
Size *hexnum `json:"size"` |
||||
ExtraData *hexdata `json:"extraData"` |
||||
GasLimit *hexnum `json:"gasLimit"` |
||||
GasUsed *hexnum `json:"gasUsed"` |
||||
UnixTimestamp *hexnum `json:"timestamp"` |
||||
Transactions []*hexdata `json:"transactions"` |
||||
Uncles []*hexdata `json:"uncles"` |
||||
} |
||||
|
||||
ext.BlockNumber = b.BlockNumber |
||||
ext.BlockHash = b.BlockHash |
||||
ext.ParentHash = b.ParentHash |
||||
ext.Nonce = b.Nonce |
||||
ext.Sha3Uncles = b.Sha3Uncles |
||||
ext.LogsBloom = b.LogsBloom |
||||
ext.TransactionRoot = b.TransactionRoot |
||||
ext.StateRoot = b.StateRoot |
||||
ext.ReceiptRoot = b.ReceiptRoot |
||||
ext.Miner = b.Miner |
||||
ext.Difficulty = b.Difficulty |
||||
ext.TotalDifficulty = b.TotalDifficulty |
||||
ext.Size = b.Size |
||||
ext.ExtraData = b.ExtraData |
||||
ext.GasLimit = b.GasLimit |
||||
ext.GasUsed = b.GasUsed |
||||
ext.UnixTimestamp = b.UnixTimestamp |
||||
ext.Transactions = make([]*hexdata, len(b.Transactions)) |
||||
for i, tx := range b.Transactions { |
||||
ext.Transactions[i] = tx.Hash |
||||
} |
||||
ext.Uncles = make([]*hexdata, len(b.Uncles)) |
||||
for i, u := range b.Uncles { |
||||
ext.Uncles[i] = u.BlockHash |
||||
} |
||||
return json.Marshal(ext) |
||||
} |
||||
} |
||||
|
||||
func NewBlockRes(block *types.Block, td *big.Int, fullTx bool) *BlockRes { |
||||
if block == nil { |
||||
return nil |
||||
} |
||||
|
||||
res := new(BlockRes) |
||||
res.fullTx = fullTx |
||||
res.BlockNumber = newHexNum(block.Number()) |
||||
res.BlockHash = newHexData(block.Hash()) |
||||
res.ParentHash = newHexData(block.ParentHash()) |
||||
res.Nonce = newHexData(block.Nonce()) |
||||
res.Sha3Uncles = newHexData(block.UncleHash()) |
||||
res.LogsBloom = newHexData(block.Bloom()) |
||||
res.TransactionRoot = newHexData(block.TxHash()) |
||||
res.StateRoot = newHexData(block.Root()) |
||||
res.ReceiptRoot = newHexData(block.ReceiptHash()) |
||||
res.Miner = newHexData(block.Coinbase()) |
||||
res.Difficulty = newHexNum(block.Difficulty()) |
||||
res.TotalDifficulty = newHexNum(td) |
||||
res.Size = newHexNum(block.Size().Int64()) |
||||
res.ExtraData = newHexData(block.Extra()) |
||||
res.GasLimit = newHexNum(block.GasLimit()) |
||||
res.GasUsed = newHexNum(block.GasUsed()) |
||||
res.UnixTimestamp = newHexNum(block.Time()) |
||||
|
||||
txs := block.Transactions() |
||||
res.Transactions = make([]*TransactionRes, len(txs)) |
||||
for i, tx := range txs { |
||||
res.Transactions[i] = NewTransactionRes(tx) |
||||
res.Transactions[i].BlockHash = res.BlockHash |
||||
res.Transactions[i].BlockNumber = res.BlockNumber |
||||
res.Transactions[i].TxIndex = newHexNum(i) |
||||
} |
||||
|
||||
uncles := block.Uncles() |
||||
res.Uncles = make([]*UncleRes, len(uncles)) |
||||
for i, uncle := range uncles { |
||||
res.Uncles[i] = NewUncleRes(uncle) |
||||
} |
||||
|
||||
return res |
||||
} |
||||
|
||||
type TransactionRes struct { |
||||
Hash *hexdata `json:"hash"` |
||||
Nonce *hexnum `json:"nonce"` |
||||
BlockHash *hexdata `json:"blockHash"` |
||||
BlockNumber *hexnum `json:"blockNumber"` |
||||
TxIndex *hexnum `json:"transactionIndex"` |
||||
From *hexdata `json:"from"` |
||||
To *hexdata `json:"to"` |
||||
Value *hexnum `json:"value"` |
||||
Gas *hexnum `json:"gas"` |
||||
GasPrice *hexnum `json:"gasPrice"` |
||||
Input *hexdata `json:"input"` |
||||
} |
||||
|
||||
func NewTransactionRes(tx *types.Transaction) *TransactionRes { |
||||
if tx == nil { |
||||
return nil |
||||
} |
||||
|
||||
var v = new(TransactionRes) |
||||
v.Hash = newHexData(tx.Hash()) |
||||
v.Nonce = newHexNum(tx.Nonce()) |
||||
// v.BlockHash =
|
||||
// v.BlockNumber =
|
||||
// v.TxIndex =
|
||||
from, _ := tx.From() |
||||
v.From = newHexData(from) |
||||
v.To = newHexData(tx.To()) |
||||
v.Value = newHexNum(tx.Value()) |
||||
v.Gas = newHexNum(tx.Gas()) |
||||
v.GasPrice = newHexNum(tx.GasPrice()) |
||||
v.Input = newHexData(tx.Data()) |
||||
return v |
||||
} |
||||
|
||||
type UncleRes struct { |
||||
BlockNumber *hexnum `json:"number"` |
||||
BlockHash *hexdata `json:"hash"` |
||||
ParentHash *hexdata `json:"parentHash"` |
||||
Nonce *hexdata `json:"nonce"` |
||||
Sha3Uncles *hexdata `json:"sha3Uncles"` |
||||
ReceiptHash *hexdata `json:"receiptHash"` |
||||
LogsBloom *hexdata `json:"logsBloom"` |
||||
TransactionRoot *hexdata `json:"transactionsRoot"` |
||||
StateRoot *hexdata `json:"stateRoot"` |
||||
Miner *hexdata `json:"miner"` |
||||
Difficulty *hexnum `json:"difficulty"` |
||||
ExtraData *hexdata `json:"extraData"` |
||||
GasLimit *hexnum `json:"gasLimit"` |
||||
GasUsed *hexnum `json:"gasUsed"` |
||||
UnixTimestamp *hexnum `json:"timestamp"` |
||||
} |
||||
|
||||
func NewUncleRes(h *types.Header) *UncleRes { |
||||
if h == nil { |
||||
return nil |
||||
} |
||||
|
||||
var v = new(UncleRes) |
||||
v.BlockNumber = newHexNum(h.Number) |
||||
v.BlockHash = newHexData(h.Hash()) |
||||
v.ParentHash = newHexData(h.ParentHash) |
||||
v.Sha3Uncles = newHexData(h.UncleHash) |
||||
v.Nonce = newHexData(h.Nonce[:]) |
||||
v.LogsBloom = newHexData(h.Bloom) |
||||
v.TransactionRoot = newHexData(h.TxHash) |
||||
v.StateRoot = newHexData(h.Root) |
||||
v.Miner = newHexData(h.Coinbase) |
||||
v.Difficulty = newHexNum(h.Difficulty) |
||||
v.ExtraData = newHexData(h.Extra) |
||||
v.GasLimit = newHexNum(h.GasLimit) |
||||
v.GasUsed = newHexNum(h.GasUsed) |
||||
v.UnixTimestamp = newHexNum(h.Time) |
||||
v.ReceiptHash = newHexData(h.ReceiptHash) |
||||
|
||||
return v |
||||
} |
||||
|
||||
// type FilterLogRes struct {
|
||||
// Hash string `json:"hash"`
|
||||
// Address string `json:"address"`
|
||||
// Data string `json:"data"`
|
||||
// BlockNumber string `json:"blockNumber"`
|
||||
// TransactionHash string `json:"transactionHash"`
|
||||
// BlockHash string `json:"blockHash"`
|
||||
// TransactionIndex string `json:"transactionIndex"`
|
||||
// LogIndex string `json:"logIndex"`
|
||||
// }
|
||||
|
||||
// type FilterWhisperRes struct {
|
||||
// Hash string `json:"hash"`
|
||||
// From string `json:"from"`
|
||||
// To string `json:"to"`
|
||||
// Expiry string `json:"expiry"`
|
||||
// Sent string `json:"sent"`
|
||||
// Ttl string `json:"ttl"`
|
||||
// Topics string `json:"topics"`
|
||||
// Payload string `json:"payload"`
|
||||
// WorkProved string `json:"workProved"`
|
||||
// }
|
||||
|
||||
type ReceiptRes struct { |
||||
TransactionHash *hexdata `json:"transactionHash"` |
||||
TransactionIndex *hexnum `json:"transactionIndex"` |
||||
BlockNumber *hexnum `json:"blockNumber"` |
||||
BlockHash *hexdata `json:"blockHash"` |
||||
CumulativeGasUsed *hexnum `json:"cumulativeGasUsed"` |
||||
GasUsed *hexnum `json:"gasUsed"` |
||||
ContractAddress *hexdata `json:"contractAddress"` |
||||
Logs *[]interface{} `json:"logs"` |
||||
} |
||||
|
||||
func NewReceiptRes(rec *types.Receipt) *ReceiptRes { |
||||
if rec == nil { |
||||
return nil |
||||
} |
||||
|
||||
var v = new(ReceiptRes) |
||||
v.TransactionHash = newHexData(rec.TxHash) |
||||
if rec.GasUsed != nil { |
||||
v.GasUsed = newHexNum(rec.GasUsed.Bytes()) |
||||
} |
||||
v.CumulativeGasUsed = newHexNum(rec.CumulativeGasUsed) |
||||
|
||||
// If the ContractAddress is 20 0x0 bytes, assume it is not a contract creation
|
||||
if bytes.Compare(rec.ContractAddress.Bytes(), bytes.Repeat([]byte{0}, 20)) != 0 { |
||||
v.ContractAddress = newHexData(rec.ContractAddress) |
||||
} |
||||
|
||||
logs := make([]interface{}, len(rec.Logs)) |
||||
for i, log := range rec.Logs { |
||||
logs[i] = NewLogRes(log) |
||||
} |
||||
v.Logs = &logs |
||||
|
||||
return v |
||||
} |
||||
|
||||
func numString(raw interface{}) (*big.Int, error) { |
||||
var number *big.Int |
||||
// Parse as integer
|
||||
num, ok := raw.(float64) |
||||
if ok { |
||||
number = big.NewInt(int64(num)) |
||||
return number, nil |
||||
} |
||||
|
||||
// Parse as string/hexstring
|
||||
str, ok := raw.(string) |
||||
if ok { |
||||
number = common.String2Big(str) |
||||
return number, nil |
||||
} |
||||
|
||||
return nil, shared.NewInvalidTypeError("", "not a number or string") |
||||
} |
||||
|
||||
func blockHeight(raw interface{}, number *int64) error { |
||||
// Parse as integer
|
||||
num, ok := raw.(float64) |
||||
if ok { |
||||
*number = int64(num) |
||||
return nil |
||||
} |
||||
|
||||
// Parse as string/hexstring
|
||||
str, ok := raw.(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("", "not a number or string") |
||||
} |
||||
|
||||
switch str { |
||||
case "earliest": |
||||
*number = 0 |
||||
case "latest": |
||||
*number = -1 |
||||
case "pending": |
||||
*number = -2 |
||||
default: |
||||
if common.HasHexPrefix(str) { |
||||
*number = common.String2Big(str).Int64() |
||||
} else { |
||||
return shared.NewInvalidTypeError("blockNumber", "is not a valid string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func blockHeightFromJson(msg json.RawMessage, number *int64) error { |
||||
var raw interface{} |
||||
if err := json.Unmarshal(msg, &raw); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
return blockHeight(raw, number) |
||||
} |
||||
@ -1,139 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"fmt" |
||||
"time" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
PersonalApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
personalMapping = map[string]personalhandler{ |
||||
"personal_listAccounts": (*personalApi).ListAccounts, |
||||
"personal_newAccount": (*personalApi).NewAccount, |
||||
"personal_unlockAccount": (*personalApi).UnlockAccount, |
||||
} |
||||
) |
||||
|
||||
// net callback handler
|
||||
type personalhandler func(*personalApi, *shared.Request) (interface{}, error) |
||||
|
||||
// net api provider
|
||||
type personalApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]personalhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new net api instance
|
||||
func NewPersonalApi(xeth *xeth.XEth, eth *eth.Ethereum, coder codec.Codec) *personalApi { |
||||
return &personalApi{ |
||||
xeth: xeth, |
||||
ethereum: eth, |
||||
methods: personalMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *personalApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *personalApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *personalApi) Name() string { |
||||
return shared.PersonalApiName |
||||
} |
||||
|
||||
func (self *personalApi) ApiVersion() string { |
||||
return PersonalApiVersion |
||||
} |
||||
|
||||
func (self *personalApi) ListAccounts(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.Accounts(), nil |
||||
} |
||||
|
||||
func (self *personalApi) NewAccount(req *shared.Request) (interface{}, error) { |
||||
args := new(NewAccountArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
var passwd string |
||||
if args.Passphrase == nil { |
||||
fe := self.xeth.Frontend() |
||||
if fe == nil { |
||||
return false, fmt.Errorf("unable to create account: unable to interact with user") |
||||
} |
||||
var ok bool |
||||
passwd, ok = fe.AskPassword() |
||||
if !ok { |
||||
return false, fmt.Errorf("unable to create account: no password given") |
||||
} |
||||
} else { |
||||
passwd = *args.Passphrase |
||||
} |
||||
am := self.ethereum.AccountManager() |
||||
acc, err := am.NewAccount(passwd) |
||||
return acc.Address.Hex(), err |
||||
} |
||||
|
||||
func (self *personalApi) UnlockAccount(req *shared.Request) (interface{}, error) { |
||||
args := new(UnlockAccountArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if args.Passphrase == nil { |
||||
fe := self.xeth.Frontend() |
||||
if fe == nil { |
||||
return false, fmt.Errorf("No password provided") |
||||
} |
||||
return fe.UnlockAccount(common.HexToAddress(args.Address).Bytes()), nil |
||||
} |
||||
|
||||
am := self.ethereum.AccountManager() |
||||
addr := common.HexToAddress(args.Address) |
||||
|
||||
err := am.TimedUnlock(addr, *args.Passphrase, time.Duration(args.Duration)*time.Second) |
||||
return err == nil, err |
||||
} |
||||
@ -1,85 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type NewAccountArgs struct { |
||||
Passphrase *string |
||||
} |
||||
|
||||
func (args *NewAccountArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) >= 1 && obj[0] != nil { |
||||
if passphrasestr, ok := obj[0].(string); ok { |
||||
args.Passphrase = &passphrasestr |
||||
} else { |
||||
return shared.NewInvalidTypeError("passphrase", "not a string") |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type UnlockAccountArgs struct { |
||||
Address string |
||||
Passphrase *string |
||||
Duration int |
||||
} |
||||
|
||||
func (args *UnlockAccountArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
args.Duration = 0 |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
if addrstr, ok := obj[0].(string); ok { |
||||
args.Address = addrstr |
||||
} else { |
||||
return shared.NewInvalidTypeError("address", "not a string") |
||||
} |
||||
|
||||
if len(obj) >= 2 && obj[1] != nil { |
||||
if passphrasestr, ok := obj[1].(string); ok { |
||||
args.Passphrase = &passphrasestr |
||||
} else { |
||||
return shared.NewInvalidTypeError("passphrase", "not a string") |
||||
} |
||||
} |
||||
|
||||
if len(obj) >= 3 && obj[2] != nil { |
||||
if duration, ok := obj[2].(float64); ok { |
||||
args.Duration = int(duration) |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
@ -1,51 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Personal_JS = ` |
||||
web3._extend({ |
||||
property: 'personal', |
||||
methods: |
||||
[ |
||||
new web3._extend.Method({ |
||||
name: 'newAccount', |
||||
call: 'personal_newAccount', |
||||
params: 1, |
||||
inputFormatter: [null], |
||||
outputFormatter: web3._extend.utils.toAddress |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'unlockAccount', |
||||
call: 'personal_unlockAccount', |
||||
params: 3, |
||||
inputFormatter: [null, null, null] |
||||
}), |
||||
new web3._extend.Method({ |
||||
name: 'lockAccount', |
||||
call: 'personal_lockAccount', |
||||
params: 1 |
||||
}) |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'listAccounts', |
||||
getter: 'personal_listAccounts' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,196 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
ShhApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
shhMapping = map[string]shhhandler{ |
||||
"shh_version": (*shhApi).Version, |
||||
"shh_post": (*shhApi).Post, |
||||
"shh_hasIdentity": (*shhApi).HasIdentity, |
||||
"shh_newIdentity": (*shhApi).NewIdentity, |
||||
"shh_newFilter": (*shhApi).NewFilter, |
||||
"shh_uninstallFilter": (*shhApi).UninstallFilter, |
||||
"shh_getMessages": (*shhApi).GetMessages, |
||||
"shh_getFilterChanges": (*shhApi).GetFilterChanges, |
||||
} |
||||
) |
||||
|
||||
func newWhisperOfflineError(method string) error { |
||||
return shared.NewNotAvailableError(method, "whisper offline") |
||||
} |
||||
|
||||
// net callback handler
|
||||
type shhhandler func(*shhApi, *shared.Request) (interface{}, error) |
||||
|
||||
// shh api provider
|
||||
type shhApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]shhhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new whisper api instance
|
||||
func NewShhApi(xeth *xeth.XEth, eth *eth.Ethereum, coder codec.Codec) *shhApi { |
||||
return &shhApi{ |
||||
xeth: xeth, |
||||
ethereum: eth, |
||||
methods: shhMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *shhApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *shhApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *shhApi) Name() string { |
||||
return shared.ShhApiName |
||||
} |
||||
|
||||
func (self *shhApi) ApiVersion() string { |
||||
return ShhApiVersion |
||||
} |
||||
|
||||
func (self *shhApi) Version(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
return w.Version(), nil |
||||
} |
||||
|
||||
func (self *shhApi) Post(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
args := new(WhisperMessageArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
err := w.Post(args.Payload, args.To, args.From, args.Topics, args.Priority, args.Ttl) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
return true, nil |
||||
} |
||||
|
||||
func (self *shhApi) HasIdentity(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
args := new(WhisperIdentityArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return w.HasIdentity(args.Identity), nil |
||||
} |
||||
|
||||
func (self *shhApi) NewIdentity(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
return w.NewIdentity(), nil |
||||
} |
||||
|
||||
func (self *shhApi) NewFilter(req *shared.Request) (interface{}, error) { |
||||
args := new(WhisperFilterArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
id := self.xeth.NewWhisperFilter(args.To, args.From, args.Topics) |
||||
return newHexNum(big.NewInt(int64(id)).Bytes()), nil |
||||
} |
||||
|
||||
func (self *shhApi) UninstallFilter(req *shared.Request) (interface{}, error) { |
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
return self.xeth.UninstallWhisperFilter(args.Id), nil |
||||
} |
||||
|
||||
func (self *shhApi) GetFilterChanges(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
// Retrieve all the new messages arrived since the last request
|
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return self.xeth.WhisperMessagesChanged(args.Id), nil |
||||
} |
||||
|
||||
func (self *shhApi) GetMessages(req *shared.Request) (interface{}, error) { |
||||
w := self.xeth.Whisper() |
||||
if w == nil { |
||||
return nil, newWhisperOfflineError(req.Method) |
||||
} |
||||
|
||||
// Retrieve all the cached messages matching a specific, existing filter
|
||||
args := new(FilterIdArgs) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return self.xeth.WhisperMessages(args.Id), nil |
||||
} |
||||
@ -1,174 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
"math/big" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type WhisperMessageArgs struct { |
||||
Payload string |
||||
To string |
||||
From string |
||||
Topics []string |
||||
Priority uint32 |
||||
Ttl uint32 |
||||
} |
||||
|
||||
func (args *WhisperMessageArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []struct { |
||||
Payload string |
||||
To string |
||||
From string |
||||
Topics []string |
||||
Priority interface{} |
||||
Ttl interface{} |
||||
} |
||||
|
||||
if err = json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
args.Payload = obj[0].Payload |
||||
args.To = obj[0].To |
||||
args.From = obj[0].From |
||||
args.Topics = obj[0].Topics |
||||
|
||||
var num *big.Int |
||||
if num, err = numString(obj[0].Priority); err != nil { |
||||
return err |
||||
} |
||||
args.Priority = uint32(num.Int64()) |
||||
|
||||
if num, err = numString(obj[0].Ttl); err != nil { |
||||
return err |
||||
} |
||||
args.Ttl = uint32(num.Int64()) |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type WhisperIdentityArgs struct { |
||||
Identity string |
||||
} |
||||
|
||||
func (args *WhisperIdentityArgs) UnmarshalJSON(b []byte) (err error) { |
||||
var obj []interface{} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
|
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
|
||||
argstr, ok := obj[0].(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("arg0", "not a string") |
||||
} |
||||
|
||||
args.Identity = argstr |
||||
|
||||
return nil |
||||
} |
||||
|
||||
type WhisperFilterArgs struct { |
||||
To string |
||||
From string |
||||
Topics [][]string |
||||
} |
||||
|
||||
// UnmarshalJSON implements the json.Unmarshaler interface, invoked to convert a
|
||||
// JSON message blob into a WhisperFilterArgs structure.
|
||||
func (args *WhisperFilterArgs) UnmarshalJSON(b []byte) (err error) { |
||||
// Unmarshal the JSON message and sanity check
|
||||
var obj []struct { |
||||
To interface{} `json:"to"` |
||||
From interface{} `json:"from"` |
||||
Topics interface{} `json:"topics"` |
||||
} |
||||
if err := json.Unmarshal(b, &obj); err != nil { |
||||
return shared.NewDecodeParamError(err.Error()) |
||||
} |
||||
if len(obj) < 1 { |
||||
return shared.NewInsufficientParamsError(len(obj), 1) |
||||
} |
||||
// Retrieve the simple data contents of the filter arguments
|
||||
if obj[0].To == nil { |
||||
args.To = "" |
||||
} else { |
||||
argstr, ok := obj[0].To.(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("to", "is not a string") |
||||
} |
||||
args.To = argstr |
||||
} |
||||
if obj[0].From == nil { |
||||
args.From = "" |
||||
} else { |
||||
argstr, ok := obj[0].From.(string) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("from", "is not a string") |
||||
} |
||||
args.From = argstr |
||||
} |
||||
// Construct the nested topic array
|
||||
if obj[0].Topics != nil { |
||||
// Make sure we have an actual topic array
|
||||
list, ok := obj[0].Topics.([]interface{}) |
||||
if !ok { |
||||
return shared.NewInvalidTypeError("topics", "is not an array") |
||||
} |
||||
// Iterate over each topic and handle nil, string or array
|
||||
topics := make([][]string, len(list)) |
||||
for idx, field := range list { |
||||
switch value := field.(type) { |
||||
case nil: |
||||
topics[idx] = []string{} |
||||
|
||||
case string: |
||||
topics[idx] = []string{value} |
||||
|
||||
case []interface{}: |
||||
topics[idx] = make([]string, len(value)) |
||||
for i, nested := range value { |
||||
switch value := nested.(type) { |
||||
case nil: |
||||
topics[idx][i] = "" |
||||
|
||||
case string: |
||||
topics[idx][i] = value |
||||
|
||||
default: |
||||
return shared.NewInvalidTypeError(fmt.Sprintf("topic[%d][%d]", idx, i), "is not a string") |
||||
} |
||||
} |
||||
default: |
||||
return shared.NewInvalidTypeError(fmt.Sprintf("topic[%d]", idx), "not a string or array") |
||||
} |
||||
} |
||||
args.Topics = topics |
||||
} |
||||
return nil |
||||
} |
||||
@ -1,34 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const Shh_JS = ` |
||||
web3._extend({ |
||||
property: 'shh', |
||||
methods: |
||||
[ |
||||
|
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'version', |
||||
getter: 'shh_version' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,92 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
TxPoolApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
txpoolMapping = map[string]txpoolhandler{ |
||||
"txpool_status": (*txPoolApi).Status, |
||||
} |
||||
) |
||||
|
||||
// net callback handler
|
||||
type txpoolhandler func(*txPoolApi, *shared.Request) (interface{}, error) |
||||
|
||||
// txpool api provider
|
||||
type txPoolApi struct { |
||||
xeth *xeth.XEth |
||||
ethereum *eth.Ethereum |
||||
methods map[string]txpoolhandler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new txpool api instance
|
||||
func NewTxPoolApi(xeth *xeth.XEth, eth *eth.Ethereum, coder codec.Codec) *txPoolApi { |
||||
return &txPoolApi{ |
||||
xeth: xeth, |
||||
ethereum: eth, |
||||
methods: txpoolMapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *txPoolApi) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *txPoolApi) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, shared.NewNotImplementedError(req.Method) |
||||
} |
||||
|
||||
func (self *txPoolApi) Name() string { |
||||
return shared.TxPoolApiName |
||||
} |
||||
|
||||
func (self *txPoolApi) ApiVersion() string { |
||||
return TxPoolApiVersion |
||||
} |
||||
|
||||
func (self *txPoolApi) Status(req *shared.Request) (interface{}, error) { |
||||
pending, queue := self.ethereum.TxPool().Stats() |
||||
return map[string]int{ |
||||
"pending": pending, |
||||
"queued": queue, |
||||
}, nil |
||||
} |
||||
@ -1,33 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
const TxPool_JS = ` |
||||
web3._extend({ |
||||
property: 'txpool', |
||||
methods: |
||||
[ |
||||
], |
||||
properties: |
||||
[ |
||||
new web3._extend.Property({ |
||||
name: 'status', |
||||
getter: 'txpool_status' |
||||
}) |
||||
] |
||||
}); |
||||
` |
||||
@ -1,226 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"strings" |
||||
|
||||
"fmt" |
||||
|
||||
"github.com/ethereum/go-ethereum/eth" |
||||
"github.com/ethereum/go-ethereum/node" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
var ( |
||||
// Mapping between the different methods each api supports
|
||||
AutoCompletion = map[string][]string{ |
||||
"admin": []string{ |
||||
"addPeer", |
||||
"datadir", |
||||
"enableUserAgent", |
||||
"exportChain", |
||||
"getContractInfo", |
||||
"httpGet", |
||||
"importChain", |
||||
"nodeInfo", |
||||
"peers", |
||||
"register", |
||||
"registerUrl", |
||||
"saveInfo", |
||||
"setGlobalRegistrar", |
||||
"setHashReg", |
||||
"setUrlHint", |
||||
"setSolc", |
||||
"sleep", |
||||
"sleepBlocks", |
||||
"startNatSpec", |
||||
"startRPC", |
||||
"stopNatSpec", |
||||
"stopRPC", |
||||
"verbosity", |
||||
}, |
||||
"db": []string{ |
||||
"getString", |
||||
"putString", |
||||
"getHex", |
||||
"putHex", |
||||
}, |
||||
"debug": []string{ |
||||
"dumpBlock", |
||||
"getBlockRlp", |
||||
"metrics", |
||||
"printBlock", |
||||
"processBlock", |
||||
"seedHash", |
||||
"setHead", |
||||
}, |
||||
"eth": []string{ |
||||
"accounts", |
||||
"blockNumber", |
||||
"call", |
||||
"contract", |
||||
"coinbase", |
||||
"compile.lll", |
||||
"compile.serpent", |
||||
"compile.solidity", |
||||
"contract", |
||||
"defaultAccount", |
||||
"defaultBlock", |
||||
"estimateGas", |
||||
"filter", |
||||
"getBalance", |
||||
"getBlock", |
||||
"getBlockTransactionCount", |
||||
"getBlockUncleCount", |
||||
"getCode", |
||||
"getNatSpec", |
||||
"getCompilers", |
||||
"gasPrice", |
||||
"getStorageAt", |
||||
"getTransaction", |
||||
"getTransactionCount", |
||||
"getTransactionFromBlock", |
||||
"getTransactionReceipt", |
||||
"getUncle", |
||||
"hashrate", |
||||
"mining", |
||||
"namereg", |
||||
"pendingTransactions", |
||||
"resend", |
||||
"sendRawTransaction", |
||||
"sendTransaction", |
||||
"sign", |
||||
"syncing", |
||||
}, |
||||
"miner": []string{ |
||||
"hashrate", |
||||
"makeDAG", |
||||
"setEtherbase", |
||||
"setExtra", |
||||
"setGasPrice", |
||||
"startAutoDAG", |
||||
"start", |
||||
"stopAutoDAG", |
||||
"stop", |
||||
}, |
||||
"net": []string{ |
||||
"peerCount", |
||||
"listening", |
||||
}, |
||||
"personal": []string{ |
||||
"listAccounts", |
||||
"newAccount", |
||||
"unlockAccount", |
||||
}, |
||||
"shh": []string{ |
||||
"post", |
||||
"newIdentity", |
||||
"hasIdentity", |
||||
"newGroup", |
||||
"addToGroup", |
||||
"filter", |
||||
}, |
||||
"txpool": []string{ |
||||
"status", |
||||
}, |
||||
"web3": []string{ |
||||
"sha3", |
||||
"version", |
||||
"fromWei", |
||||
"toWei", |
||||
"toHex", |
||||
"toAscii", |
||||
"fromAscii", |
||||
"toBigNumber", |
||||
"isAddress", |
||||
}, |
||||
} |
||||
) |
||||
|
||||
// Parse a comma separated API string to individual api's
|
||||
func ParseApiString(apistr string, codec codec.Codec, xeth *xeth.XEth, stack *node.Node) ([]shared.EthereumApi, error) { |
||||
if len(strings.TrimSpace(apistr)) == 0 { |
||||
return nil, fmt.Errorf("Empty apistr provided") |
||||
} |
||||
|
||||
names := strings.Split(apistr, ",") |
||||
apis := make([]shared.EthereumApi, len(names)) |
||||
|
||||
var eth *eth.Ethereum |
||||
if stack != nil { |
||||
if err := stack.Service(ð); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
for i, name := range names { |
||||
switch strings.ToLower(strings.TrimSpace(name)) { |
||||
case shared.AdminApiName: |
||||
apis[i] = NewAdminApi(xeth, stack, codec) |
||||
case shared.DebugApiName: |
||||
apis[i] = NewDebugApi(xeth, eth, codec) |
||||
case shared.DbApiName: |
||||
apis[i] = NewDbApi(xeth, eth, codec) |
||||
case shared.EthApiName: |
||||
apis[i] = NewEthApi(xeth, eth, codec) |
||||
case shared.MinerApiName: |
||||
apis[i] = NewMinerApi(eth, codec) |
||||
case shared.NetApiName: |
||||
apis[i] = NewNetApi(xeth, eth, codec) |
||||
case shared.ShhApiName: |
||||
apis[i] = NewShhApi(xeth, eth, codec) |
||||
case shared.TxPoolApiName: |
||||
apis[i] = NewTxPoolApi(xeth, eth, codec) |
||||
case shared.PersonalApiName: |
||||
apis[i] = NewPersonalApi(xeth, eth, codec) |
||||
case shared.Web3ApiName: |
||||
apis[i] = NewWeb3Api(xeth, codec) |
||||
case "rpc": // gives information about the RPC interface
|
||||
continue |
||||
default: |
||||
return nil, fmt.Errorf("Unknown API '%s'", name) |
||||
} |
||||
} |
||||
return apis, nil |
||||
} |
||||
|
||||
func Javascript(name string) string { |
||||
switch strings.ToLower(strings.TrimSpace(name)) { |
||||
case shared.AdminApiName: |
||||
return Admin_JS |
||||
case shared.DebugApiName: |
||||
return Debug_JS |
||||
case shared.DbApiName: |
||||
return Db_JS |
||||
case shared.EthApiName: |
||||
return Eth_JS |
||||
case shared.MinerApiName: |
||||
return Miner_JS |
||||
case shared.NetApiName: |
||||
return Net_JS |
||||
case shared.ShhApiName: |
||||
return Shh_JS |
||||
case shared.TxPoolApiName: |
||||
return TxPool_JS |
||||
case shared.PersonalApiName: |
||||
return Personal_JS |
||||
} |
||||
|
||||
return "" |
||||
} |
||||
@ -1,99 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package api |
||||
|
||||
import ( |
||||
"github.com/ethereum/go-ethereum/common" |
||||
"github.com/ethereum/go-ethereum/crypto" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/ethereum/go-ethereum/xeth" |
||||
) |
||||
|
||||
const ( |
||||
Web3ApiVersion = "1.0" |
||||
) |
||||
|
||||
var ( |
||||
// mapping between methods and handlers
|
||||
Web3Mapping = map[string]web3handler{ |
||||
"web3_sha3": (*web3Api).Sha3, |
||||
"web3_clientVersion": (*web3Api).ClientVersion, |
||||
} |
||||
) |
||||
|
||||
// web3 callback handler
|
||||
type web3handler func(*web3Api, *shared.Request) (interface{}, error) |
||||
|
||||
// web3 api provider
|
||||
type web3Api struct { |
||||
xeth *xeth.XEth |
||||
methods map[string]web3handler |
||||
codec codec.ApiCoder |
||||
} |
||||
|
||||
// create a new web3 api instance
|
||||
func NewWeb3Api(xeth *xeth.XEth, coder codec.Codec) *web3Api { |
||||
return &web3Api{ |
||||
xeth: xeth, |
||||
methods: Web3Mapping, |
||||
codec: coder.New(nil), |
||||
} |
||||
} |
||||
|
||||
// collection with supported methods
|
||||
func (self *web3Api) Methods() []string { |
||||
methods := make([]string, len(self.methods)) |
||||
i := 0 |
||||
for k := range self.methods { |
||||
methods[i] = k |
||||
i++ |
||||
} |
||||
return methods |
||||
} |
||||
|
||||
// Execute given request
|
||||
func (self *web3Api) Execute(req *shared.Request) (interface{}, error) { |
||||
if callback, ok := self.methods[req.Method]; ok { |
||||
return callback(self, req) |
||||
} |
||||
|
||||
return nil, &shared.NotImplementedError{req.Method} |
||||
} |
||||
|
||||
func (self *web3Api) Name() string { |
||||
return shared.Web3ApiName |
||||
} |
||||
|
||||
func (self *web3Api) ApiVersion() string { |
||||
return Web3ApiVersion |
||||
} |
||||
|
||||
// Calculates the sha3 over req.Params.Data
|
||||
func (self *web3Api) Sha3(req *shared.Request) (interface{}, error) { |
||||
args := new(Sha3Args) |
||||
if err := self.codec.Decode(req.Params, &args); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return common.ToHex(crypto.Sha3(common.FromHex(args.Data))), nil |
||||
} |
||||
|
||||
// returns the xeth client vrsion
|
||||
func (self *web3Api) ClientVersion(req *shared.Request) (interface{}, error) { |
||||
return self.xeth.ClientVersion(), nil |
||||
} |
||||
@ -1,65 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package codec |
||||
|
||||
import ( |
||||
"net" |
||||
"strconv" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type Codec int |
||||
|
||||
// (de)serialization support for rpc interface
|
||||
type ApiCoder interface { |
||||
// Parse message to request from underlying stream
|
||||
ReadRequest() ([]*shared.Request, bool, error) |
||||
// Parse response message from underlying stream
|
||||
ReadResponse() (interface{}, error) |
||||
// Read raw message from underlying stream
|
||||
Recv() (interface{}, error) |
||||
// Encode response to encoded form in underlying stream
|
||||
WriteResponse(interface{}) error |
||||
// Decode single message from data
|
||||
Decode([]byte, interface{}) error |
||||
// Encode msg to encoded form
|
||||
Encode(msg interface{}) ([]byte, error) |
||||
// close the underlying stream
|
||||
Close() |
||||
} |
||||
|
||||
// supported codecs
|
||||
const ( |
||||
JSON Codec = iota |
||||
nCodecs |
||||
) |
||||
|
||||
var ( |
||||
// collection with supported coders
|
||||
coders = make([]func(net.Conn) ApiCoder, nCodecs) |
||||
) |
||||
|
||||
// create a new coder instance
|
||||
func (c Codec) New(conn net.Conn) ApiCoder { |
||||
switch c { |
||||
case JSON: |
||||
return NewJsonCoder(conn) |
||||
} |
||||
|
||||
panic("codec: request for codec #" + strconv.Itoa(int(c)) + " is unavailable") |
||||
} |
||||
@ -1,149 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package codec |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
"net" |
||||
"time" |
||||
"strings" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
const ( |
||||
READ_TIMEOUT = 60 // in seconds
|
||||
MAX_REQUEST_SIZE = 1024 * 1024 |
||||
MAX_RESPONSE_SIZE = 1024 * 1024 |
||||
) |
||||
|
||||
// Json serialization support
|
||||
type JsonCodec struct { |
||||
c net.Conn |
||||
d *json.Decoder |
||||
} |
||||
|
||||
// Create new JSON coder instance
|
||||
func NewJsonCoder(conn net.Conn) ApiCoder { |
||||
return &JsonCodec{ |
||||
c: conn, |
||||
d: json.NewDecoder(conn), |
||||
} |
||||
} |
||||
|
||||
// Read incoming request and parse it to RPC request
|
||||
func (self *JsonCodec) ReadRequest() (requests []*shared.Request, isBatch bool, err error) { |
||||
deadline := time.Now().Add(READ_TIMEOUT * time.Second) |
||||
if err := self.c.SetDeadline(deadline); err != nil { |
||||
return nil, false, err |
||||
} |
||||
|
||||
var incoming json.RawMessage |
||||
err = self.d.Decode(&incoming) |
||||
if err == nil { |
||||
isBatch = incoming[0] == '[' |
||||
if isBatch { |
||||
requests = make([]*shared.Request, 0) |
||||
err = json.Unmarshal(incoming, &requests) |
||||
} else { |
||||
requests = make([]*shared.Request, 1) |
||||
var singleRequest shared.Request |
||||
if err = json.Unmarshal(incoming, &singleRequest); err == nil { |
||||
requests[0] = &singleRequest |
||||
} |
||||
} |
||||
return |
||||
} |
||||
|
||||
self.c.Close() |
||||
return nil, false, err |
||||
} |
||||
|
||||
func (self *JsonCodec) Recv() (interface{}, error) { |
||||
var msg json.RawMessage |
||||
err := self.d.Decode(&msg) |
||||
if err != nil { |
||||
self.c.Close() |
||||
return nil, err |
||||
} |
||||
|
||||
return msg, err |
||||
} |
||||
|
||||
func (self *JsonCodec) ReadResponse() (interface{}, error) { |
||||
in, err := self.Recv() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if msg, ok := in.(json.RawMessage); ok { |
||||
var req *shared.Request |
||||
if err = json.Unmarshal(msg, &req); err == nil && strings.HasPrefix(req.Method, "agent_") { |
||||
return req, nil |
||||
} |
||||
|
||||
var failure *shared.ErrorResponse |
||||
if err = json.Unmarshal(msg, &failure); err == nil && failure.Error != nil { |
||||
return failure, fmt.Errorf(failure.Error.Message) |
||||
} |
||||
|
||||
var success *shared.SuccessResponse |
||||
if err = json.Unmarshal(msg, &success); err == nil { |
||||
return success, nil |
||||
} |
||||
} |
||||
|
||||
return in, err |
||||
} |
||||
|
||||
// Decode data
|
||||
func (self *JsonCodec) Decode(data []byte, msg interface{}) error { |
||||
return json.Unmarshal(data, msg) |
||||
} |
||||
|
||||
// Encode message
|
||||
func (self *JsonCodec) Encode(msg interface{}) ([]byte, error) { |
||||
return json.Marshal(msg) |
||||
} |
||||
|
||||
// Parse JSON data from conn to obj
|
||||
func (self *JsonCodec) WriteResponse(res interface{}) error { |
||||
data, err := json.Marshal(res) |
||||
if err != nil { |
||||
self.c.Close() |
||||
return err |
||||
} |
||||
|
||||
bytesWritten := 0 |
||||
|
||||
for bytesWritten < len(data) { |
||||
n, err := self.c.Write(data[bytesWritten:]) |
||||
if err != nil { |
||||
self.c.Close() |
||||
return err |
||||
} |
||||
bytesWritten += n |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// Close decoder and encoder
|
||||
func (self *JsonCodec) Close() { |
||||
self.c.Close() |
||||
} |
||||
@ -1,157 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package codec |
||||
|
||||
import ( |
||||
"bytes" |
||||
"io" |
||||
"net" |
||||
"testing" |
||||
"time" |
||||
) |
||||
|
||||
type jsonTestConn struct { |
||||
buffer *bytes.Buffer |
||||
} |
||||
|
||||
func newJsonTestConn(data []byte) *jsonTestConn { |
||||
return &jsonTestConn{ |
||||
buffer: bytes.NewBuffer(data), |
||||
} |
||||
} |
||||
|
||||
func (self *jsonTestConn) Read(p []byte) (n int, err error) { |
||||
return self.buffer.Read(p) |
||||
} |
||||
|
||||
func (self *jsonTestConn) Write(p []byte) (n int, err error) { |
||||
return self.buffer.Write(p) |
||||
} |
||||
|
||||
func (self *jsonTestConn) Close() error { |
||||
// not implemented
|
||||
return nil |
||||
} |
||||
|
||||
func (self *jsonTestConn) LocalAddr() net.Addr { |
||||
// not implemented
|
||||
return nil |
||||
} |
||||
|
||||
func (self *jsonTestConn) RemoteAddr() net.Addr { |
||||
// not implemented
|
||||
return nil |
||||
} |
||||
|
||||
func (self *jsonTestConn) SetDeadline(t time.Time) error { |
||||
return nil |
||||
} |
||||
|
||||
func (self *jsonTestConn) SetReadDeadline(t time.Time) error { |
||||
return nil |
||||
} |
||||
|
||||
func (self *jsonTestConn) SetWriteDeadline(t time.Time) error { |
||||
return nil |
||||
} |
||||
|
||||
func TestJsonDecoderWithValidRequest(t *testing.T) { |
||||
reqdata := []byte(`{"jsonrpc":"2.0","method":"modules","params":[],"id":64}`) |
||||
decoder := newJsonTestConn(reqdata) |
||||
|
||||
jsonDecoder := NewJsonCoder(decoder) |
||||
requests, batch, err := jsonDecoder.ReadRequest() |
||||
|
||||
if err != nil { |
||||
t.Errorf("Read valid request failed - %v", err) |
||||
} |
||||
|
||||
if len(requests) != 1 { |
||||
t.Errorf("Expected to get a single request but got %d", len(requests)) |
||||
} |
||||
|
||||
if batch { |
||||
t.Errorf("Got batch indication while expecting single request") |
||||
} |
||||
|
||||
if requests[0].Id != float64(64) { |
||||
t.Errorf("Expected req.Id == 64 but got %v", requests[0].Id) |
||||
} |
||||
|
||||
if requests[0].Method != "modules" { |
||||
t.Errorf("Expected req.Method == 'modules' got '%s'", requests[0].Method) |
||||
} |
||||
} |
||||
|
||||
func TestJsonDecoderWithValidBatchRequest(t *testing.T) { |
||||
reqdata := []byte(`[{"jsonrpc":"2.0","method":"modules","params":[],"id":64}, |
||||
{"jsonrpc":"2.0","method":"modules","params":[],"id":64}]`) |
||||
decoder := newJsonTestConn(reqdata) |
||||
|
||||
jsonDecoder := NewJsonCoder(decoder) |
||||
requests, batch, err := jsonDecoder.ReadRequest() |
||||
|
||||
if err != nil { |
||||
t.Errorf("Read valid batch request failed - %v", err) |
||||
} |
||||
|
||||
if len(requests) != 2 { |
||||
t.Errorf("Expected to get two requests but got %d", len(requests)) |
||||
} |
||||
|
||||
if !batch { |
||||
t.Errorf("Got no batch indication while expecting batch request") |
||||
} |
||||
|
||||
for i := 0; i < len(requests); i++ { |
||||
if requests[i].Id != float64(64) { |
||||
t.Errorf("Expected req.Id == 64 but got %v", requests[i].Id) |
||||
} |
||||
|
||||
if requests[i].Method != "modules" { |
||||
t.Errorf("Expected req.Method == 'modules' got '%s'", requests[i].Method) |
||||
} |
||||
} |
||||
} |
||||
|
||||
func TestJsonDecoderWithInvalidIncompleteMessage(t *testing.T) { |
||||
reqdata := []byte(`{"jsonrpc":"2.0","method":"modules","pa`) |
||||
decoder := newJsonTestConn(reqdata) |
||||
|
||||
jsonDecoder := NewJsonCoder(decoder) |
||||
requests, batch, err := jsonDecoder.ReadRequest() |
||||
|
||||
if err != io.ErrUnexpectedEOF { |
||||
t.Errorf("Expected to read an incomplete request err but got %v", err) |
||||
} |
||||
|
||||
// remaining message
|
||||
decoder.Write([]byte(`rams":[],"id:64"}`)) |
||||
requests, batch, err = jsonDecoder.ReadRequest() |
||||
|
||||
if err == nil { |
||||
t.Errorf("Expected an error but got nil") |
||||
} |
||||
|
||||
if len(requests) != 0 { |
||||
t.Errorf("Expected to get no requests but got %d", len(requests)) |
||||
} |
||||
|
||||
if batch { |
||||
t.Errorf("Got batch indication while expecting non batch") |
||||
} |
||||
} |
||||
@ -1,150 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package comms |
||||
|
||||
import ( |
||||
"io" |
||||
"net" |
||||
|
||||
"fmt" |
||||
"strings" |
||||
|
||||
"strconv" |
||||
|
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
const ( |
||||
maxHttpSizeReqLength = 1024 * 1024 // 1MB
|
||||
) |
||||
|
||||
var ( |
||||
// List with all API's which are offered over the in proc interface by default
|
||||
DefaultInProcApis = shared.AllApis |
||||
|
||||
// List with all API's which are offered over the IPC interface by default
|
||||
DefaultIpcApis = shared.AllApis |
||||
|
||||
// List with API's which are offered over thr HTTP/RPC interface by default
|
||||
DefaultHttpRpcApis = strings.Join([]string{ |
||||
shared.DbApiName, shared.EthApiName, shared.NetApiName, shared.Web3ApiName, |
||||
}, ",") |
||||
) |
||||
|
||||
type EthereumClient interface { |
||||
// Close underlying connection
|
||||
Close() |
||||
// Send request
|
||||
Send(interface{}) error |
||||
// Receive response
|
||||
Recv() (interface{}, error) |
||||
// List with modules this client supports
|
||||
SupportedModules() (map[string]string, error) |
||||
} |
||||
|
||||
func handle(id int, conn net.Conn, api shared.EthereumApi, c codec.Codec) { |
||||
codec := c.New(conn) |
||||
|
||||
defer func() { |
||||
if r := recover(); r != nil { |
||||
glog.Errorf("panic: %v\n", r) |
||||
} |
||||
codec.Close() |
||||
}() |
||||
|
||||
for { |
||||
requests, isBatch, err := codec.ReadRequest() |
||||
if err == io.EOF { |
||||
return |
||||
} else if err != nil { |
||||
glog.V(logger.Debug).Infof("Closed IPC Conn %06d recv err - %v\n", id, err) |
||||
return |
||||
} |
||||
|
||||
if isBatch { |
||||
responses := make([]*interface{}, len(requests)) |
||||
responseCount := 0 |
||||
for _, req := range requests { |
||||
res, err := api.Execute(req) |
||||
if req.Id != nil { |
||||
rpcResponse := shared.NewRpcResponse(req.Id, req.Jsonrpc, res, err) |
||||
responses[responseCount] = rpcResponse |
||||
responseCount += 1 |
||||
} |
||||
} |
||||
|
||||
err = codec.WriteResponse(responses[:responseCount]) |
||||
if err != nil { |
||||
glog.V(logger.Debug).Infof("Closed IPC Conn %06d send err - %v\n", id, err) |
||||
return |
||||
} |
||||
} else { |
||||
var rpcResponse interface{} |
||||
res, err := api.Execute(requests[0]) |
||||
|
||||
rpcResponse = shared.NewRpcResponse(requests[0].Id, requests[0].Jsonrpc, res, err) |
||||
err = codec.WriteResponse(rpcResponse) |
||||
if err != nil { |
||||
glog.V(logger.Debug).Infof("Closed IPC Conn %06d send err - %v\n", id, err) |
||||
return |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Endpoint must be in the form of:
|
||||
// ${protocol}:${path}
|
||||
// e.g. ipc:/tmp/geth.ipc
|
||||
// rpc:localhost:8545
|
||||
func ClientFromEndpoint(endpoint string, c codec.Codec) (EthereumClient, error) { |
||||
if strings.HasPrefix(endpoint, "ipc:") { |
||||
cfg := IpcConfig{ |
||||
Endpoint: endpoint[4:], |
||||
} |
||||
return NewIpcClient(cfg, codec.JSON) |
||||
} |
||||
|
||||
if strings.HasPrefix(endpoint, "rpc:") { |
||||
parts := strings.Split(endpoint, ":") |
||||
addr := "http://localhost" |
||||
port := uint(8545) |
||||
if len(parts) >= 3 { |
||||
addr = parts[1] + ":" + parts[2] |
||||
} |
||||
|
||||
if len(parts) >= 4 { |
||||
p, err := strconv.Atoi(parts[3]) |
||||
|
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
port = uint(p) |
||||
} |
||||
|
||||
cfg := HttpConfig{ |
||||
ListenAddress: addr, |
||||
ListenPort: port, |
||||
} |
||||
|
||||
return NewHttpClient(cfg, codec.JSON), nil |
||||
} |
||||
|
||||
return nil, fmt.Errorf("Invalid endpoint") |
||||
} |
||||
@ -1,345 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package comms |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
"net" |
||||
"net/http" |
||||
"strings" |
||||
"sync" |
||||
"time" |
||||
|
||||
"bytes" |
||||
"io" |
||||
"io/ioutil" |
||||
|
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
"github.com/rs/cors" |
||||
) |
||||
|
||||
const ( |
||||
serverIdleTimeout = 10 * time.Second // idle keep-alive connections
|
||||
serverReadTimeout = 15 * time.Second // per-request read timeout
|
||||
serverWriteTimeout = 15 * time.Second // per-request read timeout
|
||||
) |
||||
|
||||
var ( |
||||
httpServerMu sync.Mutex |
||||
httpServer *stopServer |
||||
) |
||||
|
||||
type HttpConfig struct { |
||||
ListenAddress string |
||||
ListenPort uint |
||||
CorsDomain string |
||||
} |
||||
|
||||
// stopServer augments http.Server with idle connection tracking.
|
||||
// Idle keep-alive connections are shut down when Close is called.
|
||||
type stopServer struct { |
||||
*http.Server |
||||
l net.Listener |
||||
// connection tracking state
|
||||
mu sync.Mutex |
||||
shutdown bool // true when Stop has returned
|
||||
idle map[net.Conn]struct{} |
||||
} |
||||
|
||||
type handler struct { |
||||
codec codec.Codec |
||||
api shared.EthereumApi |
||||
} |
||||
|
||||
// StartHTTP starts listening for RPC requests sent via HTTP.
|
||||
func StartHttp(cfg HttpConfig, codec codec.Codec, api shared.EthereumApi) error { |
||||
httpServerMu.Lock() |
||||
defer httpServerMu.Unlock() |
||||
|
||||
addr := fmt.Sprintf("%s:%d", cfg.ListenAddress, cfg.ListenPort) |
||||
if httpServer != nil { |
||||
if addr != httpServer.Addr { |
||||
return fmt.Errorf("RPC service already running on %s ", httpServer.Addr) |
||||
} |
||||
return nil // RPC service already running on given host/port
|
||||
} |
||||
// Set up the request handler, wrapping it with CORS headers if configured.
|
||||
handler := http.Handler(&handler{codec, api}) |
||||
if len(cfg.CorsDomain) > 0 { |
||||
opts := cors.Options{ |
||||
AllowedMethods: []string{"POST"}, |
||||
AllowedOrigins: strings.Split(cfg.CorsDomain, " "), |
||||
} |
||||
handler = cors.New(opts).Handler(handler) |
||||
} |
||||
// Start the server.
|
||||
s, err := listenHTTP(addr, handler) |
||||
if err != nil { |
||||
glog.V(logger.Error).Infof("Can't listen on %s:%d: %v", cfg.ListenAddress, cfg.ListenPort, err) |
||||
return err |
||||
} |
||||
httpServer = s |
||||
return nil |
||||
} |
||||
|
||||
func (h *handler) ServeHTTP(w http.ResponseWriter, req *http.Request) { |
||||
w.Header().Set("Content-Type", "application/json") |
||||
|
||||
// Limit request size to resist DoS
|
||||
if req.ContentLength > maxHttpSizeReqLength { |
||||
err := fmt.Errorf("Request too large") |
||||
response := shared.NewRpcErrorResponse(-1, shared.JsonRpcVersion, -32700, err) |
||||
sendJSON(w, &response) |
||||
return |
||||
} |
||||
|
||||
defer req.Body.Close() |
||||
payload, err := ioutil.ReadAll(req.Body) |
||||
if err != nil { |
||||
err := fmt.Errorf("Could not read request body") |
||||
response := shared.NewRpcErrorResponse(-1, shared.JsonRpcVersion, -32700, err) |
||||
sendJSON(w, &response) |
||||
return |
||||
} |
||||
|
||||
c := h.codec.New(nil) |
||||
var rpcReq shared.Request |
||||
if err = c.Decode(payload, &rpcReq); err == nil { |
||||
reply, err := h.api.Execute(&rpcReq) |
||||
res := shared.NewRpcResponse(rpcReq.Id, rpcReq.Jsonrpc, reply, err) |
||||
sendJSON(w, &res) |
||||
return |
||||
} |
||||
|
||||
var reqBatch []shared.Request |
||||
if err = c.Decode(payload, &reqBatch); err == nil { |
||||
resBatch := make([]*interface{}, len(reqBatch)) |
||||
resCount := 0 |
||||
for i, rpcReq := range reqBatch { |
||||
reply, err := h.api.Execute(&rpcReq) |
||||
if rpcReq.Id != nil { // this leaves nil entries in the response batch for later removal
|
||||
resBatch[i] = shared.NewRpcResponse(rpcReq.Id, rpcReq.Jsonrpc, reply, err) |
||||
resCount += 1 |
||||
} |
||||
} |
||||
// make response omitting nil entries
|
||||
sendJSON(w, resBatch[:resCount]) |
||||
return |
||||
} |
||||
|
||||
// invalid request
|
||||
err = fmt.Errorf("Could not decode request") |
||||
res := shared.NewRpcErrorResponse(-1, shared.JsonRpcVersion, -32600, err) |
||||
sendJSON(w, res) |
||||
} |
||||
|
||||
func sendJSON(w io.Writer, v interface{}) { |
||||
if glog.V(logger.Detail) { |
||||
if payload, err := json.MarshalIndent(v, "", "\t"); err == nil { |
||||
glog.Infof("Sending payload: %s", payload) |
||||
} |
||||
} |
||||
if err := json.NewEncoder(w).Encode(v); err != nil { |
||||
glog.V(logger.Error).Infoln("Error sending JSON:", err) |
||||
} |
||||
} |
||||
|
||||
// Stop closes all active HTTP connections and shuts down the server.
|
||||
func StopHttp() { |
||||
httpServerMu.Lock() |
||||
defer httpServerMu.Unlock() |
||||
if httpServer != nil { |
||||
httpServer.Close() |
||||
httpServer = nil |
||||
} |
||||
} |
||||
|
||||
func listenHTTP(addr string, h http.Handler) (*stopServer, error) { |
||||
l, err := net.Listen("tcp", addr) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
s := &stopServer{l: l, idle: make(map[net.Conn]struct{})} |
||||
s.Server = &http.Server{ |
||||
Addr: addr, |
||||
Handler: h, |
||||
ReadTimeout: serverReadTimeout, |
||||
WriteTimeout: serverWriteTimeout, |
||||
ConnState: s.connState, |
||||
} |
||||
go s.Serve(l) |
||||
return s, nil |
||||
} |
||||
|
||||
func (s *stopServer) connState(c net.Conn, state http.ConnState) { |
||||
s.mu.Lock() |
||||
defer s.mu.Unlock() |
||||
// Close c immediately if we're past shutdown.
|
||||
if s.shutdown { |
||||
if state != http.StateClosed { |
||||
c.Close() |
||||
} |
||||
return |
||||
} |
||||
if state == http.StateIdle { |
||||
s.idle[c] = struct{}{} |
||||
} else { |
||||
delete(s.idle, c) |
||||
} |
||||
} |
||||
|
||||
func (s *stopServer) Close() { |
||||
s.mu.Lock() |
||||
defer s.mu.Unlock() |
||||
// Shut down the acceptor. No new connections can be created.
|
||||
s.l.Close() |
||||
// Drop all idle connections. Non-idle connections will be
|
||||
// closed by connState as soon as they become idle.
|
||||
s.shutdown = true |
||||
for c := range s.idle { |
||||
glog.V(logger.Detail).Infof("closing idle connection %v", c.RemoteAddr()) |
||||
c.Close() |
||||
delete(s.idle, c) |
||||
} |
||||
} |
||||
|
||||
type httpClient struct { |
||||
address string |
||||
port uint |
||||
codec codec.ApiCoder |
||||
lastRes interface{} |
||||
lastErr error |
||||
} |
||||
|
||||
// Create a new in process client
|
||||
func NewHttpClient(cfg HttpConfig, c codec.Codec) *httpClient { |
||||
return &httpClient{ |
||||
address: cfg.ListenAddress, |
||||
port: cfg.ListenPort, |
||||
codec: c.New(nil), |
||||
} |
||||
} |
||||
|
||||
func (self *httpClient) Close() { |
||||
// do nothing
|
||||
} |
||||
|
||||
func (self *httpClient) Send(req interface{}) error { |
||||
var body []byte |
||||
var err error |
||||
|
||||
self.lastRes = nil |
||||
self.lastErr = nil |
||||
|
||||
if body, err = self.codec.Encode(req); err != nil { |
||||
return err |
||||
} |
||||
|
||||
httpReq, err := http.NewRequest("POST", fmt.Sprintf("%s:%d", self.address, self.port), bytes.NewBuffer(body)) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
httpReq.Header.Set("Content-Type", "application/json") |
||||
|
||||
client := http.Client{} |
||||
resp, err := client.Do(httpReq) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
defer resp.Body.Close() |
||||
|
||||
if resp.Status == "200 OK" { |
||||
reply, _ := ioutil.ReadAll(resp.Body) |
||||
var rpcSuccessResponse shared.SuccessResponse |
||||
if err = self.codec.Decode(reply, &rpcSuccessResponse); err == nil { |
||||
self.lastRes = &rpcSuccessResponse |
||||
self.lastErr = err |
||||
return nil |
||||
} else { |
||||
var rpcErrorResponse shared.ErrorResponse |
||||
if err = self.codec.Decode(reply, &rpcErrorResponse); err == nil { |
||||
self.lastRes = &rpcErrorResponse |
||||
self.lastErr = err |
||||
return nil |
||||
} else { |
||||
return err |
||||
} |
||||
} |
||||
} |
||||
|
||||
return fmt.Errorf("Not implemented") |
||||
} |
||||
|
||||
func (self *httpClient) Recv() (interface{}, error) { |
||||
return self.lastRes, self.lastErr |
||||
} |
||||
|
||||
func (self *httpClient) SupportedModules() (map[string]string, error) { |
||||
var body []byte |
||||
var err error |
||||
|
||||
payload := shared.Request{ |
||||
Id: 1, |
||||
Jsonrpc: "2.0", |
||||
Method: "modules", |
||||
} |
||||
|
||||
if body, err = self.codec.Encode(payload); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
req, err := http.NewRequest("POST", fmt.Sprintf("%s:%d", self.address, self.port), bytes.NewBuffer(body)) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
req.Header.Set("Content-Type", "application/json") |
||||
|
||||
client := http.Client{} |
||||
resp, err := client.Do(req) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
defer resp.Body.Close() |
||||
|
||||
if resp.Status == "200 OK" { |
||||
reply, _ := ioutil.ReadAll(resp.Body) |
||||
var rpcRes shared.SuccessResponse |
||||
if err = self.codec.Decode(reply, &rpcRes); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
result := make(map[string]string) |
||||
if modules, ok := rpcRes.Result.(map[string]interface{}); ok { |
||||
for a, v := range modules { |
||||
result[a] = fmt.Sprintf("%s", v) |
||||
} |
||||
return result, nil |
||||
} |
||||
err = fmt.Errorf("Unable to parse module response - %v", rpcRes.Result) |
||||
} else { |
||||
fmt.Printf("resp.Status = %s\n", resp.Status) |
||||
fmt.Printf("err = %v\n", err) |
||||
} |
||||
|
||||
return nil, err |
||||
} |
||||
@ -1,82 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package comms |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type InProcClient struct { |
||||
api shared.EthereumApi |
||||
codec codec.Codec |
||||
lastId interface{} |
||||
lastJsonrpc string |
||||
lastErr error |
||||
lastRes interface{} |
||||
} |
||||
|
||||
// Create a new in process client
|
||||
func NewInProcClient(codec codec.Codec) *InProcClient { |
||||
return &InProcClient{ |
||||
codec: codec, |
||||
} |
||||
} |
||||
|
||||
func (self *InProcClient) Close() { |
||||
// do nothing
|
||||
} |
||||
|
||||
// Need to setup api support
|
||||
func (self *InProcClient) Initialize(offeredApi shared.EthereumApi) { |
||||
self.api = offeredApi |
||||
} |
||||
|
||||
func (self *InProcClient) Send(req interface{}) error { |
||||
if r, ok := req.(*shared.Request); ok { |
||||
self.lastId = r.Id |
||||
self.lastJsonrpc = r.Jsonrpc |
||||
self.lastRes, self.lastErr = self.api.Execute(r) |
||||
return self.lastErr |
||||
} |
||||
|
||||
return fmt.Errorf("Invalid request (%T)", req) |
||||
} |
||||
|
||||
func (self *InProcClient) Recv() (interface{}, error) { |
||||
return *shared.NewRpcResponse(self.lastId, self.lastJsonrpc, self.lastRes, self.lastErr), nil |
||||
} |
||||
|
||||
func (self *InProcClient) SupportedModules() (map[string]string, error) { |
||||
req := shared.Request{ |
||||
Id: 1, |
||||
Jsonrpc: "2.0", |
||||
Method: "modules", |
||||
} |
||||
|
||||
if res, err := self.api.Execute(&req); err == nil { |
||||
if result, ok := res.(map[string]string); ok { |
||||
return result, nil |
||||
} |
||||
} else { |
||||
return nil, err |
||||
} |
||||
|
||||
return nil, fmt.Errorf("Invalid response") |
||||
} |
||||
@ -1,158 +0,0 @@ |
||||
// Copyright 2015 The go-ethereum Authors
|
||||
// This file is part of the go-ethereum library.
|
||||
//
|
||||
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU Lesser General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public License
|
||||
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package comms |
||||
|
||||
import ( |
||||
"fmt" |
||||
"math/rand" |
||||
"net" |
||||
"os" |
||||
|
||||
"encoding/json" |
||||
|
||||
"github.com/ethereum/go-ethereum/logger" |
||||
"github.com/ethereum/go-ethereum/logger/glog" |
||||
"github.com/ethereum/go-ethereum/rpc/codec" |
||||
"github.com/ethereum/go-ethereum/rpc/shared" |
||||
) |
||||
|
||||
type Stopper interface { |
||||
Stop() |
||||
} |
||||
|
||||
type InitFunc func(conn net.Conn) (Stopper, shared.EthereumApi, error) |
||||
|
||||
type IpcConfig struct { |
||||
Endpoint string |
||||
} |
||||
|
||||
type ipcClient struct { |
||||
endpoint string |
||||
c net.Conn |
||||
codec codec.Codec |
||||
coder codec.ApiCoder |
||||
} |
||||
|
||||
func (self *ipcClient) Close() { |
||||
self.coder.Close() |
||||
} |
||||
|
||||
func (self *ipcClient) Send(msg interface{}) error { |
||||
var err error |
||||
if err = self.coder.WriteResponse(msg); err != nil { |
||||
if err = self.reconnect(); err == nil { |
||||
err = self.coder.WriteResponse(msg) |
||||
} |
||||
} |
||||
return err |
||||
} |
||||
|
||||
func (self *ipcClient) Recv() (interface{}, error) { |
||||
return self.coder.ReadResponse() |
||||
} |
||||
|
||||
func (self *ipcClient) SupportedModules() (map[string]string, error) { |
||||
req := shared.Request{ |
||||
Id: 1, |
||||
Jsonrpc: "2.0", |
||||
Method: "rpc_modules", |
||||
} |
||||
|
||||
if err := self.coder.WriteResponse(req); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
res, _ := self.coder.ReadResponse() |
||||
if sucRes, ok := res.(*shared.SuccessResponse); ok { |
||||
data, _ := json.Marshal(sucRes.Result) |
||||
modules := make(map[string]string) |
||||
if err := json.Unmarshal(data, &modules); err == nil { |
||||
return modules, nil |
||||
} |
||||
} |
||||
|
||||
// old version uses modules instead of rpc_modules, this can be removed after full migration
|
||||
req.Method = "modules" |
||||
if err := self.coder.WriteResponse(req); err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
res, err := self.coder.ReadResponse() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if sucRes, ok := res.(*shared.SuccessResponse); ok { |
||||
data, _ := json.Marshal(sucRes.Result) |
||||
modules := make(map[string]string) |
||||
err = json.Unmarshal(data, &modules) |
||||
if err == nil { |
||||
return modules, nil |
||||
} |
||||
} |
||||
|
||||
return nil, fmt.Errorf("Invalid response") |
||||
} |
||||
|
||||
// Create a new IPC client, UNIX domain socket on posix, named pipe on Windows
|
||||
func NewIpcClient(cfg IpcConfig, codec codec.Codec) (*ipcClient, error) { |
||||
return newIpcClient(cfg, codec) |
||||
} |
||||
|
||||
// Start IPC server
|
||||
func StartIpc(cfg IpcConfig, codec codec.Codec, initializer InitFunc) error { |
||||
l, err := ipcListen(cfg) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
go ipcLoop(cfg, codec, initializer, l) |
||||
return nil |
||||
} |
||||
|
||||
// CreateListener creates an listener, on Unix platforms this is a unix socket, on Windows this is a named pipe
|
||||
func CreateListener(cfg IpcConfig) (net.Listener, error) { |
||||
return ipcListen(cfg) |
||||
} |
||||
|
||||
func ipcLoop(cfg IpcConfig, codec codec.Codec, initializer InitFunc, l net.Listener) { |
||||
glog.V(logger.Info).Infof("IPC service started (%s)\n", cfg.Endpoint) |
||||
defer os.Remove(cfg.Endpoint) |
||||
defer l.Close() |
||||
for { |
||||
conn, err := l.Accept() |
||||
if err != nil { |
||||
glog.V(logger.Debug).Infof("accept: %v", err) |
||||
return |
||||
} |
||||
id := newIpcConnId() |
||||
go func() { |
||||
defer conn.Close() |
||||
glog.V(logger.Debug).Infof("new connection with id %06d started", id) |
||||
stopper, api, err := initializer(conn) |
||||
if err != nil { |
||||
glog.V(logger.Error).Infof("Unable to initialize IPC connection: %v", err) |
||||
return |
||||
} |
||||
defer stopper.Stop() |
||||
handle(id, conn, api, codec) |
||||
}() |
||||
} |
||||
} |
||||
|
||||
func newIpcConnId() int { |
||||
return rand.Int() % 1000000 |
||||
} |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue