Delete log.Fatal calls and replace with error returns. (#9086)

* Delete log.Fatal calls and replace with error returns.

* Update opcua_util.go

* Update opcua_util.go
This commit is contained in:
David Bennett 2021-04-06 17:35:54 -04:00 committed by GitHub
parent 868befcb5f
commit ef2def20dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 62 additions and 118 deletions

View File

@ -440,13 +440,16 @@ func (o *OpcUA) setupOptions() error {
if o.Certificate == "" && o.PrivateKey == "" { if o.Certificate == "" && o.PrivateKey == "" {
if o.SecurityPolicy != "None" || o.SecurityMode != "None" { if o.SecurityPolicy != "None" || o.SecurityMode != "None" {
o.Certificate, o.PrivateKey = generateCert("urn:telegraf:gopcua:client", 2048, o.Certificate, o.PrivateKey, (365 * 24 * time.Hour)) o.Certificate, o.PrivateKey, err = generateCert("urn:telegraf:gopcua:client", 2048, o.Certificate, o.PrivateKey, (365 * 24 * time.Hour))
if err != nil {
return err
}
} }
} }
o.opts = generateClientOpts(endpoints, o.Certificate, o.PrivateKey, o.SecurityPolicy, o.SecurityMode, o.AuthMethod, o.Username, o.Password, time.Duration(o.RequestTimeout)) o.opts, err = generateClientOpts(endpoints, o.Certificate, o.PrivateKey, o.SecurityPolicy, o.SecurityMode, o.AuthMethod, o.Username, o.Password, time.Duration(o.RequestTimeout))
return nil return err
} }
func (o *OpcUA) getData() error { func (o *OpcUA) getData() error {

View File

@ -31,11 +31,11 @@ func newTempDir() (string, error) {
return dir, err return dir, err
} }
func generateCert(host string, rsaBits int, certFile, keyFile string, dur time.Duration) (string, string) { func generateCert(host string, rsaBits int, certFile, keyFile string, dur time.Duration) (cert string, key string, err error) {
dir, _ := newTempDir() dir, _ := newTempDir()
if len(host) == 0 { if len(host) == 0 {
log.Fatalf("Missing required host parameter") return "", "", fmt.Errorf("missing required host parameter")
} }
if rsaBits == 0 { if rsaBits == 0 {
rsaBits = 2048 rsaBits = 2048
@ -49,7 +49,7 @@ func generateCert(host string, rsaBits int, certFile, keyFile string, dur time.D
priv, err := rsa.GenerateKey(rand.Reader, rsaBits) priv, err := rsa.GenerateKey(rand.Reader, rsaBits)
if err != nil { if err != nil {
log.Fatalf("failed to generate private key: %s", err) return "", "", fmt.Errorf("failed to generate private key: %s", err)
} }
notBefore := time.Now() notBefore := time.Now()
@ -58,7 +58,7 @@ func generateCert(host string, rsaBits int, certFile, keyFile string, dur time.D
serialNumberLimit := new(big.Int).Lsh(big.NewInt(1), 128) serialNumberLimit := new(big.Int).Lsh(big.NewInt(1), 128)
serialNumber, err := rand.Int(rand.Reader, serialNumberLimit) serialNumber, err := rand.Int(rand.Reader, serialNumberLimit)
if err != nil { if err != nil {
log.Fatalf("failed to generate serial number: %s", err) return "", "", fmt.Errorf("failed to generate serial number: %s", err)
} }
template := x509.Certificate{ template := x509.Certificate{
@ -88,33 +88,33 @@ func generateCert(host string, rsaBits int, certFile, keyFile string, dur time.D
derBytes, err := x509.CreateCertificate(rand.Reader, &template, &template, publicKey(priv), priv) derBytes, err := x509.CreateCertificate(rand.Reader, &template, &template, publicKey(priv), priv)
if err != nil { if err != nil {
log.Fatalf("Failed to create certificate: %s", err) return "", "", fmt.Errorf("failed to create certificate: %s", err)
} }
certOut, err := os.Create(certFile) certOut, err := os.Create(certFile)
if err != nil { if err != nil {
log.Fatalf("failed to open %s for writing: %s", certFile, err) return "", "", fmt.Errorf("failed to open %s for writing: %s", certFile, err)
} }
if err := pem.Encode(certOut, &pem.Block{Type: "CERTIFICATE", Bytes: derBytes}); err != nil { if err := pem.Encode(certOut, &pem.Block{Type: "CERTIFICATE", Bytes: derBytes}); err != nil {
log.Fatalf("failed to write data to %s: %s", certFile, err) return "", "", fmt.Errorf("failed to write data to %s: %s", certFile, err)
} }
if err := certOut.Close(); err != nil { if err := certOut.Close(); err != nil {
log.Fatalf("error closing %s: %s", certFile, err) return "", "", fmt.Errorf("error closing %s: %s", certFile, err)
} }
keyOut, err := os.OpenFile(keyFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600) keyOut, err := os.OpenFile(keyFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
if err != nil { if err != nil {
log.Printf("failed to open %s for writing: %s", keyFile, err) log.Printf("failed to open %s for writing: %s", keyFile, err)
return "", "" return "", "", nil
} }
if err := pem.Encode(keyOut, pemBlockForKey(priv)); err != nil { if err := pem.Encode(keyOut, pemBlockForKey(priv)); err != nil {
log.Fatalf("failed to write data to %s: %s", keyFile, err) return "", "", fmt.Errorf("failed to write data to %s: %s", keyFile, err)
} }
if err := keyOut.Close(); err != nil { if err := keyOut.Close(); err != nil {
log.Fatalf("error closing %s: %s", keyFile, err) return "", "", fmt.Errorf("error closing %s: %s", keyFile, err)
} }
return certFile, keyFile return certFile, keyFile, nil
} }
func publicKey(priv interface{}) interface{} { func publicKey(priv interface{}) interface{} {
@ -144,9 +144,8 @@ func pemBlockForKey(priv interface{}) *pem.Block {
} }
} }
// OPT FUNCTIONS //revive:disable-next-line
func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile, policy, mode, auth, username, password string, requestTimeout time.Duration) ([]opcua.Option, error) {
func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile, policy, mode, auth, username, password string, requestTimeout time.Duration) []opcua.Option {
opts := []opcua.Option{} opts := []opcua.Option{}
appuri := "urn:telegraf:gopcua:client" appuri := "urn:telegraf:gopcua:client"
appname := "Telegraf" appname := "Telegraf"
@ -157,9 +156,13 @@ func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile,
opts = append(opts, opcua.RequestTimeout(requestTimeout)) opts = append(opts, opcua.RequestTimeout(requestTimeout))
var err error
if certFile == "" && keyFile == "" { if certFile == "" && keyFile == "" {
if policy != "None" || mode != "None" { if policy != "None" || mode != "None" {
certFile, keyFile = generateCert(appuri, 2048, certFile, keyFile, (365 * 24 * time.Hour)) certFile, keyFile, err = generateCert(appuri, 2048, certFile, keyFile, (365 * 24 * time.Hour))
if err != nil {
return nil, err
}
} }
} }
@ -172,7 +175,7 @@ func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile,
} else { } else {
pk, ok := c.PrivateKey.(*rsa.PrivateKey) pk, ok := c.PrivateKey.(*rsa.PrivateKey)
if !ok { if !ok {
log.Fatalf("Invalid private key") return nil, fmt.Errorf("invalid private key")
} }
cert = c.Certificate[0] cert = c.Certificate[0]
opts = append(opts, opcua.PrivateKey(pk), opcua.Certificate(cert)) opts = append(opts, opcua.PrivateKey(pk), opcua.Certificate(cert))
@ -190,11 +193,15 @@ func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile,
secPolicy = ua.SecurityPolicyURIPrefix + policy secPolicy = ua.SecurityPolicyURIPrefix + policy
policy = "" policy = ""
default: default:
log.Fatalf("Invalid security policy: %s", policy) return nil, fmt.Errorf("invalid security policy: %s", policy)
} }
// Select the most appropriate authentication mode from server capabilities and user input // Select the most appropriate authentication mode from server capabilities and user input
authMode, authOption := generateAuth(auth, cert, username, password) authMode, authOption, err := generateAuth(auth, cert, username, password)
if err != nil {
return nil, err
}
opts = append(opts, authOption) opts = append(opts, authOption)
var secMode ua.MessageSecurityMode var secMode ua.MessageSecurityMode
@ -210,7 +217,7 @@ func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile,
secMode = ua.MessageSecurityModeSignAndEncrypt secMode = ua.MessageSecurityModeSignAndEncrypt
mode = "" mode = ""
default: default:
log.Fatalf("Invalid security mode: %s", mode) return nil, fmt.Errorf("invalid security mode: %s", mode)
} }
// Allow input of only one of sec-mode,sec-policy when choosing 'None' // Allow input of only one of sec-mode,sec-policy when choosing 'None'
@ -252,24 +259,23 @@ func generateClientOpts(endpoints []*ua.EndpointDescription, certFile, keyFile,
} }
if serverEndpoint == nil { // Didn't find an endpoint with matching policy and mode. if serverEndpoint == nil { // Didn't find an endpoint with matching policy and mode.
log.Printf("unable to find suitable server endpoint with selected sec-policy and sec-mode") return nil, fmt.Errorf("unable to find suitable server endpoint with selected sec-policy and sec-mode")
log.Fatalf("quitting")
} }
secPolicy = serverEndpoint.SecurityPolicyURI secPolicy = serverEndpoint.SecurityPolicyURI
secMode = serverEndpoint.SecurityMode secMode = serverEndpoint.SecurityMode
// Check that the selected endpoint is a valid combo // Check that the selected endpoint is a valid combo
err := validateEndpointConfig(endpoints, secPolicy, secMode, authMode) err = validateEndpointConfig(endpoints, secPolicy, secMode, authMode)
if err != nil { if err != nil {
log.Fatalf("error validating input: %s", err) return nil, fmt.Errorf("error validating input: %s", err)
} }
opts = append(opts, opcua.SecurityFromEndpoint(serverEndpoint, authMode)) opts = append(opts, opcua.SecurityFromEndpoint(serverEndpoint, authMode))
return opts return opts, nil
} }
func generateAuth(a string, cert []byte, un, pw string) (ua.UserTokenType, opcua.Option) { func generateAuth(a string, cert []byte, un, pw string) (ua.UserTokenType, opcua.Option, error) {
var err error var err error
var authMode ua.UserTokenType var authMode ua.UserTokenType
@ -284,13 +290,13 @@ func generateAuth(a string, cert []byte, un, pw string) (ua.UserTokenType, opcua
if un == "" { if un == "" {
if err != nil { if err != nil {
log.Fatalf("error reading username input: %s", err) return 0, nil, fmt.Errorf("error reading the username input: %s", err)
} }
} }
if pw == "" { if pw == "" {
if err != nil { if err != nil {
log.Fatalf("error reading username input: %s", err) return 0, nil, fmt.Errorf("error reading the password input: %s", err)
} }
} }
@ -311,7 +317,7 @@ func generateAuth(a string, cert []byte, un, pw string) (ua.UserTokenType, opcua
authOption = opcua.AuthAnonymous() authOption = opcua.AuthAnonymous()
} }
return authMode, authOption return authMode, authOption, nil
} }
func validateEndpointConfig(endpoints []*ua.EndpointDescription, secPolicy string, secMode ua.MessageSecurityMode, authMode ua.UserTokenType) error { func validateEndpointConfig(endpoints []*ua.EndpointDescription, secPolicy string, secMode ua.MessageSecurityMode, authMode ua.UserTokenType) error {

View File

@ -2,7 +2,6 @@ package github
import ( import (
"fmt" "fmt"
"log"
"time" "time"
"github.com/influxdata/telegraf" "github.com/influxdata/telegraf"
@ -107,10 +106,7 @@ func (s CommitCommentEvent) NewMetric() telegraf.Metric {
"commit": s.Comment.Commit, "commit": s.Comment.Commit,
"comment": s.Comment.Body, "comment": s.Comment.Body,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -137,10 +133,7 @@ func (s CreateEvent) NewMetric() telegraf.Metric {
"ref": s.Ref, "ref": s.Ref,
"refType": s.RefType, "refType": s.RefType,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -167,10 +160,7 @@ func (s DeleteEvent) NewMetric() telegraf.Metric {
"ref": s.Ref, "ref": s.Ref,
"refType": s.RefType, "refType": s.RefType,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -198,10 +188,7 @@ func (s DeploymentEvent) NewMetric() telegraf.Metric {
"environment": s.Deployment.Environment, "environment": s.Deployment.Environment,
"description": s.Deployment.Description, "description": s.Deployment.Description,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -232,10 +219,7 @@ func (s DeploymentStatusEvent) NewMetric() telegraf.Metric {
"depState": s.DeploymentStatus.State, "depState": s.DeploymentStatus.State,
"depDescription": s.DeploymentStatus.Description, "depDescription": s.DeploymentStatus.Description,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -260,10 +244,7 @@ func (s ForkEvent) NewMetric() telegraf.Metric {
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
"fork": s.Forkee.Repository, "fork": s.Forkee.Repository,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -288,10 +269,7 @@ func (s GollumEvent) NewMetric() telegraf.Metric {
"forks": s.Repository.Forks, "forks": s.Repository.Forks,
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -320,10 +298,7 @@ func (s IssueCommentEvent) NewMetric() telegraf.Metric {
"comments": s.Issue.Comments, "comments": s.Issue.Comments,
"body": s.Comment.Body, "body": s.Comment.Body,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -352,10 +327,7 @@ func (s IssuesEvent) NewMetric() telegraf.Metric {
"title": s.Issue.Title, "title": s.Issue.Title,
"comments": s.Issue.Comments, "comments": s.Issue.Comments,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -381,10 +353,7 @@ func (s MemberEvent) NewMetric() telegraf.Metric {
"newMember": s.Member.User, "newMember": s.Member.User,
"newMemberStatus": s.Member.Admin, "newMemberStatus": s.Member.Admin,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -407,10 +376,7 @@ func (s MembershipEvent) NewMetric() telegraf.Metric {
"newMember": s.Member.User, "newMember": s.Member.User,
"newMemberStatus": s.Member.Admin, "newMemberStatus": s.Member.Admin,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -433,10 +399,7 @@ func (s PageBuildEvent) NewMetric() telegraf.Metric {
"forks": s.Repository.Forks, "forks": s.Repository.Forks,
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -459,10 +422,7 @@ func (s PublicEvent) NewMetric() telegraf.Metric {
"forks": s.Repository.Forks, "forks": s.Repository.Forks,
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -496,10 +456,7 @@ func (s PullRequestEvent) NewMetric() telegraf.Metric {
"deletions": s.PullRequest.Deletions, "deletions": s.PullRequest.Deletions,
"changedFiles": s.PullRequest.ChangedFiles, "changedFiles": s.PullRequest.ChangedFiles,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -534,10 +491,7 @@ func (s PullRequestReviewCommentEvent) NewMetric() telegraf.Metric {
"commentFile": s.Comment.File, "commentFile": s.Comment.File,
"comment": s.Comment.Comment, "comment": s.Comment.Comment,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -566,10 +520,7 @@ func (s PushEvent) NewMetric() telegraf.Metric {
"before": s.Before, "before": s.Before,
"after": s.After, "after": s.After,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -594,10 +545,7 @@ func (s ReleaseEvent) NewMetric() telegraf.Metric {
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
"tagName": s.Release.TagName, "tagName": s.Release.TagName,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -620,10 +568,7 @@ func (s RepositoryEvent) NewMetric() telegraf.Metric {
"forks": s.Repository.Forks, "forks": s.Repository.Forks,
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -650,10 +595,7 @@ func (s StatusEvent) NewMetric() telegraf.Metric {
"commit": s.Commit, "commit": s.Commit,
"state": s.State, "state": s.State,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -678,10 +620,7 @@ func (s TeamAddEvent) NewMetric() telegraf.Metric {
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
"teamName": s.Team.Name, "teamName": s.Team.Name,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }
@ -704,9 +643,6 @@ func (s WatchEvent) NewMetric() telegraf.Metric {
"forks": s.Repository.Forks, "forks": s.Repository.Forks,
"issues": s.Repository.Issues, "issues": s.Repository.Issues,
} }
m, err := metric.New(meas, t, f, time.Now()) m, _ := metric.New(meas, t, f, time.Now())
if err != nil {
log.Fatalf("Failed to create %v event", event)
}
return m return m
} }

View File

@ -100,7 +100,6 @@ func UnrollXMLFields(data []byte, fieldsUsage map[string]int, separator string)
break break
} }
if err != nil { if err != nil {
// log.Fatal(err)
break break
} }
var parents []string var parents []string