AcraReader http buffering

cmd/acra-connector/acra-connector.go

@@ -195,7 +195,7 @@ func main() {
 
 	if *acraServerHost == "" && *acraServerConnectionString == "" {
 		log.WithField(logging.FieldKeyEventCode, logging.EventCodeErrorWrongConfiguration).
-			Errorln("Configuration error: you must pass acraserver_connection_host or acra_connection_string parameter")
+			Errorln("Configuration error: you must pass acraserver_connection_host or acraserver_connection_string parameter")
 		os.Exit(1)
 	}
 	if *acraServerHost != "" {
@@ -240,11 +240,6 @@ func main() {
 		os.Exit(1)
 	}
 
-	if *verbose {
-		logging.SetLogLevel(logging.LOG_VERBOSE)
-	} else {
-		logging.SetLogLevel(logging.LOG_DISCARD)
-	}
 	if runtime.GOOS != "linux" {
 		*disableUserCheck = true
 	}
@@ -267,7 +262,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	log.Debugf("Start listening connections")
+	log.Infof("Ready: start listening connections")
 	config := &Config{KeyStore: keyStore, KeysDir: *keysDir, ClientId: []byte(*clientId), AcraServerConnectionString: *acraServerConnectionString, ConnectionString: *connectionString, AcraServerId: []byte(*acraServerId), disableUserCheck: *disableUserCheck}
 	listener, err := network.Listen(*connectionString)
 	if err != nil {
@@ -346,6 +341,14 @@ func main() {
 	}
 
 	log.Infof("Start listening connection %s", *connectionString)
+
+	if *verbose {
+		logging.SetLogLevel(logging.LOG_VERBOSE)
+	} else {
+		log.Infof("Disabling future logs.. Set -v to see logs")
+		logging.SetLogLevel(logging.LOG_DISCARD)
+	}
+
 	for {
 		connection, err := listener.Accept()
 		if err != nil {

cmd/acra-reader/http-api/decryptor.go

@@ -2,19 +2,18 @@ package http_api
 
 import (
 	"bytes"
-	"encoding/binary"
-	"fmt"
-	"github.com/cossacklabs/acra/decryptor/base"
-	"github.com/cossacklabs/acra/keystore"
 	"github.com/cossacklabs/acra/logging"
 	"github.com/cossacklabs/acra/utils"
 	"github.com/cossacklabs/themis/gothemis/keys"
 	log "github.com/sirupsen/logrus"
-	"io/ioutil"
-	"net"
+	"fmt"
 	"net/http"
-	"os"
 	"strings"
+	"os"
+	"github.com/cossacklabs/acra/keystore"
+	"net"
+	"io/ioutil"
+	"github.com/cossacklabs/acra/decryptor/base"
 )
 
 type HTTPConnectionsDecryptor struct {
@@ -25,26 +24,24 @@ func NewHTTPConnectionsDecryptor(keystorage keystore.KeyStore) (*HTTPConnections
 	return &HTTPConnectionsDecryptor{keystorage: keystorage}, nil
 }
 
-func (decryptor *HTTPConnectionsDecryptor) SendResponseAndCloseConnection(logger *log.Entry, response *http.Response, connection net.Conn) {
-	buf := new(bytes.Buffer)
-	err := binary.Write(buf, binary.BigEndian, response)
+func (decryptor *HTTPConnectionsDecryptor) SendResponse(logger *log.Entry, response *http.Response, connection net.Conn) {
+	r, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		logger.WithError(err).WithField(logging.FieldKeyEventCode, logging.EventCodeErrorReaderCantReturnResponse).
+			Warningln("Can't read response body")
+	}
+	err = response.Body.Close()
 
 	if err != nil {
 		logger.WithError(err).WithField(logging.FieldKeyEventCode, logging.EventCodeErrorReaderCantReturnResponse).
 			Warningln("Can't convert response to binary")
 	} else {
-		_, err = connection.Write(buf.Bytes())
+		_, err = connection.Write(r)
 		if err != nil {
 			logger.WithError(err).WithField(logging.FieldKeyEventCode, logging.EventCodeErrorReaderCantReturnResponse).
 				Warningln("Can't write response to HTTP request")
 		}
 	}
-
-	err = connection.Close()
-	if err != nil {
-		logger.WithError(err).WithField(logging.FieldKeyEventCode, logging.EventCodeErrorReaderCantCloseConnection).
-			Warningln("Can't close connection of HTTP request")
-	}
 }
 
 func (decryptor *HTTPConnectionsDecryptor) ParseRequestPrepareResponse(logger *log.Entry, request *http.Request, clientId []byte) *http.Response {
@@ -110,7 +107,7 @@ func (decryptor *HTTPConnectionsDecryptor) ParseRequestPrepareResponse(logger *l
 			return responseWithMessage(request, http.StatusBadRequest, msg)
 		}
 
-		decryptedStruct, err := decryptor.decryptAcraStruct(acraStruct, zoneId, clientId)
+		decryptedStruct, err := decryptor.decryptAcraStruct(logger, acraStruct, zoneId, clientId)
 
 		if err != nil {
 			msg := fmt.Sprintf("Can't decrypt AcraStruct")
@@ -122,7 +119,7 @@ func (decryptor *HTTPConnectionsDecryptor) ParseRequestPrepareResponse(logger *l
 
 		response := emptyResponseWithStatus(request, http.StatusOK)
 		response.Header.Set("Content-Type", "application/octet-stream")
-		response.Body = ioutil.NopCloser(bytes.NewBuffer(decryptedStruct))
+		response.Body = ioutil.NopCloser(bytes.NewReader(decryptedStruct))
 		response.ContentLength = int64(len(decryptedStruct))
 		return response
 	default:
@@ -138,7 +135,7 @@ func (decryptor *HTTPConnectionsDecryptor) ParseRequestPrepareResponse(logger *l
 	return responseWithMessage(request, http.StatusBadRequest, msg)
 }
 
-func (decryptor *HTTPConnectionsDecryptor) decryptAcraStruct(acraStruct []byte, zoneId []byte, clientId []byte) ([]byte, error) {
+func (decryptor *HTTPConnectionsDecryptor) decryptAcraStruct(logger *log.Entry, acraStruct []byte, zoneId []byte, clientId []byte) ([]byte, error) {
 	var err error
 	var privateKey *keys.PrivateKey
 	var decryptionContext []byte = nil
@@ -151,6 +148,7 @@ func (decryptor *HTTPConnectionsDecryptor) decryptAcraStruct(acraStruct []byte,
 	}
 
 	if err != nil {
+		logger.Errorln("Can't load private key to decrypt AcraStruct")
 		return nil, err
 	}
 
@@ -182,8 +180,8 @@ func emptyResponseWithStatus(request *http.Request, status int) *http.Response {
 func responseWithMessage(request *http.Request, status int, body string) *http.Response {
 	response := emptyResponseWithStatus(request, status)
 	response.Header.Set("Content-Type", "text/plain")
-	response.Body = ioutil.NopCloser(bytes.NewBufferString(body))
-	response.ContentLength = int64(len(body))
+	response.Body = ioutil.NopCloser(bytes.NewReader([]byte(body)))
+	response.ContentLength = int64(len([]byte(body)))
 	return response
 }
 

cmd/acra-reader/http-api/decryptor_test.go

@@ -171,7 +171,7 @@ func TestHTTPDecryptionAcraStruct(t *testing.T) {
 	data := []byte("some data")
 
 	// not an acrastruct
-	decrypted, err := httpConnectionsDecryptor.decryptAcraStruct([]byte("some garbage not acrastruct"), nil, clientId)
+	decrypted, err := httpConnectionsDecryptor.decryptAcraStruct(nil, []byte("some garbage not acrastruct"), nil, clientId)
 	if err == nil {
 		t.Fatalf("Should not be able to decrypt garbage")
 	}
@@ -182,7 +182,7 @@ func TestHTTPDecryptionAcraStruct(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	decrypted, err = httpConnectionsDecryptor.decryptAcraStruct(acrastruct, nil, clientId)
+	decrypted, err = httpConnectionsDecryptor.decryptAcraStruct(nil, acrastruct, nil, clientId)
 	if err != nil {
 		t.Fatalf("Should be able to decrypt acrastruct without zone")
 	}
@@ -198,7 +198,7 @@ func TestHTTPDecryptionAcraStruct(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	decrypted, err = httpConnectionsDecryptor.decryptAcraStruct(acrastructWithZone, zoneId, clientId)
+	decrypted, err = httpConnectionsDecryptor.decryptAcraStruct(nil, acrastructWithZone, zoneId, clientId)
 	if err != nil {
 		t.Fatalf("Should be able to decrypt acrastruct with zone")
 	}

cmd/acra-reader/server.go

@@ -112,13 +112,23 @@ func (server *ReaderServer) HandleConnectionString(parentContext context.Context
 			logging.SetLoggerToContext(parentContext, logger)
 
 			go func() {
+				defer func () {
+					err := wrappedConnection.Close()
+					if err != nil {
+						logger.WithError(err).Errorln("Can't close wrapped connection")
+					}
+					logger.Infoln("Connection closed")
+				}()
+
 				if err := server.connectionManager.AddConnection(wrappedConnection); err != nil {
-					logger.WithError(err).Errorln("can't add connection to connection manager")
+					logger.WithError(err).Errorln("Can't add connection to connection manager")
 					return
 				}
+
 				processingFunc(parentContext, clientId, wrappedConnection)
+
 				if err := server.connectionManager.RemoveConnection(wrappedConnection); err != nil {
-					logger.WithError(err).Errorln("can't remove connection from connection manager")
+					logger.WithError(err).Errorln("Can't remove connection from connection manager")
 				}
 			}()
 		}
@@ -128,7 +138,7 @@ func (server *ReaderServer) HandleConnectionString(parentContext context.Context
 	case <-parentContext.Done():
 		log.WithError(parentContext.Err()).Debugln("Exit from handling connection string. Close all connections")
 	case outErr = <-errCh:
-		log.WithError(err).Errorln("error on accepting new connections")
+		log.WithError(err).Errorln("Error on accepting new connections")
 
 	}
 	return outErr
@@ -199,11 +209,11 @@ func (server *ReaderServer) processHTTPConnection(parentContext context.Context,
 	if err != nil {
 		logger.WithError(err).WithField(logging.FieldKeyEventCode, logging.EventCodeErrorReaderCantHandleHTTPRequest).
 			Warningln("Got new HTTP request, but can't read it")
-		server.httpDecryptor.SendResponseAndCloseConnection(logger,
+		server.httpDecryptor.SendResponse(logger,
 			server.httpDecryptor.EmptyResponseWithStatus(request, http.StatusBadRequest), connection)
 		return
 	}
 
 	response := server.httpDecryptor.ParseRequestPrepareResponse(logger, request, clientId)
-	server.httpDecryptor.SendResponseAndCloseConnection(logger, response, connection)
+	server.httpDecryptor.SendResponse(logger, response, connection)
 }