|
- // Copyright 2012 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
-
- // +build windows
-
- // Package mgr can be used to manage Windows service programs.
- // It can be used to install and remove them. It can also start,
- // stop and pause them. The package can query / change current
- // service state and config parameters.
- //
- package mgr
-
- import (
- "syscall"
- "unicode/utf16"
-
- "golang.org/x/sys/windows"
- )
-
- // Mgr is used to manage Windows service.
- type Mgr struct {
- Handle windows.Handle
- }
-
- // Connect establishes a connection to the service control manager.
- func Connect() (*Mgr, error) {
- return ConnectRemote("")
- }
-
- // ConnectRemote establishes a connection to the
- // service control manager on computer named host.
- func ConnectRemote(host string) (*Mgr, error) {
- var s *uint16
- if host != "" {
- s = syscall.StringToUTF16Ptr(host)
- }
- h, err := windows.OpenSCManager(s, nil, windows.SC_MANAGER_ALL_ACCESS)
- if err != nil {
- return nil, err
- }
- return &Mgr{Handle: h}, nil
- }
-
- // Disconnect closes connection to the service control manager m.
- func (m *Mgr) Disconnect() error {
- return windows.CloseServiceHandle(m.Handle)
- }
-
- func toPtr(s string) *uint16 {
- if len(s) == 0 {
- return nil
- }
- return syscall.StringToUTF16Ptr(s)
- }
-
- // toStringBlock terminates strings in ss with 0, and then
- // concatenates them together. It also adds extra 0 at the end.
- func toStringBlock(ss []string) *uint16 {
- if len(ss) == 0 {
- return nil
- }
- t := ""
- for _, s := range ss {
- if s != "" {
- t += s + "\x00"
- }
- }
- if t == "" {
- return nil
- }
- t += "\x00"
- return &utf16.Encode([]rune(t))[0]
- }
-
- // CreateService installs new service name on the system.
- // The service will be executed by running exepath binary.
- // Use config c to specify service parameters.
- // If service StartType is set to StartAutomatic,
- // args will be passed to svc.Handle.Execute.
- func (m *Mgr) CreateService(name, exepath string, c Config, args ...string) (*Service, error) {
- if c.StartType == 0 {
- c.StartType = StartManual
- }
- if c.ErrorControl == 0 {
- c.ErrorControl = ErrorNormal
- }
- if c.ServiceType == 0 {
- c.ServiceType = windows.SERVICE_WIN32_OWN_PROCESS
- }
- s := syscall.EscapeArg(exepath)
- for _, v := range args {
- s += " " + syscall.EscapeArg(v)
- }
- h, err := windows.CreateService(m.Handle, toPtr(name), toPtr(c.DisplayName),
- windows.SERVICE_ALL_ACCESS, c.ServiceType,
- c.StartType, c.ErrorControl, toPtr(s), toPtr(c.LoadOrderGroup),
- nil, toStringBlock(c.Dependencies), toPtr(c.ServiceStartName), toPtr(c.Password))
- if err != nil {
- return nil, err
- }
- if c.Description != "" {
- err = updateDescription(h, c.Description)
- if err != nil {
- return nil, err
- }
- }
- return &Service{Name: name, Handle: h}, nil
- }
-
- // OpenService retrieves access to service name, so it can
- // be interrogated and controlled.
- func (m *Mgr) OpenService(name string) (*Service, error) {
- h, err := windows.OpenService(m.Handle, syscall.StringToUTF16Ptr(name), windows.SERVICE_ALL_ACCESS)
- if err != nil {
- return nil, err
- }
- return &Service{Name: name, Handle: h}, nil
- }
|