-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
8a33a69
commit 367cdfa
Showing
8 changed files
with
454 additions
and
107 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
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
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,180 @@ | ||
| // SPDX-License-Identifier: GPL-2.0-or-later | ||
|
|
||
| package devices | ||
|
|
||
| import ( | ||
| "encoding/json" | ||
| "errors" | ||
| "fmt" | ||
| "strconv" | ||
| "strings" | ||
|
|
||
| log "github.com/sirupsen/logrus" | ||
|
|
||
| "github.com/redhat-partner-solutions/vse-sync-collection-tools/pkg/callbacks" | ||
| "github.com/redhat-partner-solutions/vse-sync-collection-tools/pkg/clients" | ||
| "github.com/redhat-partner-solutions/vse-sync-collection-tools/pkg/fetcher" | ||
| ) | ||
|
|
||
| type DevNetlinkDPLLInfo struct { | ||
| Timestamp string `fetcherKey:"date" json:"timestamp"` | ||
| EECState string `fetcherKey:"eec" json:"eecstate"` | ||
| PPSState string `fetcherKey:"pps" json:"state"` | ||
| } | ||
|
|
||
| // AnalyserJSON returns the json expected by the analysers | ||
| func (dpllInfo *DevNetlinkDPLLInfo) GetAnalyserFormat() ([]*callbacks.AnalyserFormatType, error) { | ||
| formatted := callbacks.AnalyserFormatType{ | ||
| ID: "dpll/states", | ||
| Data: map[string]any{ | ||
| "timestamp": dpllInfo.Timestamp, | ||
| "eecstate": dpllInfo.EECState, | ||
| "state": dpllInfo.PPSState, | ||
| }, | ||
| } | ||
| return []*callbacks.AnalyserFormatType{&formatted}, nil | ||
| } | ||
|
|
||
| type NetlinkEntry struct { | ||
| LockStatus string `json:"lock-status"` //nolint:tagliatelle // not my choice | ||
| Driver string `json:"module-name"` //nolint:tagliatelle // not my choice | ||
| ClockType string `json:"type"` //nolint:tagliatelle // not my choice | ||
| ClockID int64 `json:"clock-id"` //nolint:tagliatelle // not my choice | ||
| ID int `json:"id"` //nolint:tagliatelle // not my choice | ||
| } | ||
|
|
||
| var ( | ||
| dpllNetlinkFetcher map[int64]*fetcher.Fetcher | ||
| dpllClockIDFetcher map[string]*fetcher.Fetcher | ||
| ) | ||
|
|
||
| func init() { | ||
| dpllNetlinkFetcher = make(map[int64]*fetcher.Fetcher) | ||
| dpllClockIDFetcher = make(map[string]*fetcher.Fetcher) | ||
| } | ||
|
|
||
| func buildPostProcessDPLLNetlink(clockID int64) fetcher.PostProcessFuncType { | ||
| return func(result map[string]string) (map[string]any, error) { | ||
| processedResult := make(map[string]any) | ||
|
|
||
| entries := make([]NetlinkEntry, 0) | ||
| cleaned := strings.ReplaceAll(result["dpll-netlink"], "'", "\"") | ||
| log.Info("dpll ", cleaned) | ||
| err := json.Unmarshal([]byte(cleaned), &entries) | ||
| if err != nil { | ||
| log.Errorf("Failed to unmarshal netlink output: %s", err.Error()) | ||
| } | ||
|
|
||
| log.Info("entries: ", entries) | ||
| for _, entry := range entries { | ||
| if entry.ClockID == clockID { | ||
| processedResult[entry.ClockType] = entry.LockStatus | ||
| } | ||
| } | ||
| return processedResult, nil | ||
| } | ||
| } | ||
|
|
||
| // BuildDPLLInfoFetcher popluates the fetcher required for | ||
| // collecting the DPLLInfo | ||
| func BuildDPLLNetlinkInfoFetcher(clockID int64) error { //nolint:dupl // Further dedup risks be too abstract or fragile | ||
| fetcherInst, err := fetcher.FetcherFactory( | ||
| []*clients.Cmd{dateCmd}, | ||
| []fetcher.AddCommandArgs{ | ||
| { | ||
| Key: "dpll-netlink", | ||
| Command: "/linux/tools/net/ynl/cli.py --spec /linux/Documentation/netlink/specs/dpll.yaml --dump device-get", | ||
| Trim: true, | ||
| }, | ||
| }, | ||
| ) | ||
| if err != nil { | ||
| log.Errorf("failed to create fetcher for dpll: %s", err.Error()) | ||
| return fmt.Errorf("failed to create fetcher for dpll: %w", err) | ||
| } | ||
| dpllNetlinkFetcher[clockID] = fetcherInst | ||
| fetcherInst.SetPostProcessor(buildPostProcessDPLLNetlink(clockID)) | ||
| return nil | ||
| } | ||
|
|
||
| // GetDevDPLLInfo returns the device DPLL info for an interface. | ||
| func GetDevDPLLNetlinkInfo(ctx clients.ExecContext, clockID int64) (DevNetlinkDPLLInfo, error) { | ||
| dpllInfo := DevNetlinkDPLLInfo{} | ||
| fetcherInst, fetchedInstanceOk := dpllNetlinkFetcher[clockID] | ||
| if !fetchedInstanceOk { | ||
| err := BuildDPLLNetlinkInfoFetcher(clockID) | ||
| if err != nil { | ||
| return dpllInfo, err | ||
| } | ||
| fetcherInst, fetchedInstanceOk = dpllNetlinkFetcher[clockID] | ||
| if !fetchedInstanceOk { | ||
| return dpllInfo, errors.New("failed to create fetcher for DPLLInfo using netlink interface") | ||
| } | ||
| } | ||
| err := fetcherInst.Fetch(ctx, &dpllInfo) | ||
| if err != nil { | ||
| log.Debugf("failed to fetch dpllInfo %s", err.Error()) | ||
| return dpllInfo, fmt.Errorf("failed to fetch dpllInfo %w", err) | ||
| } | ||
| return dpllInfo, nil | ||
| } | ||
|
|
||
| func BuildClockIDFetcher(interfaceName string) error { | ||
| fetcherInst, err := fetcher.FetcherFactory( | ||
| []*clients.Cmd{dateCmd}, | ||
| []fetcher.AddCommandArgs{ | ||
| { | ||
| Key: "dpll-netlink-clock-id", | ||
| Command: fmt.Sprintf( | ||
| `export IFNAME=%s; export BUSID=$(readlink /sys/class/net/$IFNAME/device | xargs basename | cut -d ':' -f 2,3);`+ | ||
| ` echo $(("16#$(lspci -v | grep $BUSID -A 20 |grep 'Serial Number' | awk '{print $NF}' | tr -d '-')"))`, | ||
| interfaceName, | ||
| ), | ||
| Trim: true, | ||
| }, | ||
| }, | ||
| ) | ||
| if err != nil { | ||
| log.Errorf("failed to create fetcher for dpll: %s", err.Error()) | ||
| return fmt.Errorf("failed to create fetcher for dpll: %w", err) | ||
| } | ||
| fetcherInst.SetPostProcessor(postProcessDPLLNetlinkClockID) | ||
| dpllClockIDFetcher[interfaceName] = fetcherInst | ||
| return nil | ||
| } | ||
|
|
||
| func postProcessDPLLNetlinkClockID(result map[string]string) (map[string]any, error) { | ||
| processedResult := make(map[string]any) | ||
| clockID, err := strconv.ParseInt(result["dpll-netlink-clock-id"], 10, 64) | ||
| if err != nil { | ||
| return processedResult, fmt.Errorf("failed to parse int for clock id: %w", err) | ||
| } | ||
| processedResult["clockID"] = clockID | ||
| return processedResult, nil | ||
| } | ||
|
|
||
| type NetlinkClockID struct { | ||
| Timestamp string `fetcherKey:"date" json:"timestamp"` | ||
| ClockID int64 `fetcherKey:"clockID" json:"clockId"` | ||
| } | ||
|
|
||
| func GetClockID(ctx clients.ExecContext, interfaceName string) (NetlinkClockID, error) { | ||
| clockID := NetlinkClockID{} | ||
| fetcherInst, fetchedInstanceOk := dpllClockIDFetcher[interfaceName] | ||
| if !fetchedInstanceOk { | ||
| err := BuildClockIDFetcher(interfaceName) | ||
| if err != nil { | ||
| return clockID, err | ||
| } | ||
| fetcherInst, fetchedInstanceOk = dpllClockIDFetcher[interfaceName] | ||
| if !fetchedInstanceOk { | ||
| return clockID, errors.New("failed to create fetcher for DPLLInfo using netlink interface") | ||
| } | ||
| } | ||
| err := fetcherInst.Fetch(ctx, &clockID) | ||
| if err != nil { | ||
| log.Debugf("failed to fetch netlink clockID %s", err.Error()) | ||
| return clockID, fmt.Errorf("failed to fetch netlink clockID %w", err) | ||
| } | ||
| return clockID, nil | ||
| } |
Oops, something went wrong.