1
0
mirror of https://github.com/aclindsa/ofxgo.git synced 2024-11-25 04:20:05 -05:00

Added support for ofx.discovercard.com

Discover requires an exact set of headers in exact order, or it returns 403.
This commit is contained in:
Dmitry Kakurin 2018-09-21 12:57:34 -07:00
parent c6a806399a
commit 9c9a0d47ad

View File

@ -1,9 +1,15 @@
package ofxgo
import (
"bufio"
"bytes"
"crypto/tls"
"errors"
"fmt"
"io"
"net/http"
"net/url"
"strconv"
"strings"
)
@ -75,7 +81,13 @@ func RawRequest(URL string, r io.Reader) (*http.Response, error) {
return nil, errors.New("Refusing to send OFX request with possible plain-text password over non-https protocol")
}
response, err := http.Post(URL, "application/x-ofx", r)
var response *http.Response
var err error
if strings.HasPrefix(URL, "https://ofx.discovercard.com") {
response, err = customHTTPRequest(URL, r)
} else {
response, err = http.Post(URL, "application/x-ofx", r)
}
if err != nil {
return nil, err
}
@ -87,6 +99,53 @@ func RawRequest(URL string, r io.Reader) (*http.Response, error) {
return response, nil
}
func customHTTPRequest(urlStr string, r io.Reader) (*http.Response, error) {
buf, ok := r.(*bytes.Buffer)
if !ok {
return nil, errors.New("Need to know request size")
}
url, err := url.Parse(urlStr)
if err != nil {
return nil, err
}
path := url.Path
if path == "" {
path = "/"
}
// Discover requires only these headers and in this exact order, or it returns 403
reqText := "POST " + path + " HTTP/1.1\r\n" +
"Content-Type: application/x-ofx\r\n" +
"Host: " + url.Hostname() + "\r\n" +
"Content-Length: " + strconv.Itoa(buf.Len()) + "\r\n" +
"Connection: Keep-Alive\r\n" +
"\r\n"
host := url.Host
if url.Port() == "" {
host += ":443"
}
conn, err := tls.Dial("tcp", host, nil)
if err != nil {
return nil, err
}
// BUGBUG: cannot do defer conn.Close() until body is read,
// we are "leaking" a socket here, but it will be finalized
// Send request and headers
fmt.Fprint(conn, reqText)
// Send body
_, err = io.Copy(conn, r)
if err != nil {
return nil, err
}
return http.ReadResponse(bufio.NewReader(conn), nil)
}
// RawRequestCookies is RawRequest with the added feature of sending cookies
func RawRequestCookies(URL string, r io.Reader, cookies []*http.Cookie) (*http.Response, error) {
if !strings.HasPrefix(URL, "https://") {
@ -136,7 +195,10 @@ func (c *Client) RequestNoParse(r *Request) (*http.Response, error) {
// Fortunately, the initial response contains the cookie we need, so if we
// detect an empty response with cookies set that didn't have any errors,
// re-try the request while sending their cookies back to them.
if err == nil && response.ContentLength <= 0 && len(response.Cookies()) > 0 {
if err == nil &&
response.ContentLength <= 0 &&
!strings.HasPrefix(r.URL, "https://ofx.discovercard.com") &&
len(response.Cookies()) > 0 {
b, err = r.Marshal()
if err != nil {
return nil, err