install.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. // Copyright 2012 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. //go:build windows
  5. // +build windows
  6. package eventlog
  7. import (
  8. "errors"
  9. "golang.org/x/sys/windows"
  10. "golang.org/x/sys/windows/registry"
  11. )
  12. const (
  13. // Log levels.
  14. Info = windows.EVENTLOG_INFORMATION_TYPE
  15. Warning = windows.EVENTLOG_WARNING_TYPE
  16. Error = windows.EVENTLOG_ERROR_TYPE
  17. )
  18. const addKeyName = `SYSTEM\CurrentControlSet\Services\EventLog\Application`
  19. // Install modifies PC registry to allow logging with an event source src.
  20. // It adds all required keys and values to the event log registry key.
  21. // Install uses msgFile as the event message file. If useExpandKey is true,
  22. // the event message file is installed as REG_EXPAND_SZ value,
  23. // otherwise as REG_SZ. Use bitwise of log.Error, log.Warning and
  24. // log.Info to specify events supported by the new event source.
  25. func Install(src, msgFile string, useExpandKey bool, eventsSupported uint32) error {
  26. appkey, err := registry.OpenKey(registry.LOCAL_MACHINE, addKeyName, registry.CREATE_SUB_KEY)
  27. if err != nil {
  28. return err
  29. }
  30. defer appkey.Close()
  31. sk, alreadyExist, err := registry.CreateKey(appkey, src, registry.SET_VALUE)
  32. if err != nil {
  33. return err
  34. }
  35. defer sk.Close()
  36. if alreadyExist {
  37. return errors.New(addKeyName + `\` + src + " registry key already exists")
  38. }
  39. err = sk.SetDWordValue("CustomSource", 1)
  40. if err != nil {
  41. return err
  42. }
  43. if useExpandKey {
  44. err = sk.SetExpandStringValue("EventMessageFile", msgFile)
  45. } else {
  46. err = sk.SetStringValue("EventMessageFile", msgFile)
  47. }
  48. if err != nil {
  49. return err
  50. }
  51. err = sk.SetDWordValue("TypesSupported", eventsSupported)
  52. if err != nil {
  53. return err
  54. }
  55. return nil
  56. }
  57. // InstallAsEventCreate is the same as Install, but uses
  58. // %SystemRoot%\System32\EventCreate.exe as the event message file.
  59. func InstallAsEventCreate(src string, eventsSupported uint32) error {
  60. return Install(src, "%SystemRoot%\\System32\\EventCreate.exe", true, eventsSupported)
  61. }
  62. // Remove deletes all registry elements installed by the correspondent Install.
  63. func Remove(src string) error {
  64. appkey, err := registry.OpenKey(registry.LOCAL_MACHINE, addKeyName, registry.SET_VALUE)
  65. if err != nil {
  66. return err
  67. }
  68. defer appkey.Close()
  69. return registry.DeleteKey(appkey, src)
  70. }