github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/configuration/config_manager/proc_util.go (about) 1 /* 2 Copyright (C) 2022-2023 ApeCloud Co., Ltd 3 4 This file is part of KubeBlocks project 5 6 This program is free software: you can redistribute it and/or modify 7 it under the terms of the GNU Affero General Public License as published by 8 the Free Software Foundation, either version 3 of the License, or 9 (at your option) any later version. 10 11 This program is distributed in the hope that it will be useful 12 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 GNU Affero General Public License for more details. 15 16 You should have received a copy of the GNU Affero General Public License 17 along with this program. If not, see <http://www.gnu.org/licenses/>. 18 */ 19 20 package configmanager 21 22 import ( 23 "context" 24 "fmt" 25 26 "github.com/shirou/gopsutil/v3/process" 27 28 appsv1alpha1 "github.com/1aal/kubeblocks/apis/apps/v1alpha1" 29 cfgcore "github.com/1aal/kubeblocks/pkg/configuration/core" 30 ) 31 32 // findParentPIDByProcessName gets parent pid 33 func findParentPIDByProcessName(processName string, ctx ...context.Context) (PID, error) { 34 var fctx context.Context 35 36 if len(ctx) == 0 { 37 fctx = context.TODO() 38 } else { 39 fctx = ctx[0] 40 } 41 allProcess, err := process.ProcessesWithContext(fctx) 42 if err != nil { 43 return InvalidPID, err 44 } 45 46 psGraph := map[PID]int32{} 47 for _, proc := range allProcess { 48 name, err := proc.Name() 49 // OS X getting the name of the system process may fail, 50 // because OS X Process.Name function depends on sysctl and elevated permissions 51 if err != nil { 52 logger.Error(err, fmt.Sprintf("failed to get process name from pid[%d], and pass", proc.Pid)) 53 continue 54 } 55 if name != processName { 56 continue 57 } 58 ppid, err := proc.Ppid() 59 if err != nil { 60 return InvalidPID, cfgcore.WrapError(err, "failed to get parent pid from pid[%d]", proc.Pid) 61 } 62 psGraph[PID(proc.Pid)] = ppid 63 } 64 65 for key, value := range psGraph { 66 if _, ok := psGraph[PID(value)]; !ok { 67 return key, nil 68 } 69 } 70 71 return InvalidPID, cfgcore.MakeError("cannot find pid of process name: [%s]", processName) 72 } 73 74 func IsValidUnixSignal(sig appsv1alpha1.SignalType) bool { 75 _, ok := allUnixSignals[sig] 76 return ok 77 }