/
main.go
119 lines (95 loc) · 2.85 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package main
import (
"context"
"cred-alert/config"
"cred-alert/revokpb"
"crypto/tls"
"fmt"
"io"
"log"
"os"
"os/signal"
"regexp"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
flags "github.com/jessevdk/go-flags"
)
type Opts struct {
RPCServerAddress string `long:"rpc-server-address" description:"Address for RPC server." required:"true"`
RPCServerPort uint16 `long:"rpc-server-port" description:"Port for RPC server." required:"true"`
CACertPath string `long:"ca-cert-path" description:"Path to the CA certificate" required:"true"`
ClientCertPath string `long:"client-cert-path" description:"Path to the client certificate" required:"true"`
ClientKeyPath string `long:"client-key-path" description:"Path to the client private key" required:"true"`
ClientKeyPassphrase string `long:"client-key-passphrase" description:"Passphrase for the client private key, if encrypted"`
Query string `long:"query" short:"q" description:"Regular expression to search for" required:"true"`
}
func main() {
var opts Opts
_, err := flags.ParseArgs(&opts, os.Args)
if err != nil {
os.Exit(1)
}
serverAddr := fmt.Sprintf("%s:%d", opts.RPCServerAddress, opts.RPCServerPort)
clientCert, err := config.LoadCertificateFromFiles(
opts.ClientCertPath,
opts.ClientKeyPath,
opts.ClientKeyPassphrase,
)
if err != nil {
log.Fatalln(err)
}
rootCertPool, err := config.LoadCertificatePoolFromFiles(opts.CACertPath)
if err != nil {
log.Fatalln(err)
}
transportCreds := credentials.NewTLS(&tls.Config{
Certificates: []tls.Certificate{clientCert},
RootCAs: rootCertPool,
})
fmt.Println("Establishing connection...")
conn, err := grpc.Dial(
serverAddr,
grpc.WithTransportCredentials(transportCreds),
)
if err != nil {
log.Fatalf("failed to create handler: %s", err.Error())
}
defer conn.Close()
revokClient := revokpb.NewRevokSearchClient(conn)
ctx, cancel := context.WithCancel(context.Background())
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for range c {
fmt.Print("cancelling search... ")
cancel()
fmt.Println("done")
os.Exit(127)
}
}()
fmt.Println("Sending search query to server...")
stream, err := revokClient.Search(ctx, &revokpb.SearchQuery{
Regex: opts.Query,
})
if err != nil {
log.Fatalln(err.Error())
}
for {
result, err := stream.Recv()
if err == io.EOF {
break
}
if err != nil {
log.Fatalln(err.Error())
}
show(result)
}
}
var unsafeChars = regexp.MustCompile(`[[:^print:]]`)
func show(result *revokpb.SearchResult) {
location := result.GetLocation()
repo := location.GetRepository()
content := result.GetContent()
safe := unsafeChars.ReplaceAllLiteral(content, []byte{})
fmt.Printf("[%s/%s@%s] %s:%d: %s\n", repo.GetOwner(), repo.GetName(), location.GetRevision()[:7], location.GetPath(), location.GetLineNumber(), safe)
}