manager.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756
  1. // Copyright 2019 Yunion
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package esxi
  15. import (
  16. "bytes"
  17. "context"
  18. "crypto/tls"
  19. "fmt"
  20. "net"
  21. "net/http"
  22. "net/http/httputil"
  23. "net/url"
  24. "reflect"
  25. "strings"
  26. "time"
  27. xj "github.com/basgys/goxml2json"
  28. "github.com/fatih/color"
  29. "github.com/vmware/govmomi"
  30. "github.com/vmware/govmomi/object"
  31. "github.com/vmware/govmomi/property"
  32. "github.com/vmware/govmomi/session"
  33. "github.com/vmware/govmomi/view"
  34. "github.com/vmware/govmomi/vim25"
  35. "github.com/vmware/govmomi/vim25/methods"
  36. "github.com/vmware/govmomi/vim25/mo"
  37. "github.com/vmware/govmomi/vim25/soap"
  38. "github.com/vmware/govmomi/vim25/types"
  39. "moul.io/http2curl/v2"
  40. "yunion.io/x/jsonutils"
  41. "yunion.io/x/log"
  42. "yunion.io/x/pkg/errors"
  43. "yunion.io/x/pkg/utils"
  44. api "yunion.io/x/cloudmux/pkg/apis/compute"
  45. "yunion.io/x/cloudmux/pkg/cloudprovider"
  46. "yunion.io/x/cloudmux/pkg/multicloud"
  47. "yunion.io/x/cloudmux/pkg/multicloud/esxi/vcenter"
  48. )
  49. const (
  50. CLOUD_PROVIDER_VMWARE = api.CLOUD_PROVIDER_VMWARE
  51. )
  52. var (
  53. defaultDc mo.Datacenter
  54. defaultDcId = "esxi-default-datacenter"
  55. )
  56. var _ cloudprovider.ICloudRegion = (*SESXiClient)(nil)
  57. func init() {
  58. defaultDc.ManagedEntity.Name = "Datacenter"
  59. defaultDc.ManagedEntity.ExtensibleManagedObject.Self.Type = "Datacenter"
  60. defaultDc.ManagedEntity.ExtensibleManagedObject.Self.Value = defaultDcId
  61. }
  62. type ESXiClientConfig struct {
  63. cpcfg cloudprovider.ProviderConfig
  64. host string
  65. port int
  66. account string
  67. password string
  68. managed bool
  69. debug bool
  70. format string
  71. }
  72. func NewESXiClientConfig(host string, port int, account, password string) *ESXiClientConfig {
  73. cfg := &ESXiClientConfig{
  74. host: host,
  75. port: port,
  76. account: account,
  77. password: password,
  78. }
  79. return cfg
  80. }
  81. func (cfg *ESXiClientConfig) CloudproviderConfig(cpcfg cloudprovider.ProviderConfig) *ESXiClientConfig {
  82. cfg.cpcfg = cpcfg
  83. return cfg
  84. }
  85. func (cfg *ESXiClientConfig) Debug(debug bool) *ESXiClientConfig {
  86. cfg.debug = debug
  87. return cfg
  88. }
  89. func (cfg *ESXiClientConfig) Format(format string) *ESXiClientConfig {
  90. cfg.format = format
  91. return cfg
  92. }
  93. func (cfg *ESXiClientConfig) Managed(managed bool) *ESXiClientConfig {
  94. cfg.managed = managed
  95. return cfg
  96. }
  97. type SESXiClient struct {
  98. *ESXiClientConfig
  99. multicloud.SRegion
  100. multicloud.SRegionEipBase
  101. multicloud.SNoObjectStorageRegion
  102. multicloud.SRegionLbBase
  103. client *govmomi.Client
  104. context context.Context
  105. datacenters []*SDatacenter
  106. fakeVpc *sFakeVpc
  107. }
  108. func NewESXiClient(cfg *ESXiClientConfig) (*SESXiClient, error) {
  109. cfg.Managed(true)
  110. return NewESXiClient2(cfg)
  111. }
  112. func NewESXiClient2(cfg *ESXiClientConfig) (*SESXiClient, error) {
  113. cli := &SESXiClient{
  114. ESXiClientConfig: cfg,
  115. context: context.Background(),
  116. }
  117. err := cli.connect()
  118. if err != nil {
  119. return nil, err
  120. }
  121. cli.fakeVpc = &sFakeVpc{
  122. client: cli,
  123. }
  124. if !cli.IsVCenter() {
  125. err := cli.checkHostManagedByVCenter()
  126. if err != nil {
  127. if cfg.managed {
  128. cli.disconnect()
  129. return nil, err
  130. } else {
  131. log.Warningf("%s", err)
  132. }
  133. }
  134. }
  135. return cli, nil
  136. }
  137. func NewESXiClientFromJson(ctx context.Context, input jsonutils.JSONObject) (*SESXiClient, *vcenter.SVCenterAccessInfo, error) {
  138. accessInfo := new(vcenter.SVCenterAccessInfo)
  139. err := input.Unmarshal(accessInfo)
  140. if err != nil {
  141. return nil, nil, errors.Wrapf(err, "unmarshal SVCenterAccessInfo: %s", input)
  142. }
  143. c, err := NewESXiClientFromAccessInfo(ctx, accessInfo)
  144. return c, accessInfo, err
  145. }
  146. func NewESXiClientFromAccessInfo(ctx context.Context, accessInfo *vcenter.SVCenterAccessInfo) (*SESXiClient, error) {
  147. if len(accessInfo.VcenterId) > 0 {
  148. tmp, err := utils.DescryptAESBase64(accessInfo.VcenterId, accessInfo.Password)
  149. if err == nil {
  150. accessInfo.Password = tmp
  151. }
  152. }
  153. client, err := NewESXiClient(
  154. NewESXiClientConfig(
  155. accessInfo.Host,
  156. accessInfo.Port,
  157. accessInfo.Account,
  158. accessInfo.Password,
  159. ).Managed(true),
  160. )
  161. if err != nil {
  162. return nil, err
  163. }
  164. if ctx == nil {
  165. ctx = context.Background()
  166. }
  167. client.context = ctx
  168. return client, nil
  169. }
  170. func (cli *SESXiClient) getUrl() string {
  171. if cli.port == 443 || cli.port == 0 {
  172. return fmt.Sprintf("https://%s", cli.host)
  173. } else {
  174. return fmt.Sprintf("https://%s:%d", cli.host, cli.port)
  175. }
  176. }
  177. func (cli *SESXiClient) url() string {
  178. return fmt.Sprintf("%s/sdk", cli.getUrl())
  179. }
  180. var (
  181. red = color.New(color.FgRed, color.Bold).PrintlnFunc()
  182. green = color.New(color.FgGreen, color.Bold).PrintlnFunc()
  183. yellow = color.New(color.FgYellow, color.Bold).PrintlnFunc()
  184. cyan = color.New(color.FgHiCyan, color.Bold).PrintlnFunc()
  185. )
  186. func (cli *SESXiClient) connect() error {
  187. u, err := url.Parse(cli.url())
  188. if err != nil {
  189. return fmt.Errorf("Illegal url %s: %s", cli.url(), err)
  190. }
  191. var govmcli *govmomi.Client
  192. {
  193. insecure := true
  194. soapCli := soap.NewClient(u, insecure)
  195. httpClient := &soapCli.Client
  196. transport := &http.Transport{
  197. // 代理设置
  198. Proxy: cli.cpcfg.ProxyFunc,
  199. ExpectContinueTimeout: 5 * time.Second,
  200. TLSClientConfig: &tls.Config{InsecureSkipVerify: insecure},
  201. IdleConnTimeout: time.Minute * 1,
  202. TLSHandshakeTimeout: time.Minute * 1,
  203. DialContext: (&net.Dialer{
  204. Timeout: 30 * time.Second,
  205. KeepAlive: 30 * time.Second,
  206. }).DialContext,
  207. }
  208. httpClient.Transport = cloudprovider.GetCheckTransport(transport, func(req *http.Request) (func(resp *http.Response) error, error) {
  209. if cli.debug {
  210. dump, _ := httputil.DumpRequestOut(req, false)
  211. yellow(string(dump))
  212. // 忽略掉上传文件的请求,避免大量日志输出
  213. if req.Header.Get("Content-Type") != "application/octet-stream" {
  214. curlCmd, _ := http2curl.GetCurlCommand(req)
  215. cyan("CURL:", curlCmd, "\n")
  216. }
  217. }
  218. respCheck := func(resp *http.Response) error {
  219. if cli.debug {
  220. dump, _ := httputil.DumpResponse(resp, true)
  221. body := string(dump)
  222. if cli.format == "json" {
  223. obj, err := xj.Convert(bytes.NewReader(dump))
  224. if err == nil {
  225. body = string(obj.String())
  226. }
  227. }
  228. if resp.StatusCode < 300 {
  229. green(body)
  230. } else if resp.StatusCode < 400 {
  231. yellow(body)
  232. } else {
  233. red(body)
  234. }
  235. }
  236. return nil
  237. }
  238. return respCheck, nil
  239. })
  240. vimCli, err := vim25.NewClient(cli.context, soapCli)
  241. if err != nil {
  242. return err
  243. }
  244. govmcli = &govmomi.Client{
  245. Client: vimCli,
  246. SessionManager: session.NewManager(vimCli),
  247. }
  248. }
  249. userinfo := url.UserPassword(cli.account, cli.password)
  250. err = govmcli.Login(cli.context, userinfo)
  251. if err != nil {
  252. return err
  253. }
  254. cli.client = govmcli
  255. defaultDc.ManagedEntity.Parent = &cli.client.ServiceContent.RootFolder
  256. return nil
  257. }
  258. func (cli *SESXiClient) disconnect() error {
  259. if cli.client != nil {
  260. return cli.client.Logout(cli.context)
  261. }
  262. return nil
  263. }
  264. func (cli *SESXiClient) GetSubAccounts() ([]cloudprovider.SSubAccount, error) {
  265. err := cli.connect()
  266. if err != nil {
  267. return nil, err
  268. }
  269. subAccount := cloudprovider.SSubAccount{
  270. Id: cli.GetGlobalId(),
  271. Account: cli.account,
  272. Name: cli.cpcfg.Name,
  273. HealthStatus: api.CLOUD_PROVIDER_HEALTH_NORMAL,
  274. }
  275. return []cloudprovider.SSubAccount{subAccount}, nil
  276. }
  277. func (cli *SESXiClient) GetAccountId() string {
  278. return fmt.Sprintf("%s@%s:%d", cli.account, cli.host, cli.port)
  279. }
  280. func (cli *SESXiClient) GetI18n() cloudprovider.SModelI18nTable {
  281. table := cloudprovider.SModelI18nTable{}
  282. table["name"] = cloudprovider.NewSModelI18nEntry(cli.GetName()).CN(cli.GetName())
  283. return table
  284. }
  285. func (cli *SESXiClient) GetVersion() string {
  286. return cli.client.ServiceContent.About.Version
  287. }
  288. func (cli *SESXiClient) About() jsonutils.JSONObject {
  289. about := jsonutils.Marshal(&cli.client.ServiceContent.About)
  290. aboutDict := about.(*jsonutils.JSONDict)
  291. aboutDict.Add(jsonutils.NewString(cli.getEndpointType()), "endpoint_type")
  292. return aboutDict
  293. }
  294. func (cli *SESXiClient) getEndpointType() string {
  295. if cli.IsVCenter() {
  296. return "VCenter"
  297. } else {
  298. return "ESXi"
  299. }
  300. }
  301. func (cli *SESXiClient) GetUUID() string {
  302. about := cli.client.ServiceContent.About
  303. return about.InstanceUuid
  304. }
  305. func (cli *SESXiClient) fetchDatacentersFromHosts() error {
  306. var hosts []mo.HostSystem
  307. err := cli.scanAllMObjects(HOST_SYSTEM_PROPS, &hosts)
  308. if err != nil {
  309. return errors.Wrap(err, "cli.scanAllMObjects host")
  310. }
  311. dcList := make([]*SDatacenter, 0)
  312. for i := 0; i < len(hosts); i += 1 {
  313. me := newManagedObject(cli, &hosts[i], nil)
  314. dcme := me.findInParents("Datacenter")
  315. if dcme == nil {
  316. return cloudprovider.ErrNotFound
  317. }
  318. _, err := findDatacenterByMoId(dcList, dcme.Self.Value)
  319. if err == nil {
  320. continue
  321. }
  322. var moDc mo.Datacenter
  323. err = cli.reference2Object(dcme.Self, DATACENTER_PROPS, &moDc)
  324. if err != nil {
  325. return errors.Wrap(err, "cli.reference2Object")
  326. }
  327. dc, err := cli.newDatacenterFromMo(&moDc)
  328. if err != nil {
  329. return errors.Wrap(err, "cli.newDatacenterFromMo")
  330. }
  331. dcList = append(dcList, dc)
  332. }
  333. cli.datacenters = dcList
  334. return nil
  335. }
  336. func (cli *SESXiClient) fetchFakeDatacenter() error {
  337. dc, err := cli.newDatacenterFromMo(&defaultDc)
  338. if err != nil {
  339. return errors.Wrap(err, "newDatacenterFromMo")
  340. }
  341. cli.datacenters = []*SDatacenter{dc}
  342. return nil
  343. }
  344. func (cli *SESXiClient) fetchDatacenters() error {
  345. var dcs []mo.Datacenter
  346. err := cli.scanAllMObjects(DATACENTER_PROPS, &dcs)
  347. if err != nil {
  348. // log.Debugf("cli.scanAllMObjects datacenter fail %s, try cli.fetchDatacentersFromHosts", err)
  349. // err := cli.fetchDatacentersFromHosts()
  350. // if err != nil {
  351. log.Debugf("cli.fetchDatacentersFromHosts fail %s, try cli.fetchFakeDatacenter", err)
  352. return cli.fetchFakeDatacenter()
  353. // }
  354. }
  355. cli.datacenters = make([]*SDatacenter, len(dcs))
  356. for i := 0; i < len(dcs); i += 1 {
  357. dc, err := cli.newDatacenterFromMo(&dcs[i])
  358. if err != nil {
  359. return errors.Wrap(err, "cli.newDatacenterFromMo")
  360. }
  361. cli.datacenters[i] = dc
  362. }
  363. return nil
  364. }
  365. func (cli *SESXiClient) newDatacenterFromMo(mo *mo.Datacenter) (*SDatacenter, error) {
  366. dc := newDatacenter(cli, mo)
  367. err := dc.scanHosts()
  368. if err != nil {
  369. return nil, errors.Wrap(err, "dc.scanHosts")
  370. }
  371. // err = dc.scanDatastores()
  372. // if err != nil {
  373. // return nil, errors.Wrap(err, "dc.scanDatastores")
  374. // }
  375. return dc, nil
  376. }
  377. func (cli *SESXiClient) scanAllMObjects(props []string, dst interface{}) error {
  378. return cli.scanMObjects(cli.client.ServiceContent.RootFolder, props, dst)
  379. }
  380. func (cli *SESXiClient) SearchVM(id string) (*SVirtualMachine, error) {
  381. filter := property.Match{}
  382. filter["summary.config.uuid"] = id
  383. var movms []mo.VirtualMachine
  384. err := cli.scanMObjectsWithFilter(cli.client.ServiceContent.RootFolder, VIRTUAL_MACHINE_PROPS, &movms, filter)
  385. if err != nil {
  386. return nil, err
  387. }
  388. if len(movms) == 0 {
  389. return nil, errors.ErrNotFound
  390. }
  391. vm := NewVirtualMachine(cli, &movms[0], nil)
  392. dc, err := vm.fetchDatacenter()
  393. if err != nil {
  394. return nil, errors.Wrap(err, "fetchDatacenter")
  395. }
  396. vm.datacenter = dc
  397. return vm, nil
  398. }
  399. func (cli *SESXiClient) SearchTemplateVM(id string) (*SVirtualMachine, error) {
  400. filter := property.Match{}
  401. uuid := toTemplateUuid(id)
  402. filter["summary.config.uuid"] = uuid
  403. var movms []mo.VirtualMachine
  404. err := cli.scanMObjectsWithFilter(cli.client.ServiceContent.RootFolder, VIRTUAL_MACHINE_PROPS, &movms, filter)
  405. if err != nil {
  406. return nil, errors.Wrapf(err, "scanMObjectsWithFilter")
  407. }
  408. if len(movms) == 0 {
  409. return nil, errors.Wrapf(errors.ErrNotFound, "empty templates")
  410. }
  411. vm := NewVirtualMachine(cli, &movms[0], nil)
  412. if !vm.IsTemplate() {
  413. return nil, errors.Wrapf(errors.ErrNotFound, "%s is not template", vm.GetName())
  414. }
  415. dc, err := vm.fetchDatacenter()
  416. if err != nil {
  417. return nil, errors.Wrap(err, "fetchDatacenter")
  418. }
  419. vm.datacenter = dc
  420. return vm, nil
  421. }
  422. func (cli *SESXiClient) scanMObjectsWithFilter(folder types.ManagedObjectReference, props []string, dst interface{}, filter property.Match) error {
  423. dstValue := reflect.Indirect(reflect.ValueOf(dst))
  424. dstType := dstValue.Type()
  425. dstEleType := dstType.Elem()
  426. resType := dstEleType.Name()
  427. m := view.NewManager(cli.client.Client)
  428. v, err := m.CreateContainerView(cli.context, folder, []string{resType}, true)
  429. if err != nil {
  430. return errors.Wrapf(err, "m.CreateContainerView %s", resType)
  431. }
  432. defer v.Destroy(cli.context)
  433. err = v.RetrieveWithFilter(cli.context, []string{resType}, props, dst, filter)
  434. if err != nil {
  435. // hack
  436. if strings.Contains(err.Error(), "object references is empty") {
  437. return nil
  438. }
  439. return errors.Wrapf(err, "v.RetrieveWithFilter %s", resType)
  440. }
  441. return nil
  442. }
  443. func fixProps(props []string) []string {
  444. for _, key := range []string{
  445. "name", "parent",
  446. } {
  447. if !utils.IsInArray(key, props) {
  448. props = append(props, key)
  449. }
  450. }
  451. return props
  452. }
  453. func (cli *SESXiClient) scanMObjects(folder types.ManagedObjectReference, props []string, dst interface{}) error {
  454. dstValue := reflect.Indirect(reflect.ValueOf(dst))
  455. dstType := dstValue.Type()
  456. dstEleType := dstType.Elem()
  457. resType := dstEleType.Name()
  458. m := view.NewManager(cli.client.Client)
  459. v, err := m.CreateContainerView(cli.context, folder, []string{resType}, true)
  460. if err != nil {
  461. return errors.Wrapf(err, "m.CreateContainerView %s", resType)
  462. }
  463. defer v.Destroy(cli.context)
  464. err = v.Retrieve(cli.context, []string{resType}, fixProps(props), dst)
  465. if err != nil {
  466. return errors.Wrapf(err, "v.Retrieve %s", resType)
  467. }
  468. return nil
  469. }
  470. func (cli *SESXiClient) references2Objects(refs []types.ManagedObjectReference, props []string, dst interface{}) error {
  471. pc := property.DefaultCollector(cli.client.Client)
  472. err := pc.Retrieve(cli.context, refs, fixProps(props), dst)
  473. if err != nil {
  474. return errors.Wrapf(err, "Retrieve")
  475. }
  476. return nil
  477. }
  478. func (cli *SESXiClient) reference2Object(ref types.ManagedObjectReference, props []string, dst interface{}) error {
  479. pc := property.DefaultCollector(cli.client.Client)
  480. err := pc.RetrieveOne(cli.context, ref, fixProps(props), dst)
  481. if err != nil {
  482. return errors.Wrap(err, "pc.RetrieveOne")
  483. }
  484. return nil
  485. }
  486. func (cli *SESXiClient) GetDatacenters() ([]*SDatacenter, error) {
  487. if cli.datacenters == nil {
  488. err := cli.fetchDatacenters()
  489. if err != nil {
  490. return nil, err
  491. }
  492. }
  493. return cli.datacenters, nil
  494. }
  495. func (cli *SESXiClient) FindDatacenterByMoId(dcId string) (*SDatacenter, error) {
  496. dcs, err := cli.GetDatacenters()
  497. if err != nil {
  498. return nil, err
  499. }
  500. return findDatacenterByMoId(dcs, dcId)
  501. }
  502. func findDatacenterByMoId(dcs []*SDatacenter, dcId string) (*SDatacenter, error) {
  503. for i := 0; i < len(dcs); i += 1 {
  504. if dcs[i].GetId() == dcId {
  505. return dcs[i], nil
  506. }
  507. // defaultDcId means no premision to get datacenter, so return fake dc
  508. if dcs[i].GetId() == defaultDcId {
  509. return dcs[i], nil
  510. }
  511. }
  512. return nil, cloudprovider.ErrNotFound
  513. }
  514. func (cli *SESXiClient) FindHostByMoId(moId string) (cloudprovider.ICloudHost, error) {
  515. dcs, err := cli.GetDatacenters()
  516. if err != nil {
  517. return nil, err
  518. }
  519. for i := 0; i < len(dcs); i += 1 {
  520. ihost, err := dcs[i].GetIHostByMoId(moId)
  521. if err == nil {
  522. return ihost, nil
  523. }
  524. }
  525. return nil, cloudprovider.ErrNotFound
  526. }
  527. func (cli *SESXiClient) getPrivateId(idStr string) string {
  528. if len(cli.cpcfg.Id) > 0 && strings.HasPrefix(idStr, cli.cpcfg.Id) {
  529. idStr = idStr[len(cli.cpcfg.Id)+1:]
  530. }
  531. return idStr
  532. }
  533. func (cli *SESXiClient) checkHostManagedByVCenter() error {
  534. host, err := cli.FindHostByIp(cli.host)
  535. if err != nil {
  536. if errors.Cause(err) == errors.ErrNotFound {
  537. // host might be behind a NAT
  538. return nil
  539. }
  540. return errors.Wrap(err, "cli.FindHostByIp")
  541. }
  542. if host.IsManagedByVCenter() {
  543. return fmt.Errorf("ESXi host is managed by vcenter %s, please connect to vcenter instead for full management functions!", host.GetManagementServerIp())
  544. }
  545. return nil
  546. }
  547. func (cli *SESXiClient) IsHostIpExists(hostIp string) (bool, error) {
  548. searchIndex := object.NewSearchIndex(cli.client.Client)
  549. hostRef, err := searchIndex.FindByIp(cli.context, nil, cli.getPrivateId(hostIp), false)
  550. if err != nil {
  551. return false, errors.Wrapf(err, "FindByIp %s", hostIp)
  552. }
  553. if hostRef == nil {
  554. return false, nil
  555. }
  556. return true, nil
  557. }
  558. func (cli *SESXiClient) FindHostByIp(hostIp string) (*SHost, error) {
  559. searchIndex := object.NewSearchIndex(cli.client.Client)
  560. hostRef, err := searchIndex.FindByIp(cli.context, nil, cli.getPrivateId(hostIp), false)
  561. if err != nil {
  562. return nil, errors.Wrapf(err, "FindByIp %s", hostIp)
  563. }
  564. if hostRef == nil {
  565. return nil, errors.Wrapf(errors.ErrNotFound, "cannot find %s", cli.getPrivateId(hostIp))
  566. }
  567. var host mo.HostSystem
  568. err = cli.reference2Object(hostRef.Reference(), HOST_SYSTEM_PROPS, &host)
  569. if err != nil {
  570. return nil, errors.Wrapf(err, "reference2Object %s", hostIp)
  571. }
  572. h := NewHost(cli, &host, nil)
  573. if h == nil {
  574. return nil, errors.Wrap(errors.ErrInvalidStatus, "empty host mo")
  575. }
  576. return h, nil
  577. }
  578. func (cli *SESXiClient) acquireCloneTicket() (string, error) {
  579. manager := session.NewManager(cli.client.Client)
  580. return manager.AcquireCloneTicket(cli.context)
  581. }
  582. func (cli *SESXiClient) IsVCenter() bool {
  583. return cli.client.Client.IsVC()
  584. }
  585. func (cli *SESXiClient) IsValid() bool {
  586. return cli.client.Client.Valid()
  587. }
  588. func (cli *SESXiClient) GetCapabilities() []string {
  589. caps := []string{
  590. cloudprovider.CLOUD_CAPABILITY_PROJECT,
  591. cloudprovider.CLOUD_CAPABILITY_COMPUTE,
  592. cloudprovider.CLOUD_CAPABILITY_NETWORK,
  593. // cloudprovider.CLOUD_CAPABILITY_LOADBALANCER,
  594. // cloudprovider.CLOUD_CAPABILITY_OBJECTSTORE,
  595. // cloudprovider.CLOUD_CAPABILITY_RDS,
  596. // cloudprovider.CLOUD_CAPABILITY_CACHE,
  597. // cloudprovider.CLOUD_CAPABILITY_EVENT,
  598. }
  599. return caps
  600. }
  601. func (cli *SESXiClient) FindVMByPrivateID(idstr string) (*SVirtualMachine, error) {
  602. searchIndex := object.NewSearchIndex(cli.client.Client)
  603. instanceUuid := true
  604. vmRef, err := searchIndex.FindByUuid(cli.context, nil, idstr, true, &instanceUuid)
  605. if err != nil {
  606. return nil, errors.Wrap(err, "searchIndex.FindByUuid fail")
  607. }
  608. if vmRef == nil {
  609. return nil, fmt.Errorf("cannot find %s", idstr)
  610. }
  611. var vm mo.VirtualMachine
  612. err = cli.reference2Object(vmRef.Reference(), VIRTUAL_MACHINE_PROPS, &vm)
  613. if err != nil {
  614. return nil, errors.Wrap(err, "reference2Object fail")
  615. }
  616. ret := NewVirtualMachine(cli, &vm, nil)
  617. if ret == nil {
  618. return nil, errors.Error("invalid vm")
  619. }
  620. return ret, nil
  621. }
  622. func (cli *SESXiClient) DoExtendDiskOnline(_vm *SVirtualMachine, _disk *SVirtualDisk, newSizeMb int64) error {
  623. disk := _disk.getVirtualDisk()
  624. disk.CapacityInKB = newSizeMb * 1024
  625. devSepc := types.VirtualDeviceConfigSpec{Operation: types.VirtualDeviceConfigSpecOperationEdit, Device: disk}
  626. spec := types.VirtualMachineConfigSpec{DeviceChange: []types.BaseVirtualDeviceConfigSpec{&devSepc}}
  627. vm := object.NewVirtualMachine(cli.client.Client, _vm.getVirtualMachine().Reference())
  628. task, err := vm.Reconfigure(cli.context, spec)
  629. if err != nil {
  630. return errors.Wrapf(err, "vm reconfigure failed")
  631. }
  632. return task.Wait(cli.context)
  633. }
  634. func (cli *SESXiClient) ExtendDisk(url string, newSizeMb int64) error {
  635. param := types.ExtendVirtualDisk_Task{
  636. This: *cli.client.Client.ServiceContent.VirtualDiskManager,
  637. Name: url,
  638. NewCapacityKb: newSizeMb * 1024,
  639. }
  640. response, err := methods.ExtendVirtualDisk_Task(cli.context, cli.client, &param)
  641. if err != nil {
  642. return errors.Wrapf(err, "extend virtualdisk task failed")
  643. }
  644. log.Debugf("extend virtual disk task response: %s", response.Returnval.String())
  645. return nil
  646. }
  647. func (cli *SESXiClient) CopyDisk(ctx context.Context, src, dst string, isForce bool) error {
  648. dm := object.NewVirtualDiskManager(cli.client.Client)
  649. task, err := dm.CopyVirtualDisk(ctx, src, nil, dst, nil, nil, isForce)
  650. if err != nil {
  651. return errors.Wrap(err, "CopyVirtualDisk")
  652. }
  653. return task.Wait(ctx)
  654. }
  655. func (cli *SESXiClient) MoveDisk(ctx context.Context, src, dst string, isForce bool) error {
  656. dm := object.NewVirtualDiskManager(cli.client.Client)
  657. task, err := dm.MoveVirtualDisk(ctx, src, nil, dst, nil, isForce)
  658. if err != nil {
  659. return errors.Wrap(err, "MoveVirtualDisk")
  660. }
  661. return task.Wait(ctx)
  662. }