-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
create vendor dir for implants and script to update it
- Loading branch information
1 parent
d94c596
commit 8c8dd28
Showing
1,519 changed files
with
454,342 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| package implant | ||
|
|
||
| //go:generate ./scripts/update-vendor.sh |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "fmt" | ||
| "go/parser" | ||
| "go/token" | ||
| "io/fs" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
| ) | ||
|
|
||
| // These are the directories imports will be taken from. If the implant | ||
| // needs files outside of these dirs in the future, add the dirs to this | ||
| // slice. | ||
| var srcDirs = []string{ | ||
| "../sliver", | ||
| "../../protobuf/commonpb", | ||
| "../../protobuf/dnspb", | ||
| "../../protobuf/sliverpb", | ||
| } | ||
|
|
||
| func fatalf(format string, a ...interface{}) { | ||
| panic(fmt.Sprintf(format, a...)) | ||
| } | ||
|
|
||
| func getImports(imports map[string]struct{}, dir string) error { | ||
| fset := token.NewFileSet() | ||
| return filepath.WalkDir(dir, func(path string, d fs.DirEntry, err error) error { | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if d.IsDir() { | ||
| return nil | ||
| } | ||
| if !strings.HasSuffix(path, ".go") { | ||
| return nil | ||
| } | ||
|
|
||
| file, err := parser.ParseFile(fset, path, nil, parser.ImportsOnly) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| for _, imp := range file.Imports { | ||
| impPath := strings.ToLower(imp.Path.Value) | ||
| // skip local packages, those shouldn't be vendored | ||
| if strings.HasPrefix(impPath, `"github.com/bishopfox/sliver`) { | ||
| continue | ||
| } | ||
|
|
||
| imports[imp.Path.Value] = struct{}{} | ||
| } | ||
|
|
||
| return nil | ||
| }) | ||
| } | ||
|
|
||
| func main() { | ||
| if len(os.Args) < 2 { | ||
| fatalf("expected temporary directory argument") | ||
| } | ||
|
|
||
| dir := os.Args[1] | ||
| imports := make(map[string]struct{}) | ||
|
|
||
| // Parse Go files and build a unique list of all their 3rd party | ||
| // imports so we can build a vendor dir from them. Stdlib packages | ||
| // will be included as well but 'go mod vendor' will ignore them. | ||
| for _, srcDir := range srcDirs { | ||
| err := getImports(imports, srcDir) | ||
| if err != nil { | ||
| fatalf("error walking directory: %v", err) | ||
| } | ||
| } | ||
|
|
||
| // Create a Go source file containing only import statements of all | ||
| // the packages an implant might need, no matter what OS, transport, | ||
| // etc is configured. | ||
| var buf bytes.Buffer | ||
| buf.WriteString("package sliver\n\nimport (\n") | ||
| for imp := range imports { | ||
| buf.WriteString("\t _ ") | ||
| buf.WriteString(imp) | ||
| buf.WriteByte('\n') | ||
| } | ||
| buf.WriteString(")\n") | ||
|
|
||
| err := os.WriteFile(filepath.Join(dir, "imports.go"), buf.Bytes(), 0660) | ||
| if err != nil { | ||
| fatalf("error creating file: %v", err) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,27 @@ | ||
| #!/bin/sh | ||
|
|
||
| set -e | ||
|
|
||
| # we are expecting from run from sliver/implant via 'go generate' | ||
| cd scripts | ||
|
|
||
| # copy Go module related files | ||
| pwd="$(pwd)" | ||
| tempDir="$(mktemp -d -p $pwd)" | ||
| cp ../go-mod "${tempDir}/go.mod" | ||
| cp ../go-sum "${tempDir}/go.sum" | ||
| mv ../vendor "${tempDir}/vendor" | ||
|
|
||
| # build Go file with all imported packages | ||
| go run update-vendor.go "$tempDir" | ||
| cd "$tempDir" | ||
| # update vendor dir | ||
| go mod tidy -compat=1.17 | ||
| go mod vendor | ||
|
|
||
| # move updated files back | ||
| mv "${tempDir}/vendor" ../../vendor | ||
| mv "${tempDir}/go.mod" ../../go-mod | ||
| mv "${tempDir}/go.sum" ../../go-sum | ||
| cd .. | ||
| rm -rf "$tempDir" |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Oops, something went wrong.