-
Notifications
You must be signed in to change notification settings - Fork 16
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add TLS communication between Galadriel Server and Harvester #146
Changes from 3 commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1 +1,3 @@ | ||
package cli | ||
|
||
// TODO: add tests |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,32 @@ | ||
-----BEGIN CERTIFICATE----- | ||
MIIFkjCCA3qgAwIBAgIUaCjXS2q5FyByi//FJRIoVoxwNLowDQYJKoZIhvcNAQEL | ||
BQAwMTELMAkGA1UEBhMCVVMxDzANBgNVBAoMBlNQSUZGRTERMA8GA1UEAwwIcGVy | ||
ZnRlc3QwHhcNMjMwMzA4MTgwMjAxWhcNMzQwNTI1MTgwMjAxWjAxMQswCQYDVQQG | ||
EwJVUzEPMA0GA1UECgwGU1BJRkZFMREwDwYDVQQDDAhwZXJmdGVzdDCCAiIwDQYJ | ||
KoZIhvcNAQEBBQADggIPADCCAgoCggIBAN3NNPUn6NK7PthpCcGMboyWU/jusjHc | ||
DALVbPsnGaSq2fup/d+7bQ7ElY4R26fK7KibIh8S1s6uGgrjBnOKkJRmVhxxtqZC | ||
C8devw/zWEdEAY5VeVHlN7gzKBJpmy3U3HBwhx6eFqwB85enz/9Y6FgWU5jbTkw5 | ||
SyPGMo+jiUPWBhaKH2tGoa3kujbrTcCjyRWYu6jyk6ZjSyJXWriY526HZwaFl6/c | ||
C64AYGdnRSPAFYKi0s4FILd11kbgno2UqZqUb+LkuYyrMBMd9VPjISQKGmKprcIU | ||
6KY79LTELse5lOHFmJWbELpUtyWOglWEpTlwWdr4AmJwcfS9po6FY2CLmDIi9Xgy | ||
aLYJwLh4xllqSg1GCx44FIeOqIVjJYNbWxF1EHPk9ZCaRNemFV6PRns4ePCZnjtp | ||
hZjSXxkIXsJfu5CHjn88OIEby0rxNCFREMZZ/o22UXI9Iuq72/SjS4nT5W7Wo/K3 | ||
m4YvrbfIBVxPTwoK4w5fLXJPa5rcPgRbtch3UQX6/gLY3akzfFH+mGlv2V06fxRx | ||
hbrQIcZGyPf5TvugW0UIcFLv3RW1ScxQjcvajPDEZzpLArLiC/lnD3rmXUgau7SQ | ||
4nUqHT/STL7KPDu0XMa7omSQfqljwed978Rg1gqNiKOB0WSu9+phGxLakad5/k+e | ||
mbtLkG0RZiA5AgMBAAGjgaEwgZ4wDwYDVR0TAQH/BAUwAwEB/zAdBgNVHQ4EFgQU | ||
E13a4pFoH0GGLDghCoqJ3N68kJIwbAYDVR0jBGUwY4AUE13a4pFoH0GGLDghCoqJ | ||
3N68kJKhNaQzMDExCzAJBgNVBAYTAlVTMQ8wDQYDVQQKDAZTUElGRkUxETAPBgNV | ||
BAMMCHBlcmZ0ZXN0ghRoKNdLarkXIHKL/8UlEihWjHA0ujANBgkqhkiG9w0BAQsF | ||
AAOCAgEAh4rjrwnUI0gEtHgKXCnx5rcT2DBAy/9xAOxdsDOVqheiDWoC0NK9Iet8 | ||
m/01uwY5v+PJ4gmdVGO3V/FOvb3KOAiilZPFmI0jUYRpceUTh7FkRsZPLFeGYJ7K | ||
OPJ6BzX1Y+aHcMK6jaTtHrgxS2gt1cWEB4nSYFQKKVqCmExrZpiFDw0ZY3RTFXSc | ||
LA1jV1/1OD05G6UxdHSNbTzQRlNaLPeE+EL9eyiIF2RxvnQul6qZwed35nh1TqjN | ||
n+nNSH8LSZxdwQLNEn0CYQ2LuUeAr6EdvfFibse2fyTU+HBSyuH4SRXjvFC4WuSr | ||
pPtnGPtKwVVCNhCYiixOMPCTOL68ZWK7XC7AaHSX7pD7DuRZU2WfAJ4gSD/jUJel | ||
ku2Kxl2NXeXrlkzMs6Ud022mLL/sFSd4WLuB/G+fk7wgwdezk4syISHSGQ0+Xech | ||
QqpvPqj/+hi3C2yUxzZojxVSBJkB9+Z8LFrk8qpO0VREO5KD/9444Ric/CTgjQmg | ||
qfGlCd4P+2YYEd8em7VIE5d/yDJ8rGLT45d7AwepWJzXEZBb1DMDcGVbSSISDyCh | ||
ZAmmTw37cxEfp+7nKAGor64lX3DbRiPQV/MGv6MhNPNRyBPH7kkeZuE6ewh8bdfV | ||
8YAhxXpLl11iLC4bZbj2AGlZ+Cr0WEVaX7g7S82cdrmc208zBR8= | ||
-----END CERTIFICATE----- |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
package constants | ||
|
||
const ( | ||
GaladrielServerName = "galadriel-server" | ||
) |
Original file line number | Diff line number | Diff line change | ||||||||||||||||||||||||||||||||||||||
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
@@ -3,12 +3,17 @@ package client | |||||||||||||||||||||||||||||||||||||||
import ( | ||||||||||||||||||||||||||||||||||||||||
"bytes" | ||||||||||||||||||||||||||||||||||||||||
"context" | ||||||||||||||||||||||||||||||||||||||||
"crypto/tls" | ||||||||||||||||||||||||||||||||||||||||
"crypto/x509" | ||||||||||||||||||||||||||||||||||||||||
"encoding/json" | ||||||||||||||||||||||||||||||||||||||||
"fmt" | ||||||||||||||||||||||||||||||||||||||||
"io" | ||||||||||||||||||||||||||||||||||||||||
"net" | ||||||||||||||||||||||||||||||||||||||||
"net/http" | ||||||||||||||||||||||||||||||||||||||||
"os" | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
"github.com/HewlettPackard/galadriel/pkg/common" | ||||||||||||||||||||||||||||||||||||||||
"github.com/HewlettPackard/galadriel/pkg/common/constants" | ||||||||||||||||||||||||||||||||||||||||
"github.com/HewlettPackard/galadriel/pkg/common/telemetry" | ||||||||||||||||||||||||||||||||||||||||
"github.com/sirupsen/logrus" | ||||||||||||||||||||||||||||||||||||||||
) | ||||||||||||||||||||||||||||||||||||||||
|
@@ -29,23 +34,31 @@ type GaladrielServerClient interface { | |||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
type client struct { | ||||||||||||||||||||||||||||||||||||||||
c http.Client | ||||||||||||||||||||||||||||||||||||||||
address string | ||||||||||||||||||||||||||||||||||||||||
c *http.Client | ||||||||||||||||||||||||||||||||||||||||
address *net.TCPAddr | ||||||||||||||||||||||||||||||||||||||||
token string | ||||||||||||||||||||||||||||||||||||||||
logger logrus.FieldLogger | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
func NewGaladrielServerClient(address, token string) (GaladrielServerClient, error) { | ||||||||||||||||||||||||||||||||||||||||
// NewGaladrielServerClient creates a new Galadriel Server client, using the given token to authenticate | ||||||||||||||||||||||||||||||||||||||||
// and the given trustBundlePath to validate the server certificate. | ||||||||||||||||||||||||||||||||||||||||
func NewGaladrielServerClient(address *net.TCPAddr, token string, trustBundlePath string) (GaladrielServerClient, error) { | ||||||||||||||||||||||||||||||||||||||||
c, err := createTLSClient(trustBundlePath) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
return nil, fmt.Errorf("failed to create TLS client: %w", err) | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
return &client{ | ||||||||||||||||||||||||||||||||||||||||
c: *http.DefaultClient, | ||||||||||||||||||||||||||||||||||||||||
address: "http://" + address, | ||||||||||||||||||||||||||||||||||||||||
c: c, | ||||||||||||||||||||||||||||||||||||||||
address: address, | ||||||||||||||||||||||||||||||||||||||||
token: token, | ||||||||||||||||||||||||||||||||||||||||
logger: logrus.WithField(telemetry.SubsystemName, telemetry.GaladrielServerClient), | ||||||||||||||||||||||||||||||||||||||||
}, nil | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
func (c *client) Connect(ctx context.Context, token string) error { | ||||||||||||||||||||||||||||||||||||||||
url := c.address + onboardPath | ||||||||||||||||||||||||||||||||||||||||
url := fmt.Sprintf("https://%s%s", c.address.String(), onboardPath) | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
req, err := http.NewRequestWithContext(ctx, http.MethodConnect, url, nil) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
return err | ||||||||||||||||||||||||||||||||||||||||
|
@@ -77,7 +90,7 @@ func (c *client) SyncFederatedBundles(ctx context.Context, req *common.SyncBundl | |||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
c.logger.Debugf("Sending post federated bundles updates:\n%s", b) | ||||||||||||||||||||||||||||||||||||||||
url := c.address + postBundleSyncPath | ||||||||||||||||||||||||||||||||||||||||
url := c.address.String() + postBundleSyncPath | ||||||||||||||||||||||||||||||||||||||||
r, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(b)) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
return nil, fmt.Errorf("failed to create request: %v", err) | ||||||||||||||||||||||||||||||||||||||||
|
@@ -117,7 +130,7 @@ func (c *client) PostBundle(ctx context.Context, req *common.PostBundleRequest) | |||||||||||||||||||||||||||||||||||||||
return fmt.Errorf("failed to marshal push bundle request: %v", err) | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
url := c.address + postBundlePath | ||||||||||||||||||||||||||||||||||||||||
url := c.address.String() + postBundlePath | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
r, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(b)) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
|
@@ -147,6 +160,32 @@ func (c *client) PostBundle(ctx context.Context, req *common.PostBundleRequest) | |||||||||||||||||||||||||||||||||||||||
return nil | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
func createTLSClient(trustBundlePath string) (*http.Client, error) { | ||||||||||||||||||||||||||||||||||||||||
caCert, err := os.ReadFile(trustBundlePath) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
return nil, err | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
caCertPool := x509.NewCertPool() | ||||||||||||||||||||||||||||||||||||||||
ok := caCertPool.AppendCertsFromPEM(caCert) | ||||||||||||||||||||||||||||||||||||||||
if !ok { | ||||||||||||||||||||||||||||||||||||||||
return nil, fmt.Errorf("failed to append CA certificates") | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
tr := &http.Transport{ | ||||||||||||||||||||||||||||||||||||||||
TLSClientConfig: &tls.Config{ | ||||||||||||||||||||||||||||||||||||||||
RootCAs: caCertPool, | ||||||||||||||||||||||||||||||||||||||||
ServerName: constants.GaladrielServerName, | ||||||||||||||||||||||||||||||||||||||||
}, | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
client := &http.Client{ | ||||||||||||||||||||||||||||||||||||||||
Transport: tr, | ||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
return client, nil | ||||||||||||||||||||||||||||||||||||||||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
Suggested change
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I'd actually have the client and transport on a single sentence as you suggested (or two sentences as it is now) and the return statement as it is. It makes the actual return values easier to read. There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I agree refactored it to:
I don't like to nest more than two struct literals, it becomes hard to read. |
||||||||||||||||||||||||||||||||||||||||
} | ||||||||||||||||||||||||||||||||||||||||
|
||||||||||||||||||||||||||||||||||||||||
func readBody(resp *http.Response) (string, error) { | ||||||||||||||||||||||||||||||||||||||||
bodyBytes, err := io.ReadAll(resp.Body) | ||||||||||||||||||||||||||||||||||||||||
if err != nil { | ||||||||||||||||||||||||||||||||||||||||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Did you considered moving some of the consts defined in our telemetry package here?