mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-09-24 16:03:43 +08:00
332 lines
8.1 KiB
Go
332 lines
8.1 KiB
Go
// Copyright 2019 Yunion
|
|
//
|
|
// 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 cloudprovider
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"strings"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
"yunion.io/x/pkg/errors"
|
|
"yunion.io/x/pkg/util/osprofile"
|
|
|
|
"yunion.io/x/onecloud/pkg/util/ansible"
|
|
"yunion.io/x/onecloud/pkg/util/billing"
|
|
"yunion.io/x/onecloud/pkg/util/cloudinit"
|
|
"yunion.io/x/onecloud/pkg/util/seclib2"
|
|
)
|
|
|
|
type TOsType string
|
|
|
|
var (
|
|
OsTypeLinux = TOsType(osprofile.OS_TYPE_LINUX)
|
|
OsTypeWindows = TOsType(osprofile.OS_TYPE_WINDOWS)
|
|
)
|
|
|
|
type SDistDefaultAccount struct {
|
|
// 操作系统发行版
|
|
OsDistribution string
|
|
// 默认用户名
|
|
DefaultAccount string
|
|
// 是否可更改
|
|
Changeable bool
|
|
}
|
|
|
|
type SOsDefaultAccount struct {
|
|
// 默认用户名
|
|
DefaultAccount string
|
|
// 是否可更改用户名
|
|
Changeable bool
|
|
// 禁止使用的账号
|
|
DisabledAccounts []string
|
|
// 各操作系统发行版的默认用户名信息
|
|
DistAccounts []SDistDefaultAccount
|
|
}
|
|
|
|
type SDefaultAccount struct {
|
|
Linux SOsDefaultAccount
|
|
Windows SOsDefaultAccount
|
|
}
|
|
|
|
type StorageInfo struct {
|
|
StorageType string
|
|
MaxSizeGb int
|
|
MinSizeGb int
|
|
Resizable bool
|
|
StepSizeGb int
|
|
}
|
|
|
|
type Storage struct {
|
|
DataDisk []StorageInfo
|
|
SysDisk []StorageInfo
|
|
}
|
|
|
|
type SInstanceCapability struct {
|
|
Provider string
|
|
Hypervisor string
|
|
Storages Storage
|
|
|
|
DefaultAccount SDefaultAccount
|
|
}
|
|
|
|
type SDiskInfo struct {
|
|
StorageExternalId string
|
|
StorageType string
|
|
SizeGB int
|
|
Name string
|
|
}
|
|
|
|
type GuestDiskCreateOptions struct {
|
|
SizeMb int
|
|
UUID string
|
|
Driver string
|
|
StorageId string
|
|
}
|
|
|
|
const (
|
|
CLOUD_SHELL = "cloud-shell"
|
|
CLOUD_SHELL_WITHOUT_ENCRYPT = "cloud-shell-without-encrypt"
|
|
CLOUD_CONFIG = "cloud-config"
|
|
CLOUD_POWER_SHELL = "powershell"
|
|
CLOUD_EC2 = "ec2"
|
|
)
|
|
|
|
type SPublicIpInfo struct {
|
|
PublicIpBw int
|
|
PublicIpChargeType TElasticipChargeType
|
|
}
|
|
|
|
type ServerStopOptions struct {
|
|
IsForce bool
|
|
StopCharging bool
|
|
}
|
|
|
|
type SManagedVMCreateConfig struct {
|
|
Name string
|
|
NameEn string
|
|
Hostname string
|
|
ExternalImageId string
|
|
ImageType string
|
|
OsType string
|
|
OsDistribution string
|
|
OsVersion string
|
|
InstanceType string // InstanceType 不为空时，直接采用InstanceType创建机器。
|
|
Cpu int
|
|
MemoryMB int
|
|
ExternalNetworkId string
|
|
ExternalVpcId string
|
|
IpAddr string
|
|
Description string
|
|
SysDisk SDiskInfo
|
|
DataDisks []SDiskInfo
|
|
PublicKey string
|
|
ExternalSecgroupId string
|
|
ExternalSecgroupIds []string
|
|
Account string
|
|
Password string
|
|
UserData string
|
|
ProjectId string
|
|
|
|
SPublicIpInfo
|
|
|
|
Tags map[string]string
|
|
|
|
BillingCycle *billing.SBillingCycle
|
|
|
|
IsNeedInjectPasswordByCloudInit bool
|
|
UserDataType string
|
|
WindowsUserDataType string
|
|
IsWindowsUserDataTypeNeedEncode bool
|
|
}
|
|
|
|
type SManagedVMChangeConfig struct {
|
|
Cpu int
|
|
MemoryMB int
|
|
InstanceType string
|
|
}
|
|
|
|
type SManagedVMRebuildRootConfig struct {
|
|
Account string
|
|
Password string
|
|
ImageId string
|
|
PublicKey string
|
|
SysSizeGB int
|
|
OsType string
|
|
}
|
|
|
|
func (vmConfig *SManagedVMCreateConfig) GetConfig(config *jsonutils.JSONDict) error {
|
|
err := config.Unmarshal(vmConfig, "desc")
|
|
if err != nil {
|
|
return errors.Wrapf(err, "config.Unmarshal")
|
|
}
|
|
if !vmConfig.IsNeedInjectPasswordByCloudInit {
|
|
if len(vmConfig.UserData) > 0 {
|
|
_, err := cloudinit.ParseUserData(vmConfig.UserData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
if publicKey, _ := config.GetString("public_key"); len(publicKey) > 0 {
|
|
vmConfig.PublicKey = publicKey
|
|
}
|
|
//目前所写的userData格式仅支持Linux
|
|
if strings.ToLower(vmConfig.OsType) == strings.ToLower(osprofile.OS_TYPE_LINUX) {
|
|
adminPublicKey, _ := config.GetString("admin_public_key")
|
|
projectPublicKey, _ := config.GetString("project_public_key")
|
|
vmConfig.UserData = generateUserData(adminPublicKey, projectPublicKey, vmConfig.UserData)
|
|
}
|
|
|
|
resetPassword := jsonutils.QueryBoolean(config, "reset_password", false)
|
|
vmConfig.Password, _ = config.GetString("password")
|
|
if resetPassword && len(vmConfig.Password) == 0 {
|
|
vmConfig.Password = seclib2.RandomPassword2(12)
|
|
}
|
|
if vmConfig.IsNeedInjectPasswordByCloudInit {
|
|
err = vmConfig.InjectPasswordByCloudInit()
|
|
if err != nil {
|
|
return errors.Wrapf(err, "InjectPasswordByCloudInit")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func generateUserData(adminPublicKey, projectPublicKey, oUserData string) string {
|
|
var oCloudConfig *cloudinit.SCloudConfig
|
|
|
|
if len(oUserData) > 0 {
|
|
oCloudConfig, _ = cloudinit.ParseUserData(oUserData)
|
|
}
|
|
|
|
ansibleUser := cloudinit.NewUser(ansible.PUBLIC_CLOUD_ANSIBLE_USER)
|
|
ansibleUser.SshKey(adminPublicKey).SshKey(projectPublicKey).SudoPolicy(cloudinit.USER_SUDO_NOPASSWD)
|
|
|
|
cloudConfig := cloudinit.SCloudConfig{
|
|
DisableRoot: 0,
|
|
SshPwauth: cloudinit.SSH_PASSWORD_AUTH_ON,
|
|
|
|
Users: []cloudinit.SUser{
|
|
ansibleUser,
|
|
},
|
|
}
|
|
|
|
if oCloudConfig != nil {
|
|
cloudConfig.Merge(oCloudConfig)
|
|
}
|
|
|
|
return cloudConfig.UserData()
|
|
}
|
|
|
|
func (vmConfig *SManagedVMCreateConfig) GetUserData() (string, error) {
|
|
if len(vmConfig.UserData) == 0 {
|
|
return "", nil
|
|
}
|
|
oUserData, err := cloudinit.ParseUserData(vmConfig.UserData)
|
|
if err != nil {
|
|
// 用户输入非标准cloud-init数据
|
|
if !vmConfig.IsNeedInjectPasswordByCloudInit {
|
|
return base64.StdEncoding.EncodeToString([]byte(vmConfig.UserData)), nil
|
|
}
|
|
return "", err
|
|
}
|
|
if strings.ToLower(vmConfig.OsType) == strings.ToLower(osprofile.OS_TYPE_LINUX) {
|
|
switch vmConfig.UserDataType {
|
|
case CLOUD_SHELL:
|
|
return oUserData.UserDataScriptBase64(), nil
|
|
case CLOUD_SHELL_WITHOUT_ENCRYPT:
|
|
return oUserData.UserDataScript(), nil
|
|
default:
|
|
return oUserData.UserDataBase64(), nil
|
|
}
|
|
} else {
|
|
userData := ""
|
|
switch vmConfig.WindowsUserDataType {
|
|
case CLOUD_EC2:
|
|
userData = oUserData.UserDataEc2()
|
|
default:
|
|
userData = oUserData.UserDataPowerShell()
|
|
}
|
|
if vmConfig.IsWindowsUserDataTypeNeedEncode {
|
|
userData = base64.StdEncoding.EncodeToString([]byte(userData))
|
|
}
|
|
return userData, nil
|
|
}
|
|
}
|
|
|
|
func (vmConfig *SManagedVMCreateConfig) InjectPasswordByCloudInit() error {
|
|
loginUser := cloudinit.NewUser(vmConfig.Account)
|
|
loginUser.SudoPolicy(cloudinit.USER_SUDO_NOPASSWD)
|
|
if len(vmConfig.PublicKey) > 0 {
|
|
loginUser.SshKey(vmConfig.PublicKey)
|
|
}
|
|
if len(vmConfig.Password) > 0 {
|
|
loginUser.Password(vmConfig.Password)
|
|
}
|
|
|
|
cloudconfig := cloudinit.SCloudConfig{
|
|
DisableRoot: 0,
|
|
SshPwauth: cloudinit.SSH_PASSWORD_AUTH_ON,
|
|
Users: []cloudinit.SUser{
|
|
loginUser,
|
|
},
|
|
}
|
|
|
|
if len(vmConfig.UserData) > 0 {
|
|
oCloudConfig, err := cloudinit.ParseUserData(vmConfig.UserData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cloudconfig.Merge(oCloudConfig)
|
|
}
|
|
vmConfig.UserData = cloudconfig.UserData()
|
|
return nil
|
|
}
|
|
|
|
// +onecloud:model-api-gen
|
|
type ServerVncInput struct {
|
|
// 是否使用原生vnc控制台，此选项仅对openstack有效
|
|
// default: false
|
|
Origin bool `json:"origin"`
|
|
}
|
|
|
|
// +onecloud:model-api-gen
|
|
type ServerVncOutput struct {
|
|
Id string `json:"id"`
|
|
|
|
// baremetal
|
|
HostId string `json:"host_id"`
|
|
Zone string `json:"zone"`
|
|
|
|
// kvm host ip
|
|
Host string `json:"host"`
|
|
Protocol string `json:"protocol"`
|
|
Port int64 `json:"port"`
|
|
|
|
Url string `json:"url"`
|
|
InstanceId string `json:"instance_id"`
|
|
InstanceName string `json:"instance_name"`
|
|
Password string `json:"password"`
|
|
VncPassword string `json:"vnc_password"`
|
|
|
|
OsName string `json:"os_name"`
|
|
|
|
// cloudpods
|
|
ApiServer string `json:"api_server"`
|
|
ConnectParams string `json:"connect_params"`
|
|
Session string `json:"session"`
|
|
|
|
Hypervisor string `json:"hypervisor"`
|
|
}
|