replaced arborist with arborist bundle & common func for analysis

This commit is contained in:
Sahilb315
2025-04-09 00:58:14 +05:30
parent e1e57a3ecd
commit a190d945ec
6 changed files with 69485 additions and 75 deletions
+43
View File
@@ -0,0 +1,43 @@
package analyser
import (
"context"
"fmt"
"buf.build/gen/go/safedep/api/grpc/go/safedep/services/malysis/v1/malysisv1grpc"
malysisv1pb "buf.build/gen/go/safedep/api/protocolbuffers/go/safedep/messages/malysis/v1"
packagev1 "buf.build/gen/go/safedep/api/protocolbuffers/go/safedep/messages/package/v1"
malysisv1 "buf.build/gen/go/safedep/api/protocolbuffers/go/safedep/services/malysis/v1"
)
func SubmitPackageForAnalysis(client malysisv1grpc.MalwareAnalysisServiceClient,
ecosystem packagev1.Ecosystem, name string,
version string) (string, error) {
req := &malysisv1.AnalyzePackageRequest{
Target: &malysisv1pb.PackageAnalysisTarget{
PackageVersion: &packagev1.PackageVersion{
Package: &packagev1.Package{
Ecosystem: ecosystem,
Name: name,
},
Version: version,
},
},
}
resp, err := client.AnalyzePackage(context.Background(), req)
if err != nil {
return "", fmt.Errorf("failed to analyze %s@%s: %v", name, version, err)
}
return resp.GetAnalysisId(), nil
}
func GetAnalysisReport(client malysisv1grpc.MalwareAnalysisServiceClient,
analysisId string) (*malysisv1pb.Report, error) {
analysisReportReq := &malysisv1.GetAnalysisReportRequest{
AnalysisId: analysisId,
}
reportResp, err := client.GetAnalysisReport(context.Background(), analysisReportReq)
if err != nil {
return nil, fmt.Errorf("failed to get analysis report: %v", err)
}
return reportResp.GetReport(), nil
}
+27
View File
@@ -0,0 +1,27 @@
package analyser
import (
"fmt"
"net/http"
"os"
"buf.build/gen/go/safedep/api/grpc/go/safedep/services/malysis/v1/malysisv1grpc"
drygrpc "github.com/safedep/dry/adapters/grpc"
"google.golang.org/grpc"
)
func GetMalwareAnalysisClient() (malysisv1grpc.MalwareAnalysisServiceClient, error) {
tok := os.Getenv("SAFEDEP_API_KEY")
tenantId := os.Getenv("SAFEDEP_TENANT_ID")
if tok == "" || tenantId == "" {
return nil, fmt.Errorf("SAFEDEP_API_KEY and SAFEDEP_TENANT_ID must be set")
}
headers := http.Header{}
headers.Set("x-tenant-id", tenantId)
cc, err := drygrpc.GrpcClient("pmg-pkg-scan", "api.safedep.io", "443",
tok, headers, []grpc.DialOption{})
if err != nil {
return nil, fmt.Errorf("failed to create gRPC client: %v", err)
}
return malysisv1grpc.NewMalwareAnalysisServiceClient(cc), nil
}