-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.go
94 lines (84 loc) · 3.36 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package iwbemobjectsink
import (
"context"
"fmt"
"strings"
"unicode/utf16"
dcerpc "github.com/oiweiwei/go-msrpc/dcerpc"
errors "github.com/oiweiwei/go-msrpc/dcerpc/errors"
uuid "github.com/oiweiwei/go-msrpc/midl/uuid"
iunknown "github.com/oiweiwei/go-msrpc/msrpc/dcom/iunknown/v0"
ndr "github.com/oiweiwei/go-msrpc/ndr"
)
var (
_ = context.Background
_ = fmt.Errorf
_ = utf16.Encode
_ = strings.TrimPrefix
_ = ndr.ZeroString
_ = (*uuid.UUID)(nil)
_ = (*dcerpc.SyntaxID)(nil)
_ = (*errors.Error)(nil)
_ = iunknown.GoPackage
)
// IWbemObjectSink server interface.
type ObjectSinkServer interface {
// IUnknown base class.
iunknown.UnknownServer
// When the IWbemObjectSink::Indicate method is called, the apObjArray parameter MUST
// contain additional result objects as an array of an IWbemClassObject, sent by the
// client to the server. The IWbemObjectSink::Indicate method has the following syntax,
// expressed in Microsoft Interface Definition Language (MIDL).
//
// Return Values: This method MUST return an HRESULT value that MUST indicate the status
// of the method call.
//
// When the IWbemObjectSink::Indicate method is called for the first time, the server
// that implements the ObjectArray structure MUST return WBEM_S_NEW_STYLE if the execution
// of the method succeeds. If a server does not implement the ObjectArray structure,
// it MUST return WBEM_S_NO_ERROR for successful execution of the method.
//
// If the server implements the ObjectArray structure and WBEM_S_NEW_STYLE is returned
// during the first call to the IWbemObjectSink::Indicate method, the server MUST support
// subsequent calls to the IWbemObjectSink::Indicate method by using both the DCOM Remote
// Protocol marshaling and the ObjectArray structure as specified in section 2.2.14.
Indicate(context.Context, *IndicateRequest) (*IndicateResponse, error)
// When the IWbemObjectSink::SetStatus method is called, the parameter MUST contain
// the result of the operation or the progress status information.
//
// Return Values: This method MUST return an HRESULT value that MUST indicate the status
// of the method call. The server MUST return WBEM_S_NO_ERROR (specified in section
// 2.2.11) to indicate the successful completion of the method.
SetStatus(context.Context, *SetStatusRequest) (*SetStatusResponse, error)
}
func RegisterObjectSinkServer(conn dcerpc.Conn, o ObjectSinkServer, opts ...dcerpc.Option) {
conn.RegisterServer(NewObjectSinkServerHandle(o), append(opts, dcerpc.WithAbstractSyntax(ObjectSinkSyntaxV0_0))...)
}
func NewObjectSinkServerHandle(o ObjectSinkServer) dcerpc.ServerHandle {
return func(ctx context.Context, opNum int, r ndr.Reader) (dcerpc.Operation, error) {
return ObjectSinkServerHandle(ctx, o, opNum, r)
}
}
func ObjectSinkServerHandle(ctx context.Context, o ObjectSinkServer, opNum int, r ndr.Reader) (dcerpc.Operation, error) {
if opNum < 3 {
// IUnknown base method.
return iunknown.UnknownServerHandle(ctx, o, opNum, r)
}
switch opNum {
case 3: // Indicate
in := &IndicateRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.Indicate(ctx, in)
return resp.xxx_ToOp(ctx), err
case 4: // SetStatus
in := &SetStatusRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.SetStatus(ctx, in)
return resp.xxx_ToOp(ctx), err
}
return nil, nil
}