2022/03/30 v2.1.0

README.MD

@@ -17,7 +17,7 @@
 
 > 1. 账号管理
 > 2. 用户管理
-> 3. 电源选项
+> 3. 电源管理
 > 4. 实用工具(软件安装等)
 > 5. 文件共享
 
@@ -81,6 +81,7 @@
 - 2020/10/27 v1.10.2 优化软件安装和重启小工具的程序逻辑,更新64位的程序清单和新ico文件
 - 2020/11/05 v1.10.3 更新备份策略
 - 2022/03/29 v2.0.0 删除备份策略,同步域计算机信息到cmdb,添加账户、主机管理管理,唤醒策略更新,日志统一收集处理,运维平台添加管理员指令
+- 2022/03/30 v2.1.0 支持通过cmdb获取字段来唤醒主机,日志输出优化,支持通过运维平台打开远程终端
 
 ## 五、其他说明
 

adct/adct.go

@@ -13,6 +13,11 @@ const INFO = pub.INFO
 const ERROR = pub.ERROR
 const NULL = pub.NULL
 
+var f struct {
+	Force bool
+	Quick bool
+}
+
 func main() {
 
 	var reportText = flag.String(pub.Msg_Report, "", "report to AD")
@@ -22,9 +27,14 @@ func main() {
 	var hourText = flag.Bool("hour", false, "exec task at the time")
 	var cmdbtokenText = flag.Bool("cmdbtoken", false, "")
 	var loglevel = flag.String("loglevel", INFO, "")
+	var force = flag.Bool("force", false, "force exec,ignore other")
+	var quick = flag.Bool("quick", false, "quickly exec")
 
 	flag.Parse()
 
+	f.Force = *force
+	f.Quick = *quick
+
 	if *loglevel != "" {
 		loglevelSet(strings.ToLower(*loglevel))
 	}
@@ -116,12 +126,36 @@ func wolMac(mac *string) {
 	pub.Wol_enter("", mac)
 }
 func wolHost(pc *string) {
-	// coding
-	var mac string
-	pub.LOG(INFO, "wol_host", fmt.Sprintf("pc:%s,mac:%s", *pc, mac))
-	pub.Wol_enter(*pc, &mac)
+	var hostname string = *pc
+
+	// 判断主机名的格式
+	if _, err := pub.GetHostnamePart(hostname); err != nil {
+		pub.LOG(ERROR, "WOL_HOST", err)
+		return
+	}
+	mac, err := pub.CmdbGetWordFromKV(pub.Cmdb_Product_name, hostname, pub.Cmdb_Mac)
+	if err != nil {
+		pub.LOG(ERROR, "WOL_HOST", err)
+		return
+	}
 
-	//暂未开发
+	// 快速时，跳过远程连接
+	if !f.Quick {
+		pub.LOG(INFO, "WOL_HOST", "测试目标连通性")
+		alive, err := pub.PSTest_Connection(hostname)
+		if err != nil {
+			pub.LOG(ERROR, "WOL_HOST", err)
+			return
+		}
+		if alive {
+			pub.LOG(INFO, "WOL_HOST", fmt.Sprintf("pc:%s,mac:%s 已经运行", *pc, mac))
+			return
+		}
+	}
+
+	// 开始唤醒
+	pub.LOG(INFO, "WOL_HOST", fmt.Sprintf("pc:%s,mac:%s", *pc, mac))
+	pub.Wol_enter(*pc, &mac)
 }
 func wolALL() {
 	// coding get pclist

daemon/daemon.go

@@ -18,6 +18,8 @@ func main() {
 }
 
 func listenMsg() {
+	pub.LOG(pub.INFO, pub.NULL, "start MSG daemon!")
+
 	http.HandleFunc("/", pub.DealMsgEnter)
 	http.ListenAndServe(pub.DaemonListen, nil)
 }

public/cmdb.go

@@ -12,21 +12,18 @@ import (
 
 var client = http.Client{}
 
-const cmdb_Dept string = "bumen"
-const cmdb_Product_name string = "product_name"
-const cmdb_Username string = "username"
-
+// 将键值转化为json
 func cmdbKVtoJSON(k, v string) (string, error) {
 
 	var err error
 	var b []byte
 	switch k {
-	case cmdb_Product_name:
+	case Cmdb_Product_name:
 		var a CIRecorProduct_name
 		a.Product_name = v
 		b, err = json.Marshal(&a)
 
-	case cmdb_Username:
+	case Cmdb_Username:
 		var a CIRecordUsername
 		a.Username = v
 
@@ -35,6 +32,8 @@ func cmdbKVtoJSON(k, v string) (string, error) {
 
 	return string(b), err
 }
+
+// 键值将上传到cmdb，如果存在返回ci，不存在将添加记录
 func cmdbExistRecord(k, v string) (int, bool) {
 
 	var count string = "1"
@@ -76,7 +75,7 @@ func cmdbExistRecord(k, v string) (int, bool) {
 			LOG(ERROR, CMDB, err)
 			return 0, false
 		} else {
-			l := fmt.Sprintf("CMDB\nnew host %s:%s", k, v)
+			l := fmt.Sprintf("CMDB\nnew hostanme %s", v)
 			LOG(INFO, "CMDB_NEW", l)
 			SendWxworkTextToAdmins(l)
 		}
@@ -90,6 +89,7 @@ func cmdbExistRecord(k, v string) (int, bool) {
 	return 0, false
 }
 
+// 将json格式的字段根据ci更新到cmdb
 func cmdbUpdateRecordFromJSON(datastr string, ci int) error {
 
 	var cmdbPath string = fmt.Sprintf("api/v0.1/ci/%d", ci)
@@ -111,6 +111,8 @@ func cmdbUpdateRecordFromJSON(datastr string, ci int) error {
 	}
 	return nil
 }
+
+// 获取token并保存于注册表
 func CmdbTokenUpdate() {
 	type res struct {
 		Token string `json:"token"`
@@ -137,6 +139,8 @@ func CmdbTokenUpdate() {
 		LOG(INFO, msg_type, "token更新完成")
 	}
 }
+
+// 根据键值获取CI
 func cmdbGetCI(k, v string) (int, error) {
 
 	var count string = "1"
@@ -157,14 +161,84 @@ func cmdbGetCI(k, v string) (int, error) {
 	LOG(DEBUG, "CMDB_EXITE", string(data))
 	switch cir.Numfound {
 	case 0:
-		return 0, fmt.Errorf("不存在记录")
+		return 0, Error_CMDB_No_Record
 	case 1:
 		return cir.R[0].Ci_id, nil
 	default:
-		return 0, fmt.Errorf("存在多条记录")
+		return 0, Error_CMDB_Many_Record
+
+	}
+}
+
+// 根据键值获取1个字段
+func CmdbGetWordFromKV(k, v, w string) (string, error) {
+
+	var count string = "1"
+	var cmdbPath string = "api/v0.1/ci/s?q="
+	search := fmt.Sprintf("%s:%s", k, v)
+	url := fmt.Sprintf("%s/%s_type:%s,%s&count=%s", cmdbUrl, cmdbPath, cmdbCITypeComputer, search, count)
+	LOG(DEBUG, CMDB, url)
+	var cir CIRecordReq
+	data, err := cmdbGET(url)
+	if err != nil {
+		LOG(ERROR, CMDB, err)
+		return "", err
+	}
+	if err := json.Unmarshal(data, &cir); err != nil {
+		LOG(ERROR, CMDB, err)
+		return "", err
+	}
+	LOG(DEBUG, "CMDB_EXITE", string(data))
+	switch cir.Numfound {
+	case 0:
+		return "", Error_CMDB_No_Record
+	case 1:
+		return getVFromK(&cir.R[0], w), nil
+
+	default:
+		return "", Error_CMDB_Many_Record
 
 	}
 }
+
+// 根据键值获取2个字段
+func CmdbGetWordFromKV2(k, v, a, b string) (string, string, error) {
+
+	var count string = "1"
+	var cmdbPath string = "api/v0.1/ci/s?q="
+	search := fmt.Sprintf("%s:%s", k, v)
+	url := fmt.Sprintf("%s/%s_type:%s,%s&count=%s", cmdbUrl, cmdbPath, cmdbCITypeComputer, search, count)
+	LOG(DEBUG, CMDB, url)
+	var cir CIRecordReq
+	data, err := cmdbGET(url)
+	if err != nil {
+		LOG(ERROR, CMDB, err)
+		return "", "", err
+	}
+	if err := json.Unmarshal(data, &cir); err != nil {
+		LOG(ERROR, CMDB, err)
+		return "", "", err
+	}
+	LOG(DEBUG, "CMDB_EXITE", string(data))
+	switch cir.Numfound {
+	case 0:
+		return "", "", Error_CMDB_No_Record
+	case 1:
+		return getVFromK(&cir.R[0], a), getVFromK(&cir.R[0], b), nil
+
+	default:
+		return "", "", Error_CMDB_Many_Record
+
+	}
+}
+func getVFromK(cir *CIRecordResult, k string) string {
+	var r string
+	switch k {
+	case Cmdb_Mac:
+		r = cir.Mac
+	}
+	return r
+}
 func cmdbPOST(url string, reqdata []byte) ([]byte, error) {
 	buf := bytes.NewBuffer(nil)
 	buf.Write(reqdata)

public/cmdbjson.go

@@ -6,6 +6,11 @@ import (
 	"sync"
 )
 
+const cmdb_Dept string = "bumen"
+const Cmdb_Product_name string = "product_name"
+const Cmdb_Username string = "username"
+const Cmdb_Mac string = "Mac"
+
 type CIRecordReq struct {
 	Counter  CIRecordComputer `json:"counter"`
 	Facet    CIRecordFacet    `json:"facet"`
@@ -150,7 +155,7 @@ func (cir *CIRecordResult) position(location, dept string) string {
 			resutlt = cmdbPosition1
 		case "TXHD":
 			resutlt = cmdbPosition1
-		case "TXMT":
+		case "TXWJ":
 			resutlt = cmdbPosition1
 		case "JS":
 			resutlt = cmdbPosition1

public/env.go

@@ -8,7 +8,7 @@ const SymbolHostname = "-"
 
 // AD-Control //
 
-const Version string = "2.0.0"
+const Version string = "2.1.0"
 const Host_adserver string = "ADSERVER"
 const Host_thserver string = "THSERVER"
 const Host_zzserver string = "ZZSERVER"
@@ -18,6 +18,20 @@ const ADServerDaemon string = Host_adserver + DaemonListen
 const ADServerLogDaemon string = Host_adserver + DaemonLogListen
 const THServerDaemon string = Host_thserver + DaemonListen
 
+//
+//
+// AD User env //
+//
+
+const AD_User_main_admin string = "zhiweiadserver"
+const AD_User_tengfei string = "tengfei"
+const AD_User_taotengfei string = "taotengfei"
+const AD_User_ADControl string = "ad-control@zhiweireach.com"
+const AD_User_ADControl_key string = "golang&*90-="
+
+// users //
+const Users_Timeout int = 3
+
 // user //
 
 var User_computername string = GetEnv("COMPUTERNAME")
@@ -42,15 +56,11 @@ const SMB_ADSoftPlus string = SMB_ADSoft + `plus\`
 // Local Folder //
 
 var Dir_userprofile string = GetEnv("USERPROFILE")
-var Dir_appdata string = GetEnv("appdata")
 var Dir_tmp string = GetEnv("tmp")
-var Dir_Windows string = GetEnv("windir") + `\`
 
 const Dir_ADsoftware_local string = `H:\software\softlike\`
 
 // application //
-
-const App_adct string = SMB_ADSoftApplication + "adct.exe"
 const App_MultiSplit string = SMB_ADSoftOthTool + "win_MultiSplit.exe"
 
 // reg info //
@@ -60,9 +70,9 @@ const (
 	HKLM
 	HKU
 )
-const reg_LM_ADCONTROL string = `software\AD-Control`
 
 // reg value //
+const reg_LM_ADCONTROL string = `software\AD-Control`
 const Reg_Key_cmdbtoken string = "cmdb_token"
 const Reg_Value_ON string = "ON"
 const Reg_Value_OFF string = "OFF"

public/error.go

@@ -21,8 +21,11 @@ var Error_Not_Msg = errors.New("isn't defined msg")
 
 var Error_WXWork_Down = errors.New(Error_WXWork_Down_STR)
 var Error_WXWork_Down_STR = "后台更新中..."
-
 var Error_WXWork_NotDominUser_STR = "您的主机暂时不支持该功能"
 
 const Error_CMDB_UPDATE_REQ = "更新失败"
 const Sccessful_CMDB_UPDATE_REQ = "更新成功"
+
+// cmdb
+var Error_CMDB_No_Record = errors.New("CMDB_No_Record")
+var Error_CMDB_Many_Record = errors.New("_CMDB_Many_Record")

public/ldap.go

@@ -4,13 +4,26 @@ import (
 	ldap "gopkg.in/ldap.v3"
 )
 
+func ldapBase() (*ldap.Conn, error) {
+	l, err := ldap.DialURL("ldap://127.0.0.1:389")
+	if err != nil {
+		LOG(ERROR, NULL, err)
+
+		return nil, err
+	}
+	if err = l.Bind(AD_User_ADControl, AD_User_ADControl_key); err != nil {
+		LOG(ERROR, NULL, err)
+		return nil, err
+	}
+	return l, err
+}
 func LDAP_Get_PCGroup() []string {
 	l, err := ldap.DialURL("ldap://127.0.0.1:389")
 	if err != nil {
 		LOG(ERROR, NULL, err)
 	}
 	defer l.Close()
-	if err = l.Bind(AD_User_zhiwei, "1Q2W3e4r"); err != nil {
+	if err = l.Bind(AD_User_ADControl, AD_User_ADControl_key); err != nil {
 		LOG(ERROR, NULL, err)
 	}
 	searchRequest := ldap.NewSearchRequest(
@@ -35,14 +48,8 @@ func LDAP_Get_PCGroup() []string {
 	return pcgroup
 }
 func LDAP_nameTopinyin(name string) string {
-	l, err := ldap.DialURL("ldap://127.0.0.1:389")
-	if err != nil {
-		LOG(ERROR, NULL, err)
-	}
-	defer l.Close()
-	if err = l.Bind(AD_User_zhiwei, "1Q2W3e4r"); err != nil {
-		LOG(ERROR, NULL, err)
-	}
+	l, err := ldapBase()
+
 	searchRequest := ldap.NewSearchRequest(
 		//"ou=情报部门,dc=zhiweireach,dc=com",
 		"ou=情报部门,dc=zhiweireach,dc=com",
@@ -56,6 +63,32 @@ func LDAP_nameTopinyin(name string) string {
 	if err != nil {
 		LOG(ERROR, NULL, err)
 	}
+	defer l.Close()
 	return sr.Entries[0].GetAttributeValue("sAMAccountName")
 
 }
+
+func LDAP_CheckExistName(name string) (bool, error) {
+	var r bool
+	l, _ := ldapBase()
+	defer l.Close()
+
+	searchRequest := ldap.NewSearchRequest(
+		"dc=zhiweireach,dc=com",
+		ldap.ScopeWholeSubtree, ldap.NeverDerefAliases, 0, 0,
+		false,
+		"(cn="+name+")",
+		[]string{"cn"},
+		nil,
+	)
+	sr, err := l.Search(searchRequest)
+	if err != nil {
+		return false, err
+	}
+	if len(sr.Entries) == 0 {
+		r = false
+	} else {
+		r = true
+	}
+	return r, err
+}

public/msg_deal.go

@@ -7,10 +7,7 @@ import (
 	"strings"
 )
 
-//
-//
 // 消息入口
-//
 
 func DealMsgEnter(w http.ResponseWriter, r *http.Request) {
 
@@ -21,48 +18,39 @@ func DealMsgEnter(w http.ResponseWriter, r *http.Request) {
 		LOG(ERROR, NULL, err)
 		w.Write([]byte(""))
 	}
-	ret := DealMsg(msgtype, url, unmsg)
-	w.Write(ret)
 
+	w.Write(DealMsg(msgtype, url, unmsg))
 }
+
+// 处理消息
 func DealMsg(msgtype, url string, unmsg []byte) []byte {
 	var ret []byte
 
 	switch msgtype {
 	// 报告 消息
 	case Msg_Report:
-		var pmsg MJreport
-		// LOG(INFO, "REPORT", string(unmsg))
-		ParseJson(unmsg, &pmsg)
-		ice := pmsg.MsgDeal()
+		var msg MJreport
+		ParseJson(unmsg, &msg)
+		ice := msg.MsgDeal()
 		ret = ReParseJson(&ice)
 
 	// 执行 消息
 	case Msg_Exec:
-		var pmsg MJexec
-		// LOG(INFO, "EXEC", string(unmsg))
-		ParseJson(unmsg, &pmsg)
-		ice := pmsg.MsgDeal()
+		var msg MJexec
+		ParseJson(unmsg, &msg)
+		ice := msg.MsgDeal()
 		ret = ReParseJson(&ice)
 
 	// 唤醒 消息
 	case Msg_Wake:
-		var pmsg MJwake
-		ParseJson(unmsg, &pmsg)
-		ice := pmsg.MsgDeal()
-		ret = ReParseJson(&ice)
-
-	// 错误 消息
-	case Msg_Error:
-		LOG(ERROR, NULL, string(unmsg))
-		var pmsg MJerror
-		ParseJson(unmsg, &pmsg)
-		ice := pmsg.MsgDeal()
+		var msg MJwake
+		ParseJson(unmsg, &msg)
+		ice := msg.MsgDeal()
 		ret = ReParseJson(&ice)
 
 	default:
-		pmsg := ParseWX(url, &unmsg)
-		go pmsg.MsgDealSend()
+		msg := ParseWX(url, &unmsg)
+		go msg.MsgDealSend()
 	}
 	return ret
 }
@@ -83,14 +71,14 @@ func (rep *MJreport) MsgDeal() interface{} {
 		// 服务器
 		case Msg_status_commit:
 			// 第一次仅仅会更新部分字段，第二次会更新全部字段
-			if ci, r := cmdbExistRecord(cmdb_Product_name, rep.Computername); !r {
+			if ci, r := cmdbExistRecord(Cmdb_Product_name, rep.Computername); !r {
 				break
 			} else {
 				err := cmdbUpdateRecordFromJSON(rep.DataStr, ci)
 				if err != nil {
-					LOG(ERROR, "CMDB_UPDATE", err)
+					LOG(ERROR, rep.Instruction, err)
 				} else {
-					LOG(INFO, "CMDB_UPDATE", rep.Computername)
+					LOG(INFO, rep.Instruction, rep.Computername)
 				}
 			}
 		// 客户端处理
@@ -98,25 +86,18 @@ func (rep *MJreport) MsgDeal() interface{} {
 			return nil
 		}
 
-		// coding
-		// rep.Reg_Write_ADDC(
-		// 	Reg_Name_TurnOffTime,
-		// rep.Reg_Write_ADDC(
-		// 	Reg_Name_PowerStatus,
-		// 	Reg_Value_OFF)
-
 	case Msg_Report_WillTurnOnPC:
 		switch rep.Status {
 		// 服务器
 		case Msg_status_commit:
-			if ci, r := cmdbExistRecord(cmdb_Product_name, rep.Computername); !r {
+			if ci, r := cmdbExistRecord(Cmdb_Product_name, rep.Computername); !r {
 				break
 			} else {
 				err := cmdbUpdateRecordFromJSON(rep.DataStr, ci)
 				if err != nil {
-					LOG(ERROR, "CMDB_UPDATE", err)
+					LOG(ERROR, rep.Instruction, err)
 				} else {
-					LOG(INFO, "CMDB_UPDATE", rep.Computername)
+					LOG(INFO, rep.Instruction, rep.Computername)
 				}
 			}
 		// 客户端
@@ -124,22 +105,6 @@ func (rep *MJreport) MsgDeal() interface{} {
 			return nil
 		}
 
-		// coding
-		// 如果是被开机唤醒
-		// id = rep.Reg_Query_ADDC(Reg_Name_BeWakedID)
-		// if id != "" {
-		// 	SendWxworkTextToAUser(id, "主机已启动,一切就绪~")
-		// 	// rep.Reg_Write_ADDC(Reg_Name_BeWakedID, "")
-		// }
-		// coding
-		// 记录 开机 时间
-		// rep.Reg_Write_ADDC(
-		// 	Reg_Name_TurnOnTime,
-		// 记录 电源状态
-		// rep.Reg_Write_ADDC(
-		// 	Reg_Name_PowerStatus,
-		// 	Reg_Value_ON)
-
 	case Msg_Report_WillLogInUser:
 	case Msg_Report_WillLogOutUser:
 
@@ -148,13 +113,13 @@ func (rep *MJreport) MsgDeal() interface{} {
 		switch rep.Status {
 		// 服务器
 		case Msg_status_commit:
-			ci, err := cmdbGetCI(cmdb_Product_name, rep.Computername)
+			ci, err := cmdbGetCI(Cmdb_Product_name, rep.Computername)
 			if err != nil {
 				rep.DataStr = Error_CMDB_UPDATE_REQ
 				LOG(ERROR, rep.Instruction, err)
 				break
 			}
-			a, err := cmdbKVtoJSON(cmdb_Username, rep.DataStr)
+			a, err := cmdbKVtoJSON(Cmdb_Username, rep.DataStr)
 			if err != nil {
 				rep.DataStr = Error_CMDB_UPDATE_REQ
 				LOG(ERROR, rep.Instruction, err)
@@ -165,7 +130,7 @@ func (rep *MJreport) MsgDeal() interface{} {
 				LOG(ERROR, rep.Instruction, err)
 				break
 			}
-			LOG(INFO, rep.Instruction, fmt.Sprintf("%s is %s's PrimryComputer", cmdb_Product_name, rep.DataStr))
+			LOG(INFO, rep.Instruction, fmt.Sprintf("%s is %s's PrimryComputer", Cmdb_Product_name, rep.DataStr))
 			rep.DataStr = Sccessful_CMDB_UPDATE_REQ
 
 		// 客户端
@@ -248,12 +213,17 @@ func (exec *MJexec) MsgDeal() interface{} {
 		} else {
 			PsExec_sfdic(exec.Computername, Dir_ADsoftware_local+exec.Command, exec.SessionID)
 		}
-
 		exec.Status = Msg_Exec_State_Over
+
 	case Msg_Exec_OpenAdminEXE:
 		PsExec_sdi(exec.Computername, exec.Command, exec.SessionID)
 		exec.Status = Msg_Exec_State_Over
 
+	case Msg_Exec_OpenRemoteTerminal:
+		exec.Command = "powershell"
+		SendWxworkTextToAdmins(exec.ToWXWrokAdmin())
+		exec.Status = Msg_Exec_State_Over
+
 	default:
 		Execcmd_nowait(exec.Command)
 	}
@@ -275,13 +245,6 @@ func (wake *MJwake) MsgDeal() interface{} {
 	return nil
 }
 
-// 关于 error 错误 的 消息处理
-
-func (err *MJerror) MsgDeal() interface{} {
-	//SendWX(err.Instruction)
-	return nil
-}
-
 // 关于 wxwork 企业微信 的 消息处理
 
 func (wxwork *MJwxwork) MsgDealSend() {
@@ -289,31 +252,32 @@ func (wxwork *MJwxwork) MsgDealSend() {
 	switch wxwork.Instruction {
 	case WXEK_zwDevopsPowerWake:
 
-		wxwork.SendText("收到唤醒指令，正在处理,")
+		if c, err := LDAP_CheckExistName(wxwork.Name); err != nil {
+			LOG(ERROR, wxwork.Instruction, err)
+		} else {
+			if !c {
+				wxwork.SendText("请先申请知微域的个人账号!")
+				return
+			}
+		}
 
-		//var domainUser bool = true
 		var pc, mac string
 		var err error
-
-		// coding
-		// 在注册表中查询 域用户 记录
-		// if err = Reg_IfExist_item(Reg_Root_ac, Reg_Path_home_du+wxwork.Name); err == nil {
-		// 	pc = wxwork.Reg_Query_ADUsers(Reg_Name_PrimayComputerName)
-		// 	mac = wxwork.Reg_Query_ADUsers(Reg_Name_MACAddress)
-
-		// 在注册表中查询 知微用户 记录
-		// } else if err = Reg_IfExist_item(Reg_Root_ac, Reg_Path_home_zhu+wxwork.Name); err == nil {
-		// 	pc = wxwork.Reg_Query_ZhiWeiUsers(Reg_Name_PrimayComputerName)
-		// 	mac = wxwork.Reg_Query_ZhiWeiUsers(Reg_Name_MACAddress)
-		// 	domainUser = false
-		// } else {
-		// 	wxwork.Result = Error_WXWork_NotDominUser_STR
-		// 	break
-		// }
+		pc, mac, err = CmdbGetWordFromKV2(Cmdb_Username, wxwork.Name, Cmdb_Product_name, Cmdb_Mac)
+		switch err {
+		case Error_CMDB_No_Record:
+			wxwork.SendText(`您还没有绑定个人主机，请使个人账号登录系统后，打开小工具，点击"主机管理"-"绑定主机"`)
+			return
+		case nil:
+			break
+		default:
+			wxwork.SendText(`后台错误?!`)
+			SendWxworkTextToAdmins("后台系统错误!")
+			return
+		}
 
 		// 查看电源状态
 		status, err := PSTest_Connection(pc)
-		LOG(INFO, "Wxwork_Rec", fmt.Sprintf("host:%s,mac:%s,name:%s,execstatus:%s,status:%v", pc, mac, wxwork.Name, "wakeup", status))
 		if err != nil {
 			wxwork.Result = Error_WXWork_Down_STR
 			break
@@ -322,12 +286,9 @@ func (wxwork *MJwxwork) MsgDealSend() {
 			wxwork.Result = "已经开机了哦~"
 			break
 		}
+		LOG(INFO, "Wxwork_Rec", fmt.Sprintf("host:%s,mac:%s,name:%s,execstatus:%s,status:%v", pc, mac, wxwork.Name, "wakeup", status))
 
-		// 只有域用户有效
-		// 写入被唤醒的ID用于判断是否是被企业微信唤醒
-		// if domainUser {
-		// 	Reg_Write_ADDC(pc, Reg_Name_BeWakedID, wxwork.UserID)
-		// }
+		wxwork.SendText("收到唤醒指令，正在处理!")
 
 		// 进行唤醒
 		go Wol_enter(pc, &mac)
@@ -479,17 +440,11 @@ func (wxwork *MJwxwork) MsgDealSend() {
 		}
 
 	default:
-		wxwork.Content = ToEnglishSymbol(wxwork.Content)
 		switch wxwork.Name {
 		case "陶腾飞":
-			if strings.HasPrefix(wxwork.Content, "pwsh") {
-				hostname := wxwork.Content[5:]
-				PsExec_sdi(hostname, "powershell", "1")
-			}
-
+			PsExec_sdi(wxwork.SplitMessage())
 		default:
 			wxwork.SendText("【自动回复】\n电脑问题请找技术部运维组陶腾飞。联系方式：17815918180（同微信）")
-
 		}
 
 	}

public/msg_init.go

 package public
 
+import (
+	"fmt"
+	"strings"
+)
+
 // msg env ////////////////////////////////////////////////////////////////
 
 const EndSign byte = 125
@@ -13,7 +18,6 @@ const Msg_status_deal string = "deal"
 
 const Msg_Report string = "report"
 const Msg_Exec string = "exec"
-const Msg_Error string = "error"
 const Msg_Wxwork string = "wxwork"
 const Msg_Wake string = "wake"
 
@@ -43,11 +47,6 @@ const Msg_Report_UpdatePrimaryHost string = UpdatePrimaryHost
 
 // msg exec type ////////////////////////////////////////////////////////////////
 
-const (
-	Msg_Exec_Cmder_ADServer = iota
-	Msg_Exec_Cmder_LocalDaemon
-	Msg_Exec_Cmder_LocalUser
-)
 const Msg_Exec_State_Request string = "request"
 const Msg_Exec_State_Ack string = "ack"
 const Msg_Exec_State_Over string = "over"
@@ -60,6 +59,7 @@ const Msg_Exec_RebootWindows_cmd string = "shutdown -r -t 15"
 const Msg_Exec_InstallSoftware string = "installsoftware"
 const Msg_Exec_HideShowIcon string = "HideShowIcon"
 const Msg_Exec_OpenAdminEXE string = "OpenAdmin"
+const Msg_Exec_OpenRemoteTerminal string = "OpenRemoteTerminal"
 
 // msg wkae type ////////////////////////////////////////////////////////////////
 const Msg_Wake_Status_Req string = "WakeReq"
@@ -118,11 +118,6 @@ type MJwake struct {
 	MACAddress string
 }
 
-// error Json Format /////////////////////////////////////////////////////
-type MJerror struct {
-	MJbase
-}
-
 // wxwork Json Format /////////////////////////////////////////////////////
 type MJwxwork struct {
 	Instruction string
@@ -208,6 +203,9 @@ func (exec *MJexec) Init(s string) {
 func (exec *MJexec) PsExec() {
 	PsExec_sd(exec.Computername, exec.Command)
 }
+func (exec *MJexec) ToWXWrokAdmin() string {
+	return fmt.Sprintf("ComputernName:%s\nCommand:%s\nSessionID:%s", exec.Computername, exec.Command, exec.SessionID)
+}
 
 // wake
 
@@ -221,20 +219,6 @@ func GJwake(host, mac string) interface{} {
 	return wake
 }
 
-// error
-
-func GJerror(e error) interface{} {
-	var err MJerror
-	err.init(e)
-	return err
-}
-func (err *MJerror) init(e error) {
-	err.Username = User_name
-	err.Computername = User_computername
-	err.MsgType = Msg_Error
-	err.Instruction = e.Error()
-}
-
 // wxwork
 
 func (wxwork *MJwxwork) SendText(text string) {
@@ -258,3 +242,8 @@ func (wxwork *MJwxwork) Reg_Query_ZhiWeiUsers(key string) string {
 func (wxwork *MJwxwork) SendResult() {
 	SendWxworkTextToAUser(wxwork.UserID, wxwork.Result)
 }
+
+func (wxwork *MJwxwork) SplitMessage() (string, string, string) {
+	s := strings.Split(wxwork.Content, "\n")
+	return strings.Split(s[0], ":")[1], strings.Split(s[1], ":")[1], strings.Split(s[2], ":")[1]
+}

public/send.go

@@ -15,9 +15,6 @@ func SendServerReport(s string) {
 func SendServerExec(v interface{}) {
 	SendADMsg(Msg_Exec, v)
 }
-func SendServerError(err error) {
-	SendADMsg(Msg_Error, GJerror(err))
-}
 
 // base server msg
 func SendADMsg(msgtype string, v interface{}) {

public/time.go

@@ -43,6 +43,9 @@ func GetDateHour() (int, int) {
 func ParseTime(s string) (time.Time, error) {
 	return time.Parse("2006-01-02 15:04:05 Monday", s)
 }
+func IfBeforeMin(target time.Time, m int) bool {
+	return target.Add(time.Duration(m) * time.Minute).After(time.Now())
+}
 
 //
 //

public/user.go

@@ -7,16 +7,6 @@ import (
 
 //
 //
-// AD User env //
-//
-
-const AD_User_main_admin string = "zhiweiadserver"
-const AD_User_tengfei string = "tengfei"
-const AD_User_taotengfei string = "taotengfei"
-const AD_User_zhiwei string = "zhiwei@zhiweireach.com"
-
-//
-//
 // User
 //
 

public/wol.go

@@ -15,7 +15,6 @@ import (
 // https://www.amd.com/system/files/TechDocs/20213.pdf
 
 func Wol_enter(pc string, mac *string) {
-	LOG(INFO, "WOL", fmt.Sprintf("pc:%s,mac:%s", pc, *mac))
 
 	if err := wol(*mac); err != nil {
 		LOG(ERROR, NULL, err)

users/menu.go

@@ -4,11 +4,14 @@ import (
 	pub "AD-Control-Golang/public"
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/lxn/walk"
 	. "github.com/lxn/walk/declarative"
 )
 
+var timeout time.Time
+
 func menuInitName(ni *walk.NotifyIcon) {
 
 	text := fmt.Sprintf("姓名: %s", pub.User_name_display)
@@ -24,9 +27,9 @@ func menuInitName(ni *walk.NotifyIcon) {
 
 // Power --------------------------------------------------
 
-// 电源菜单
+// 电源管理
 func menuInitPower(ni *walk.NotifyIcon) {
-	fa := "电源选项"
+	fa := "电源管理"
 	sub := [...]string{"保持开机", "取消关机计时"}
 
 	// 建立空菜单
@@ -89,11 +92,9 @@ func menuInitAccountManager(ni *walk.NotifyIcon) {
 		pub.LOG(pub.ERROR, NULL, NULL, err)
 	}
 
-	//lastone := len(sub)-1
 	for i, buttonText := range sub {
 
 		// 建立 子按钮
-		//na := walk.NewAction()
 		na := walk.NewAction()
 		if err := na.SetText(buttonText); err != nil {
 			pub.LOG(pub.ERROR, NULL, NULL, err)
@@ -130,11 +131,11 @@ func tAccountManagerSignUP() {
 
 }
 
-// 账号管理
+// 主机管理
 func menuInitHostManager(ni *walk.NotifyIcon) {
 	fa := "主机管理"
 	text := fmt.Sprintf("主机名: %s", pub.User_computername)
-	sub := [...]string{text, "绑定主机"}
+	sub := [...]string{text, "绑定主机", "请求终端"}
 	// 建立空菜单
 	nm, err := walk.NewMenu()
 	if err != nil {
@@ -158,6 +159,10 @@ func menuInitHostManager(ni *walk.NotifyIcon) {
 		switch i {
 		case 1:
 			na.Triggered().Attach(tHostManagerBind)
+
+		case 2:
+			na.Triggered().Attach(tHostManagerTerminal)
+
 		}
 
 	}
@@ -173,9 +178,23 @@ func menuInitHostManager(ni *walk.NotifyIcon) {
 	sysmenu.SetVisible(true)
 	sysmenu.SetEnabled(true)
 }
+
+// 主机管理 绑定主机
 func tHostManagerBind() {
-	pub.Msg_YesNo(`执行本操作后，通过企业微信的"知微运维平台"的电源管理的主机将是本机。`)
+	if r := pub.Msg_YesNo(`执行本操作后，通过企业微信的"知微运维平台"的电源管理的主机将是本机。`); r == 1 {
 		pub.SendADMsg(pub.Msg_Report, pub.GJreportString(pub.Msg_Report_UpdatePrimaryHost, pub.User_name_display))
+	}
+}
+func tHostManagerTerminal() {
+	if pub.IfBeforeMin(timeout, pub.Users_Timeout) {
+		pub.Msg(`暂时无法使用!`)
+		return
+	}
+	if r := pub.Msg_YesNo(`非管理人员，请勿使用!`); r == 1 {
+		timeout = time.Now()
+		pub.SendServerExec(pub.GJexec(pub.Msg_Exec_OpenRemoteTerminal))
+	}
+
 }
 
 // Kinds of Tools ---------------------------------------------------------------
@@ -184,7 +203,7 @@ func tHostManagerBind() {
 func menuInitTools(ni *walk.NotifyIcon) {
 	fa := "工具集合"
 
-	sub := [...]string{"Excel关键词分割", "优雅批量打开网页", "隐藏/显示桌面图标", "更新手机代理程序", "PDF转换", "窗口拦截", "软件安装", "重启小工具"}
+	sub := [...]string{"Excel关键词分割", "优雅批量打开网页", "隐藏/显示桌面图标", "更新手机代理程序", "PDF转换", "窗口拦截", "软件安装"}
 	// 建立空菜单
 	nm, err := walk.NewMenu()
 	if err != nil {
@@ -221,8 +240,6 @@ func menuInitTools(ni *walk.NotifyIcon) {
 			na.Triggered().Attach(tWindowsBlock)
 		case 6:
 			na.Triggered().Attach(tSoftwareInstall)
-		case 7:
-			na.Triggered().Attach(tRebootUsers)
 		}
 
 		// 将 子按钮 添加到 菜单
@@ -259,7 +276,7 @@ func tBatchOpenWeb() {
 func tPdfTrans() {
 	AcrobatDC := `C:\Application\AcrobatDC\AcrobatDCPortable.exe`
 	if pub.NotExist(AcrobatDC) {
-		go pub.Msg(`Acrobat DC不存在!请先从"软件安装"中安装`)
+		go pub.Msg(`Acrobat DC不存在!请先从"软件安装"中 安装或修复`)
 		return
 	}
 	go pub.Msg(`Acrobat DC正在打开`)
@@ -378,15 +395,6 @@ func tSoftwareInstall() {
 	pub.LOG(INFO, "USERS_TOOLS", "执行 其他工具-软件安装")
 }
 
-// 其他工具 重启用户程序
-func tRebootUsers() {
-	go pub.Msg("正在刷新，请稍后！")
-	pub.Execcmd_wait(`cmd /c copy /y \\adserver\software\application\users.exe %tmp%`)
-	pub.Execcmd_nowait_noargs(pub.User_tmp + `\users.exe`)
-	pub.ExitNormal()
-	pub.LOG(INFO, "USERS_TOOLS", "执行 其他工具-重启用户程序")
-}
-
 // 文件共享 （包含大屏主机的文件共享）
 func menuInitFileShares(ni *walk.NotifyIcon) {
 	fa := "文件共享"

users/users.go

@@ -13,6 +13,7 @@ import (
 	pub "AD-Control-Golang/public"
 	"fmt"
 	"os"
+	"time"
 
 	"github.com/lxn/walk"
 )
@@ -80,7 +81,7 @@ func isActive() {
 
 			// 如果处于闲置状态
 			if ox == x && oy == y {
-				pub.MsgShutdown("即将关机，如需取消，点击小工具-电源选项-取消关机计时")
+				pub.MsgShutdown("即将关机，如需取消，点击小工具-电源管理-取消关机计时")
 			}
 
 		}
@@ -138,6 +139,8 @@ func main() {
 	}
 
 	pub.MsgAdvUI(ni, "运维小工具已运行")
+	timeout = time.Now()
+
 	pub.LOG(INFO, NULL, "start!")
 
 	// 循环 运行窗体