-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.go
113 lines (99 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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package ifsrmfilescreen
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"
ifsrmfilescreenbase "github.com/oiweiwei/go-msrpc/msrpc/dcom/fsrm/ifsrmfilescreenbase/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)
_ = ifsrmfilescreenbase.GoPackage
)
// IFsrmFileScreen server interface.
type FileScreenServer interface {
// IFsrmFileScreenBase base class.
ifsrmfilescreenbase.FileScreenBaseServer
// Path operation.
GetPath(context.Context, *GetPathRequest) (*GetPathResponse, error)
// SourceTemplateName operation.
GetSourceTemplateName(context.Context, *GetSourceTemplateNameRequest) (*GetSourceTemplateNameResponse, error)
// MatchesSourceTemplate operation.
GetMatchesSourceTemplate(context.Context, *GetMatchesSourceTemplateRequest) (*GetMatchesSourceTemplateResponse, error)
// UserSid operation.
GetUserSID(context.Context, *GetUserSIDRequest) (*GetUserSIDResponse, error)
// UserAccount operation.
GetUserAccount(context.Context, *GetUserAccountRequest) (*GetUserAccountResponse, error)
// ApplyTemplate operation.
ApplyTemplate(context.Context, *ApplyTemplateRequest) (*ApplyTemplateResponse, error)
}
func RegisterFileScreenServer(conn dcerpc.Conn, o FileScreenServer, opts ...dcerpc.Option) {
conn.RegisterServer(NewFileScreenServerHandle(o), append(opts, dcerpc.WithAbstractSyntax(FileScreenSyntaxV0_0))...)
}
func NewFileScreenServerHandle(o FileScreenServer) dcerpc.ServerHandle {
return func(ctx context.Context, opNum int, r ndr.Reader) (dcerpc.Operation, error) {
return FileScreenServerHandle(ctx, o, opNum, r)
}
}
func FileScreenServerHandle(ctx context.Context, o FileScreenServer, opNum int, r ndr.Reader) (dcerpc.Operation, error) {
if opNum < 18 {
// IFsrmFileScreenBase base method.
return ifsrmfilescreenbase.FileScreenBaseServerHandle(ctx, o, opNum, r)
}
switch opNum {
case 18: // Path
in := &GetPathRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.GetPath(ctx, in)
return resp.xxx_ToOp(ctx), err
case 19: // SourceTemplateName
in := &GetSourceTemplateNameRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.GetSourceTemplateName(ctx, in)
return resp.xxx_ToOp(ctx), err
case 20: // MatchesSourceTemplate
in := &GetMatchesSourceTemplateRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.GetMatchesSourceTemplate(ctx, in)
return resp.xxx_ToOp(ctx), err
case 21: // UserSid
in := &GetUserSIDRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.GetUserSID(ctx, in)
return resp.xxx_ToOp(ctx), err
case 22: // UserAccount
in := &GetUserAccountRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.GetUserAccount(ctx, in)
return resp.xxx_ToOp(ctx), err
case 23: // ApplyTemplate
in := &ApplyTemplateRequest{}
if err := in.UnmarshalNDR(ctx, r); err != nil {
return nil, err
}
resp, err := o.ApplyTemplate(ctx, in)
return resp.xxx_ToOp(ctx), err
}
return nil, nil
}