278 lines
7.0 KiB
Go
278 lines
7.0 KiB
Go
/*
|
|
Copyright (c) 2017-2018 VMware, Inc. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package simulator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/session"
|
|
"github.com/vmware/govmomi/vim25/methods"
|
|
"github.com/vmware/govmomi/vim25/mo"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
type SessionManager struct {
|
|
mo.SessionManager
|
|
|
|
ServiceHostName string
|
|
|
|
sessions map[string]Session
|
|
}
|
|
|
|
func NewSessionManager(ref types.ManagedObjectReference) object.Reference {
|
|
s := &SessionManager{
|
|
sessions: make(map[string]Session),
|
|
}
|
|
s.Self = ref
|
|
return s
|
|
}
|
|
|
|
func (s *SessionManager) Login(ctx *Context, login *types.Login) soap.HasFault {
|
|
body := &methods.LoginBody{}
|
|
|
|
if login.Locale == "" {
|
|
login.Locale = session.Locale
|
|
}
|
|
|
|
if login.UserName == "" || login.Password == "" || ctx.Session != nil {
|
|
body.Fault_ = invalidLogin
|
|
} else {
|
|
session := Session{
|
|
UserSession: types.UserSession{
|
|
Key: uuid.New().String(),
|
|
UserName: login.UserName,
|
|
FullName: login.UserName,
|
|
LoginTime: time.Now(),
|
|
LastActiveTime: time.Now(),
|
|
Locale: login.Locale,
|
|
MessageLocale: login.Locale,
|
|
},
|
|
Registry: NewRegistry(),
|
|
}
|
|
|
|
ctx.SetSession(session, true)
|
|
|
|
body.Res = &types.LoginResponse{
|
|
Returnval: session.UserSession,
|
|
}
|
|
}
|
|
|
|
return body
|
|
}
|
|
|
|
func (s *SessionManager) Logout(ctx *Context, _ *types.Logout) soap.HasFault {
|
|
session := ctx.Session
|
|
delete(s.sessions, session.Key)
|
|
|
|
ctx.postEvent(&types.UserLogoutSessionEvent{
|
|
IpAddress: session.IpAddress,
|
|
UserAgent: session.UserAgent,
|
|
SessionId: session.Key,
|
|
LoginTime: &session.LoginTime,
|
|
})
|
|
|
|
return &methods.LogoutBody{Res: new(types.LogoutResponse)}
|
|
}
|
|
|
|
func (s *SessionManager) TerminateSession(ctx *Context, req *types.TerminateSession) soap.HasFault {
|
|
body := new(methods.TerminateSessionBody)
|
|
|
|
for _, id := range req.SessionId {
|
|
if id == ctx.Session.Key {
|
|
body.Fault_ = Fault("", new(types.InvalidArgument))
|
|
return body
|
|
}
|
|
delete(s.sessions, id)
|
|
}
|
|
|
|
body.Res = new(types.TerminateSessionResponse)
|
|
return body
|
|
}
|
|
|
|
func (s *SessionManager) AcquireCloneTicket(ctx *Context, _ *types.AcquireCloneTicket) soap.HasFault {
|
|
session := *ctx.Session
|
|
session.Key = uuid.New().String()
|
|
s.sessions[session.Key] = session
|
|
|
|
return &methods.AcquireCloneTicketBody{
|
|
Res: &types.AcquireCloneTicketResponse{
|
|
Returnval: session.Key,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *SessionManager) CloneSession(ctx *Context, ticket *types.CloneSession) soap.HasFault {
|
|
body := new(methods.CloneSessionBody)
|
|
|
|
session, exists := s.sessions[ticket.CloneTicket]
|
|
|
|
if exists {
|
|
delete(s.sessions, ticket.CloneTicket) // A clone ticket can only be used once
|
|
session.Key = uuid.New().String()
|
|
ctx.SetSession(session, true)
|
|
|
|
body.Res = &types.CloneSessionResponse{
|
|
Returnval: session.UserSession,
|
|
}
|
|
} else {
|
|
body.Fault_ = invalidLogin
|
|
}
|
|
|
|
return body
|
|
}
|
|
|
|
func (s *SessionManager) AcquireGenericServiceTicket(ticket *types.AcquireGenericServiceTicket) soap.HasFault {
|
|
return &methods.AcquireGenericServiceTicketBody{
|
|
Res: &types.AcquireGenericServiceTicketResponse{
|
|
Returnval: types.SessionManagerGenericServiceTicket{
|
|
Id: uuid.New().String(),
|
|
HostName: s.ServiceHostName,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// internalContext is the session for use by the in-memory client (Service.RoundTrip)
|
|
var internalContext = &Context{
|
|
Context: context.Background(),
|
|
Session: &Session{
|
|
UserSession: types.UserSession{
|
|
Key: uuid.New().String(),
|
|
},
|
|
Registry: NewRegistry(),
|
|
},
|
|
}
|
|
|
|
var invalidLogin = Fault("Login failure", new(types.InvalidLogin))
|
|
|
|
// Context provides per-request Session management.
|
|
type Context struct {
|
|
req *http.Request
|
|
res http.ResponseWriter
|
|
m *SessionManager
|
|
|
|
context.Context
|
|
Session *Session
|
|
Caller *types.ManagedObjectReference
|
|
}
|
|
|
|
// mapSession maps an HTTP cookie to a Session.
|
|
func (c *Context) mapSession() {
|
|
if cookie, err := c.req.Cookie(soap.SessionCookieName); err == nil {
|
|
if val, ok := c.m.sessions[cookie.Value]; ok {
|
|
c.SetSession(val, false)
|
|
}
|
|
}
|
|
}
|
|
|
|
// SetSession should be called after successful authentication.
|
|
func (c *Context) SetSession(session Session, login bool) {
|
|
session.UserAgent = c.req.UserAgent()
|
|
session.IpAddress = strings.Split(c.req.RemoteAddr, ":")[0]
|
|
session.LastActiveTime = time.Now()
|
|
|
|
c.m.sessions[session.Key] = session
|
|
c.Session = &session
|
|
|
|
if login {
|
|
http.SetCookie(c.res, &http.Cookie{
|
|
Name: soap.SessionCookieName,
|
|
Value: session.Key,
|
|
})
|
|
|
|
c.postEvent(&types.UserLoginSessionEvent{
|
|
SessionId: session.Key,
|
|
IpAddress: session.IpAddress,
|
|
UserAgent: session.UserAgent,
|
|
Locale: session.Locale,
|
|
})
|
|
}
|
|
}
|
|
|
|
// WithLock holds a lock for the given object while then given function is run.
|
|
func (c *Context) WithLock(obj mo.Reference, f func()) {
|
|
if c.Caller != nil && *c.Caller == obj.Reference() {
|
|
// Internal method invocation, obj is already locked
|
|
f()
|
|
return
|
|
}
|
|
Map.WithLock(obj, f)
|
|
}
|
|
|
|
// postEvent wraps EventManager.PostEvent for internal use, with a lock on the EventManager.
|
|
func (c *Context) postEvent(events ...types.BaseEvent) {
|
|
m := Map.EventManager()
|
|
c.WithLock(m, func() {
|
|
for _, event := range events {
|
|
m.PostEvent(c, &types.PostEvent{EventToPost: event})
|
|
}
|
|
})
|
|
}
|
|
|
|
// Session combines a UserSession and a Registry for per-session managed objects.
|
|
type Session struct {
|
|
types.UserSession
|
|
*Registry
|
|
}
|
|
|
|
// Put wraps Registry.Put, setting the moref value to include the session key.
|
|
func (s *Session) Put(item mo.Reference) mo.Reference {
|
|
ref := item.Reference()
|
|
if ref.Value == "" {
|
|
ref.Value = fmt.Sprintf("session[%s]%s", s.Key, uuid.New())
|
|
}
|
|
s.Registry.setReference(item, ref)
|
|
return s.Registry.Put(item)
|
|
}
|
|
|
|
// Get wraps Registry.Get, session-izing singleton objects such as SessionManager and the root PropertyCollector.
|
|
func (s *Session) Get(ref types.ManagedObjectReference) mo.Reference {
|
|
obj := s.Registry.Get(ref)
|
|
if obj != nil {
|
|
return obj
|
|
}
|
|
|
|
// Return a session "view" of certain singleton objects
|
|
switch ref.Type {
|
|
case "SessionManager":
|
|
// Clone SessionManager so the PropertyCollector can properly report CurrentSession
|
|
m := *Map.SessionManager()
|
|
m.CurrentSession = &s.UserSession
|
|
|
|
// TODO: we could maintain SessionList as part of the SessionManager singleton
|
|
for _, session := range m.sessions {
|
|
m.SessionList = append(m.SessionList, session.UserSession)
|
|
}
|
|
|
|
return &m
|
|
case "PropertyCollector":
|
|
if ref == Map.content().PropertyCollector {
|
|
return s.Put(NewPropertyCollector(ref))
|
|
}
|
|
}
|
|
|
|
return Map.Get(ref)
|
|
}
|