config.go 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915
  1. // Copyright 2016 The etcd Authors
  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 embed
  15. import (
  16. "crypto/tls"
  17. "fmt"
  18. "io/ioutil"
  19. "net"
  20. "net/http"
  21. "net/url"
  22. "os"
  23. "path/filepath"
  24. "strings"
  25. "sync"
  26. "time"
  27. "go.etcd.io/etcd/etcdserver"
  28. "go.etcd.io/etcd/etcdserver/api/v3compactor"
  29. "go.etcd.io/etcd/pkg/flags"
  30. "go.etcd.io/etcd/pkg/logutil"
  31. "go.etcd.io/etcd/pkg/netutil"
  32. "go.etcd.io/etcd/pkg/srv"
  33. "go.etcd.io/etcd/pkg/tlsutil"
  34. "go.etcd.io/etcd/pkg/transport"
  35. "go.etcd.io/etcd/pkg/types"
  36. bolt "go.etcd.io/bbolt"
  37. "go.uber.org/zap"
  38. "go.uber.org/zap/zapcore"
  39. "golang.org/x/crypto/bcrypt"
  40. "google.golang.org/grpc"
  41. "sigs.k8s.io/yaml"
  42. )
  43. const (
  44. ClusterStateFlagNew = "new"
  45. ClusterStateFlagExisting = "existing"
  46. DefaultName = "default"
  47. DefaultMaxSnapshots = 5
  48. DefaultMaxWALs = 5
  49. DefaultMaxTxnOps = uint(128)
  50. DefaultMaxRequestBytes = 1.5 * 1024 * 1024
  51. DefaultGRPCKeepAliveMinTime = 5 * time.Second
  52. DefaultGRPCKeepAliveInterval = 2 * time.Hour
  53. DefaultGRPCKeepAliveTimeout = 20 * time.Second
  54. DefaultListenPeerURLs = "http://localhost:2380"
  55. DefaultListenClientURLs = "http://localhost:2379"
  56. DefaultLogOutput = "default"
  57. JournalLogOutput = "systemd/journal"
  58. StdErrLogOutput = "stderr"
  59. StdOutLogOutput = "stdout"
  60. // DefaultStrictReconfigCheck is the default value for "--strict-reconfig-check" flag.
  61. // It's enabled by default.
  62. DefaultStrictReconfigCheck = true
  63. // DefaultEnableV2 is the default value for "--enable-v2" flag.
  64. // v2 API is disabled by default.
  65. DefaultEnableV2 = false
  66. // maxElectionMs specifies the maximum value of election timeout.
  67. // More details are listed in ../Documentation/tuning.md#time-parameters.
  68. maxElectionMs = 50000
  69. // backend freelist map type
  70. freelistMapType = "map"
  71. )
  72. var (
  73. ErrConflictBootstrapFlags = fmt.Errorf("multiple discovery or bootstrap flags are set. " +
  74. "Choose one of \"initial-cluster\", \"discovery\" or \"discovery-srv\"")
  75. ErrUnsetAdvertiseClientURLsFlag = fmt.Errorf("--advertise-client-urls is required when --listen-client-urls is set explicitly")
  76. DefaultInitialAdvertisePeerURLs = "http://localhost:2380"
  77. DefaultAdvertiseClientURLs = "http://localhost:2379"
  78. defaultHostname string
  79. defaultHostStatus error
  80. )
  81. var (
  82. // CompactorModePeriodic is periodic compaction mode
  83. // for "Config.AutoCompactionMode" field.
  84. // If "AutoCompactionMode" is CompactorModePeriodic and
  85. // "AutoCompactionRetention" is "1h", it automatically compacts
  86. // compacts storage every hour.
  87. CompactorModePeriodic = v3compactor.ModePeriodic
  88. // CompactorModeRevision is revision-based compaction mode
  89. // for "Config.AutoCompactionMode" field.
  90. // If "AutoCompactionMode" is CompactorModeRevision and
  91. // "AutoCompactionRetention" is "1000", it compacts log on
  92. // revision 5000 when the current revision is 6000.
  93. // This runs every 5-minute if enough of logs have proceeded.
  94. CompactorModeRevision = v3compactor.ModeRevision
  95. )
  96. func init() {
  97. defaultHostname, defaultHostStatus = netutil.GetDefaultHost()
  98. }
  99. // Config holds the arguments for configuring an etcd server.
  100. type Config struct {
  101. Name string `json:"name"`
  102. Dir string `json:"data-dir"`
  103. WalDir string `json:"wal-dir"`
  104. SnapshotCount uint64 `json:"snapshot-count"`
  105. // SnapshotCatchUpEntries is the number of entries for a slow follower
  106. // to catch-up after compacting the raft storage entries.
  107. // We expect the follower has a millisecond level latency with the leader.
  108. // The max throughput is around 10K. Keep a 5K entries is enough for helping
  109. // follower to catch up.
  110. // WARNING: only change this for tests.
  111. // Always use "DefaultSnapshotCatchUpEntries"
  112. SnapshotCatchUpEntries uint64
  113. MaxSnapFiles uint `json:"max-snapshots"`
  114. MaxWalFiles uint `json:"max-wals"`
  115. // TickMs is the number of milliseconds between heartbeat ticks.
  116. // TODO: decouple tickMs and heartbeat tick (current heartbeat tick = 1).
  117. // make ticks a cluster wide configuration.
  118. TickMs uint `json:"heartbeat-interval"`
  119. ElectionMs uint `json:"election-timeout"`
  120. // InitialElectionTickAdvance is true, then local member fast-forwards
  121. // election ticks to speed up "initial" leader election trigger. This
  122. // benefits the case of larger election ticks. For instance, cross
  123. // datacenter deployment may require longer election timeout of 10-second.
  124. // If true, local node does not need wait up to 10-second. Instead,
  125. // forwards its election ticks to 8-second, and have only 2-second left
  126. // before leader election.
  127. //
  128. // Major assumptions are that:
  129. // - cluster has no active leader thus advancing ticks enables faster
  130. // leader election, or
  131. // - cluster already has an established leader, and rejoining follower
  132. // is likely to receive heartbeats from the leader after tick advance
  133. // and before election timeout.
  134. //
  135. // However, when network from leader to rejoining follower is congested,
  136. // and the follower does not receive leader heartbeat within left election
  137. // ticks, disruptive election has to happen thus affecting cluster
  138. // availabilities.
  139. //
  140. // Disabling this would slow down initial bootstrap process for cross
  141. // datacenter deployments. Make your own tradeoffs by configuring
  142. // --initial-election-tick-advance at the cost of slow initial bootstrap.
  143. //
  144. // If single-node, it advances ticks regardless.
  145. //
  146. // See https://github.com/etcd-io/etcd/issues/9333 for more detail.
  147. InitialElectionTickAdvance bool `json:"initial-election-tick-advance"`
  148. // BackendBatchInterval is the maximum time before commit the backend transaction.
  149. BackendBatchInterval time.Duration `json:"backend-batch-interval"`
  150. // BackendBatchLimit is the maximum operations before commit the backend transaction.
  151. BackendBatchLimit int `json:"backend-batch-limit"`
  152. QuotaBackendBytes int64 `json:"quota-backend-bytes"`
  153. MaxTxnOps uint `json:"max-txn-ops"`
  154. MaxRequestBytes uint `json:"max-request-bytes"`
  155. LPUrls, LCUrls []url.URL
  156. APUrls, ACUrls []url.URL
  157. ClientTLSInfo transport.TLSInfo
  158. ClientAutoTLS bool
  159. PeerTLSInfo transport.TLSInfo
  160. PeerAutoTLS bool
  161. // CipherSuites is a list of supported TLS cipher suites between
  162. // client/server and peers. If empty, Go auto-populates the list.
  163. // Note that cipher suites are prioritized in the given order.
  164. CipherSuites []string `json:"cipher-suites"`
  165. ClusterState string `json:"initial-cluster-state"`
  166. DNSCluster string `json:"discovery-srv"`
  167. DNSClusterServiceName string `json:"discovery-srv-name"`
  168. Dproxy string `json:"discovery-proxy"`
  169. Durl string `json:"discovery"`
  170. InitialCluster string `json:"initial-cluster"`
  171. InitialClusterToken string `json:"initial-cluster-token"`
  172. StrictReconfigCheck bool `json:"strict-reconfig-check"`
  173. EnableV2 bool `json:"enable-v2"`
  174. // AutoCompactionMode is either 'periodic' or 'revision'.
  175. AutoCompactionMode string `json:"auto-compaction-mode"`
  176. // AutoCompactionRetention is either duration string with time unit
  177. // (e.g. '5m' for 5-minute), or revision unit (e.g. '5000').
  178. // If no time unit is provided and compaction mode is 'periodic',
  179. // the unit defaults to hour. For example, '5' translates into 5-hour.
  180. AutoCompactionRetention string `json:"auto-compaction-retention"`
  181. // GRPCKeepAliveMinTime is the minimum interval that a client should
  182. // wait before pinging server. When client pings "too fast", server
  183. // sends goaway and closes the connection (errors: too_many_pings,
  184. // http2.ErrCodeEnhanceYourCalm). When too slow, nothing happens.
  185. // Server expects client pings only when there is any active streams
  186. // (PermitWithoutStream is set false).
  187. GRPCKeepAliveMinTime time.Duration `json:"grpc-keepalive-min-time"`
  188. // GRPCKeepAliveInterval is the frequency of server-to-client ping
  189. // to check if a connection is alive. Close a non-responsive connection
  190. // after an additional duration of Timeout. 0 to disable.
  191. GRPCKeepAliveInterval time.Duration `json:"grpc-keepalive-interval"`
  192. // GRPCKeepAliveTimeout is the additional duration of wait
  193. // before closing a non-responsive connection. 0 to disable.
  194. GRPCKeepAliveTimeout time.Duration `json:"grpc-keepalive-timeout"`
  195. // PreVote is true to enable Raft Pre-Vote.
  196. // If enabled, Raft runs an additional election phase
  197. // to check whether it would get enough votes to win
  198. // an election, thus minimizing disruptions.
  199. // TODO: enable by default in 3.5.
  200. PreVote bool `json:"pre-vote"`
  201. CORS map[string]struct{}
  202. // HostWhitelist lists acceptable hostnames from HTTP client requests.
  203. // Client origin policy protects against "DNS Rebinding" attacks
  204. // to insecure etcd servers. That is, any website can simply create
  205. // an authorized DNS name, and direct DNS to "localhost" (or any
  206. // other address). Then, all HTTP endpoints of etcd server listening
  207. // on "localhost" becomes accessible, thus vulnerable to DNS rebinding
  208. // attacks. See "CVE-2018-5702" for more detail.
  209. //
  210. // 1. If client connection is secure via HTTPS, allow any hostnames.
  211. // 2. If client connection is not secure and "HostWhitelist" is not empty,
  212. // only allow HTTP requests whose Host field is listed in whitelist.
  213. //
  214. // Note that the client origin policy is enforced whether authentication
  215. // is enabled or not, for tighter controls.
  216. //
  217. // By default, "HostWhitelist" is "*", which allows any hostnames.
  218. // Note that when specifying hostnames, loopback addresses are not added
  219. // automatically. To allow loopback interfaces, leave it empty or set it "*",
  220. // or add them to whitelist manually (e.g. "localhost", "127.0.0.1", etc.).
  221. //
  222. // CVE-2018-5702 reference:
  223. // - https://bugs.chromium.org/p/project-zero/issues/detail?id=1447#c2
  224. // - https://github.com/transmission/transmission/pull/468
  225. // - https://github.com/etcd-io/etcd/issues/9353
  226. HostWhitelist map[string]struct{}
  227. // UserHandlers is for registering users handlers and only used for
  228. // embedding etcd into other applications.
  229. // The map key is the route path for the handler, and
  230. // you must ensure it can't be conflicted with etcd's.
  231. UserHandlers map[string]http.Handler `json:"-"`
  232. // ServiceRegister is for registering users' gRPC services. A simple usage example:
  233. // cfg := embed.NewConfig()
  234. // cfg.ServerRegister = func(s *grpc.Server) {
  235. // pb.RegisterFooServer(s, &fooServer{})
  236. // pb.RegisterBarServer(s, &barServer{})
  237. // }
  238. // embed.StartEtcd(cfg)
  239. ServiceRegister func(*grpc.Server) `json:"-"`
  240. AuthToken string `json:"auth-token"`
  241. BcryptCost uint `json:"bcrypt-cost"`
  242. ExperimentalInitialCorruptCheck bool `json:"experimental-initial-corrupt-check"`
  243. ExperimentalCorruptCheckTime time.Duration `json:"experimental-corrupt-check-time"`
  244. ExperimentalEnableV2V3 string `json:"experimental-enable-v2v3"`
  245. // ExperimentalBackendFreelistType specifies the type of freelist that boltdb backend uses (array and map are supported types).
  246. ExperimentalBackendFreelistType string `json:"experimental-backend-bbolt-freelist-type"`
  247. // ExperimentalEnableLeaseCheckpoint enables primary lessor to persist lease remainingTTL to prevent indefinite auto-renewal of long lived leases.
  248. ExperimentalEnableLeaseCheckpoint bool `json:"experimental-enable-lease-checkpoint"`
  249. ExperimentalCompactionBatchLimit int `json:"experimental-compaction-batch-limit"`
  250. // ForceNewCluster starts a new cluster even if previously started; unsafe.
  251. ForceNewCluster bool `json:"force-new-cluster"`
  252. EnablePprof bool `json:"enable-pprof"`
  253. Metrics string `json:"metrics"`
  254. ListenMetricsUrls []url.URL
  255. ListenMetricsUrlsJSON string `json:"listen-metrics-urls"`
  256. // Logger is logger options: "zap", "capnslog".
  257. // WARN: "capnslog" is being deprecated in v3.5.
  258. Logger string `json:"logger"`
  259. // LogLevel configures log level. Only supports debug, info, warn, error, panic, or fatal. Default 'info'.
  260. LogLevel string `json:"log-level"`
  261. // LogOutputs is either:
  262. // - "default" as os.Stderr,
  263. // - "stderr" as os.Stderr,
  264. // - "stdout" as os.Stdout,
  265. // - file path to append server logs to.
  266. // It can be multiple when "Logger" is zap.
  267. LogOutputs []string `json:"log-outputs"`
  268. // ZapLoggerBuilder is used to build the zap logger.
  269. ZapLoggerBuilder func(*Config) error
  270. // logger logs server-side operations. The default is nil,
  271. // and "setupLogging" must be called before starting server.
  272. // Do not set logger directly.
  273. loggerMu *sync.RWMutex
  274. logger *zap.Logger
  275. // loggerConfig is server logger configuration for Raft logger.
  276. // Must be either: "loggerConfig != nil" or "loggerCore != nil && loggerWriteSyncer != nil".
  277. loggerConfig *zap.Config
  278. // loggerCore is "zapcore.Core" for raft logger.
  279. // Must be either: "loggerConfig != nil" or "loggerCore != nil && loggerWriteSyncer != nil".
  280. loggerCore zapcore.Core
  281. loggerWriteSyncer zapcore.WriteSyncer
  282. // EnableGRPCGateway is false to disable grpc gateway.
  283. EnableGRPCGateway bool `json:"enable-grpc-gateway"`
  284. // TO BE DEPRECATED
  285. // DeprecatedLogOutput is to be deprecated in v3.5.
  286. // Just here for safe migration in v3.4.
  287. DeprecatedLogOutput []string `json:"log-output"`
  288. // Debug is true, to enable debug level logging.
  289. // WARNING: to be deprecated in 3.5. Use "--log-level=debug" instead.
  290. Debug bool `json:"debug"`
  291. // LogPkgLevels is being deprecated in v3.5.
  292. // Only valid if "logger" option is "capnslog".
  293. // WARN: DO NOT USE THIS!
  294. LogPkgLevels string `json:"log-package-levels"`
  295. }
  296. // configYAML holds the config suitable for yaml parsing
  297. type configYAML struct {
  298. Config
  299. configJSON
  300. }
  301. // configJSON has file options that are translated into Config options
  302. type configJSON struct {
  303. LPUrlsJSON string `json:"listen-peer-urls"`
  304. LCUrlsJSON string `json:"listen-client-urls"`
  305. APUrlsJSON string `json:"initial-advertise-peer-urls"`
  306. ACUrlsJSON string `json:"advertise-client-urls"`
  307. CORSJSON string `json:"cors"`
  308. HostWhitelistJSON string `json:"host-whitelist"`
  309. ClientSecurityJSON securityConfig `json:"client-transport-security"`
  310. PeerSecurityJSON securityConfig `json:"peer-transport-security"`
  311. }
  312. type securityConfig struct {
  313. CertFile string `json:"cert-file"`
  314. KeyFile string `json:"key-file"`
  315. CertAuth bool `json:"client-cert-auth"`
  316. TrustedCAFile string `json:"trusted-ca-file"`
  317. AutoTLS bool `json:"auto-tls"`
  318. }
  319. // NewConfig creates a new Config populated with default values.
  320. func NewConfig() *Config {
  321. lpurl, _ := url.Parse(DefaultListenPeerURLs)
  322. apurl, _ := url.Parse(DefaultInitialAdvertisePeerURLs)
  323. lcurl, _ := url.Parse(DefaultListenClientURLs)
  324. acurl, _ := url.Parse(DefaultAdvertiseClientURLs)
  325. cfg := &Config{
  326. MaxSnapFiles: DefaultMaxSnapshots,
  327. MaxWalFiles: DefaultMaxWALs,
  328. Name: DefaultName,
  329. SnapshotCount: etcdserver.DefaultSnapshotCount,
  330. SnapshotCatchUpEntries: etcdserver.DefaultSnapshotCatchUpEntries,
  331. MaxTxnOps: DefaultMaxTxnOps,
  332. MaxRequestBytes: DefaultMaxRequestBytes,
  333. GRPCKeepAliveMinTime: DefaultGRPCKeepAliveMinTime,
  334. GRPCKeepAliveInterval: DefaultGRPCKeepAliveInterval,
  335. GRPCKeepAliveTimeout: DefaultGRPCKeepAliveTimeout,
  336. TickMs: 100,
  337. ElectionMs: 1000,
  338. InitialElectionTickAdvance: true,
  339. LPUrls: []url.URL{*lpurl},
  340. LCUrls: []url.URL{*lcurl},
  341. APUrls: []url.URL{*apurl},
  342. ACUrls: []url.URL{*acurl},
  343. ClusterState: ClusterStateFlagNew,
  344. InitialClusterToken: "etcd-cluster",
  345. StrictReconfigCheck: DefaultStrictReconfigCheck,
  346. Metrics: "basic",
  347. EnableV2: DefaultEnableV2,
  348. CORS: map[string]struct{}{"*": {}},
  349. HostWhitelist: map[string]struct{}{"*": {}},
  350. AuthToken: "simple",
  351. BcryptCost: uint(bcrypt.DefaultCost),
  352. PreVote: false, // TODO: enable by default in v3.5
  353. loggerMu: new(sync.RWMutex),
  354. logger: nil,
  355. Logger: "capnslog",
  356. DeprecatedLogOutput: []string{DefaultLogOutput},
  357. LogOutputs: []string{DefaultLogOutput},
  358. Debug: false,
  359. LogLevel: logutil.DefaultLogLevel,
  360. LogPkgLevels: "",
  361. }
  362. cfg.InitialCluster = cfg.InitialClusterFromName(cfg.Name)
  363. return cfg
  364. }
  365. func logTLSHandshakeFailure(conn *tls.Conn, err error) {
  366. state := conn.ConnectionState()
  367. remoteAddr := conn.RemoteAddr().String()
  368. serverName := state.ServerName
  369. if len(state.PeerCertificates) > 0 {
  370. cert := state.PeerCertificates[0]
  371. ips, dns := cert.IPAddresses, cert.DNSNames
  372. plog.Infof("rejected connection from %q (error %q, ServerName %q, IPAddresses %q, DNSNames %q)", remoteAddr, err.Error(), serverName, ips, dns)
  373. } else {
  374. plog.Infof("rejected connection from %q (error %q, ServerName %q)", remoteAddr, err.Error(), serverName)
  375. }
  376. }
  377. func ConfigFromFile(path string) (*Config, error) {
  378. cfg := &configYAML{Config: *NewConfig()}
  379. if err := cfg.configFromFile(path); err != nil {
  380. return nil, err
  381. }
  382. return &cfg.Config, nil
  383. }
  384. func (cfg *configYAML) configFromFile(path string) error {
  385. b, err := ioutil.ReadFile(path)
  386. if err != nil {
  387. return err
  388. }
  389. defaultInitialCluster := cfg.InitialCluster
  390. err = yaml.Unmarshal(b, cfg)
  391. if err != nil {
  392. return err
  393. }
  394. if cfg.LPUrlsJSON != "" {
  395. u, err := types.NewURLs(strings.Split(cfg.LPUrlsJSON, ","))
  396. if err != nil {
  397. fmt.Fprintf(os.Stderr, "unexpected error setting up listen-peer-urls: %v\n", err)
  398. os.Exit(1)
  399. }
  400. cfg.LPUrls = []url.URL(u)
  401. }
  402. if cfg.LCUrlsJSON != "" {
  403. u, err := types.NewURLs(strings.Split(cfg.LCUrlsJSON, ","))
  404. if err != nil {
  405. fmt.Fprintf(os.Stderr, "unexpected error setting up listen-client-urls: %v\n", err)
  406. os.Exit(1)
  407. }
  408. cfg.LCUrls = []url.URL(u)
  409. }
  410. if cfg.APUrlsJSON != "" {
  411. u, err := types.NewURLs(strings.Split(cfg.APUrlsJSON, ","))
  412. if err != nil {
  413. fmt.Fprintf(os.Stderr, "unexpected error setting up initial-advertise-peer-urls: %v\n", err)
  414. os.Exit(1)
  415. }
  416. cfg.APUrls = []url.URL(u)
  417. }
  418. if cfg.ACUrlsJSON != "" {
  419. u, err := types.NewURLs(strings.Split(cfg.ACUrlsJSON, ","))
  420. if err != nil {
  421. fmt.Fprintf(os.Stderr, "unexpected error setting up advertise-peer-urls: %v\n", err)
  422. os.Exit(1)
  423. }
  424. cfg.ACUrls = []url.URL(u)
  425. }
  426. if cfg.ListenMetricsUrlsJSON != "" {
  427. u, err := types.NewURLs(strings.Split(cfg.ListenMetricsUrlsJSON, ","))
  428. if err != nil {
  429. fmt.Fprintf(os.Stderr, "unexpected error setting up listen-metrics-urls: %v\n", err)
  430. os.Exit(1)
  431. }
  432. cfg.ListenMetricsUrls = []url.URL(u)
  433. }
  434. if cfg.CORSJSON != "" {
  435. uv := flags.NewUniqueURLsWithExceptions(cfg.CORSJSON, "*")
  436. cfg.CORS = uv.Values
  437. }
  438. if cfg.HostWhitelistJSON != "" {
  439. uv := flags.NewUniqueStringsValue(cfg.HostWhitelistJSON)
  440. cfg.HostWhitelist = uv.Values
  441. }
  442. // If a discovery flag is set, clear default initial cluster set by InitialClusterFromName
  443. if (cfg.Durl != "" || cfg.DNSCluster != "") && cfg.InitialCluster == defaultInitialCluster {
  444. cfg.InitialCluster = ""
  445. }
  446. if cfg.ClusterState == "" {
  447. cfg.ClusterState = ClusterStateFlagNew
  448. }
  449. copySecurityDetails := func(tls *transport.TLSInfo, ysc *securityConfig) {
  450. tls.CertFile = ysc.CertFile
  451. tls.KeyFile = ysc.KeyFile
  452. tls.ClientCertAuth = ysc.CertAuth
  453. tls.TrustedCAFile = ysc.TrustedCAFile
  454. }
  455. copySecurityDetails(&cfg.ClientTLSInfo, &cfg.ClientSecurityJSON)
  456. copySecurityDetails(&cfg.PeerTLSInfo, &cfg.PeerSecurityJSON)
  457. cfg.ClientAutoTLS = cfg.ClientSecurityJSON.AutoTLS
  458. cfg.PeerAutoTLS = cfg.PeerSecurityJSON.AutoTLS
  459. return cfg.Validate()
  460. }
  461. func updateCipherSuites(tls *transport.TLSInfo, ss []string) error {
  462. if len(tls.CipherSuites) > 0 && len(ss) > 0 {
  463. return fmt.Errorf("TLSInfo.CipherSuites is already specified (given %v)", ss)
  464. }
  465. if len(ss) > 0 {
  466. cs := make([]uint16, len(ss))
  467. for i, s := range ss {
  468. var ok bool
  469. cs[i], ok = tlsutil.GetCipherSuite(s)
  470. if !ok {
  471. return fmt.Errorf("unexpected TLS cipher suite %q", s)
  472. }
  473. }
  474. tls.CipherSuites = cs
  475. }
  476. return nil
  477. }
  478. // Validate ensures that '*embed.Config' fields are properly configured.
  479. func (cfg *Config) Validate() error {
  480. if err := cfg.setupLogging(); err != nil {
  481. return err
  482. }
  483. if err := checkBindURLs(cfg.LPUrls); err != nil {
  484. return err
  485. }
  486. if err := checkBindURLs(cfg.LCUrls); err != nil {
  487. return err
  488. }
  489. if err := checkBindURLs(cfg.ListenMetricsUrls); err != nil {
  490. return err
  491. }
  492. if err := checkHostURLs(cfg.APUrls); err != nil {
  493. addrs := cfg.getAPURLs()
  494. return fmt.Errorf(`--initial-advertise-peer-urls %q must be "host:port" (%v)`, strings.Join(addrs, ","), err)
  495. }
  496. if err := checkHostURLs(cfg.ACUrls); err != nil {
  497. addrs := cfg.getACURLs()
  498. return fmt.Errorf(`--advertise-client-urls %q must be "host:port" (%v)`, strings.Join(addrs, ","), err)
  499. }
  500. // Check if conflicting flags are passed.
  501. nSet := 0
  502. for _, v := range []bool{cfg.Durl != "", cfg.InitialCluster != "", cfg.DNSCluster != ""} {
  503. if v {
  504. nSet++
  505. }
  506. }
  507. if cfg.ClusterState != ClusterStateFlagNew && cfg.ClusterState != ClusterStateFlagExisting {
  508. return fmt.Errorf("unexpected clusterState %q", cfg.ClusterState)
  509. }
  510. if nSet > 1 {
  511. return ErrConflictBootstrapFlags
  512. }
  513. if cfg.TickMs <= 0 {
  514. return fmt.Errorf("--heartbeat-interval must be >0 (set to %dms)", cfg.TickMs)
  515. }
  516. if cfg.ElectionMs <= 0 {
  517. return fmt.Errorf("--election-timeout must be >0 (set to %dms)", cfg.ElectionMs)
  518. }
  519. if 5*cfg.TickMs > cfg.ElectionMs {
  520. return fmt.Errorf("--election-timeout[%vms] should be at least as 5 times as --heartbeat-interval[%vms]", cfg.ElectionMs, cfg.TickMs)
  521. }
  522. if cfg.ElectionMs > maxElectionMs {
  523. return fmt.Errorf("--election-timeout[%vms] is too long, and should be set less than %vms", cfg.ElectionMs, maxElectionMs)
  524. }
  525. // check this last since proxying in etcdmain may make this OK
  526. if cfg.LCUrls != nil && cfg.ACUrls == nil {
  527. return ErrUnsetAdvertiseClientURLsFlag
  528. }
  529. switch cfg.AutoCompactionMode {
  530. case "":
  531. case CompactorModeRevision, CompactorModePeriodic:
  532. default:
  533. return fmt.Errorf("unknown auto-compaction-mode %q", cfg.AutoCompactionMode)
  534. }
  535. return nil
  536. }
  537. // PeerURLsMapAndToken sets up an initial peer URLsMap and cluster token for bootstrap or discovery.
  538. func (cfg *Config) PeerURLsMapAndToken(which string) (urlsmap types.URLsMap, token string, err error) {
  539. token = cfg.InitialClusterToken
  540. switch {
  541. case cfg.Durl != "":
  542. urlsmap = types.URLsMap{}
  543. // If using discovery, generate a temporary cluster based on
  544. // self's advertised peer URLs
  545. urlsmap[cfg.Name] = cfg.APUrls
  546. token = cfg.Durl
  547. case cfg.DNSCluster != "":
  548. clusterStrs, cerr := cfg.GetDNSClusterNames()
  549. lg := cfg.logger
  550. if cerr != nil {
  551. if lg != nil {
  552. lg.Warn("failed to resolve during SRV discovery", zap.Error(cerr))
  553. } else {
  554. plog.Errorf("couldn't resolve during SRV discovery (%v)", cerr)
  555. }
  556. return nil, "", cerr
  557. }
  558. for _, s := range clusterStrs {
  559. if lg != nil {
  560. lg.Info("got bootstrap from DNS for etcd-server", zap.String("node", s))
  561. } else {
  562. plog.Noticef("got bootstrap from DNS for etcd-server at %s", s)
  563. }
  564. }
  565. clusterStr := strings.Join(clusterStrs, ",")
  566. if strings.Contains(clusterStr, "https://") && cfg.PeerTLSInfo.TrustedCAFile == "" {
  567. cfg.PeerTLSInfo.ServerName = cfg.DNSCluster
  568. }
  569. urlsmap, err = types.NewURLsMap(clusterStr)
  570. // only etcd member must belong to the discovered cluster.
  571. // proxy does not need to belong to the discovered cluster.
  572. if which == "etcd" {
  573. if _, ok := urlsmap[cfg.Name]; !ok {
  574. return nil, "", fmt.Errorf("cannot find local etcd member %q in SRV records", cfg.Name)
  575. }
  576. }
  577. default:
  578. // We're statically configured, and cluster has appropriately been set.
  579. urlsmap, err = types.NewURLsMap(cfg.InitialCluster)
  580. }
  581. return urlsmap, token, err
  582. }
  583. // GetDNSClusterNames uses DNS SRV records to get a list of initial nodes for cluster bootstrapping.
  584. func (cfg *Config) GetDNSClusterNames() ([]string, error) {
  585. var (
  586. clusterStrs []string
  587. cerr error
  588. serviceNameSuffix string
  589. )
  590. if cfg.DNSClusterServiceName != "" {
  591. serviceNameSuffix = "-" + cfg.DNSClusterServiceName
  592. }
  593. lg := cfg.GetLogger()
  594. // Use both etcd-server-ssl and etcd-server for discovery.
  595. // Combine the results if both are available.
  596. clusterStrs, cerr = srv.GetCluster("https", "etcd-server-ssl"+serviceNameSuffix, cfg.Name, cfg.DNSCluster, cfg.APUrls)
  597. if cerr != nil {
  598. clusterStrs = make([]string, 0)
  599. }
  600. if lg != nil {
  601. lg.Info(
  602. "get cluster for etcd-server-ssl SRV",
  603. zap.String("service-scheme", "https"),
  604. zap.String("service-name", "etcd-server-ssl"+serviceNameSuffix),
  605. zap.String("server-name", cfg.Name),
  606. zap.String("discovery-srv", cfg.DNSCluster),
  607. zap.Strings("advertise-peer-urls", cfg.getAPURLs()),
  608. zap.Strings("found-cluster", clusterStrs),
  609. zap.Error(cerr),
  610. )
  611. }
  612. defaultHTTPClusterStrs, httpCerr := srv.GetCluster("http", "etcd-server"+serviceNameSuffix, cfg.Name, cfg.DNSCluster, cfg.APUrls)
  613. if httpCerr != nil {
  614. clusterStrs = append(clusterStrs, defaultHTTPClusterStrs...)
  615. }
  616. if lg != nil {
  617. lg.Info(
  618. "get cluster for etcd-server SRV",
  619. zap.String("service-scheme", "http"),
  620. zap.String("service-name", "etcd-server"+serviceNameSuffix),
  621. zap.String("server-name", cfg.Name),
  622. zap.String("discovery-srv", cfg.DNSCluster),
  623. zap.Strings("advertise-peer-urls", cfg.getAPURLs()),
  624. zap.Strings("found-cluster", clusterStrs),
  625. zap.Error(httpCerr),
  626. )
  627. }
  628. return clusterStrs, cerr
  629. }
  630. func (cfg Config) InitialClusterFromName(name string) (ret string) {
  631. if len(cfg.APUrls) == 0 {
  632. return ""
  633. }
  634. n := name
  635. if name == "" {
  636. n = DefaultName
  637. }
  638. for i := range cfg.APUrls {
  639. ret = ret + "," + n + "=" + cfg.APUrls[i].String()
  640. }
  641. return ret[1:]
  642. }
  643. func (cfg Config) IsNewCluster() bool { return cfg.ClusterState == ClusterStateFlagNew }
  644. func (cfg Config) ElectionTicks() int { return int(cfg.ElectionMs / cfg.TickMs) }
  645. func (cfg Config) defaultPeerHost() bool {
  646. return len(cfg.APUrls) == 1 && cfg.APUrls[0].String() == DefaultInitialAdvertisePeerURLs
  647. }
  648. func (cfg Config) defaultClientHost() bool {
  649. return len(cfg.ACUrls) == 1 && cfg.ACUrls[0].String() == DefaultAdvertiseClientURLs
  650. }
  651. func (cfg *Config) ClientSelfCert() (err error) {
  652. if !cfg.ClientAutoTLS {
  653. return nil
  654. }
  655. if !cfg.ClientTLSInfo.Empty() {
  656. if cfg.logger != nil {
  657. cfg.logger.Warn("ignoring client auto TLS since certs given")
  658. } else {
  659. plog.Warningf("ignoring client auto TLS since certs given")
  660. }
  661. return nil
  662. }
  663. chosts := make([]string, len(cfg.LCUrls))
  664. for i, u := range cfg.LCUrls {
  665. chosts[i] = u.Host
  666. }
  667. cfg.ClientTLSInfo, err = transport.SelfCert(cfg.logger, filepath.Join(cfg.Dir, "fixtures", "client"), chosts)
  668. if err != nil {
  669. return err
  670. }
  671. return updateCipherSuites(&cfg.ClientTLSInfo, cfg.CipherSuites)
  672. }
  673. func (cfg *Config) PeerSelfCert() (err error) {
  674. if !cfg.PeerAutoTLS {
  675. return nil
  676. }
  677. if !cfg.PeerTLSInfo.Empty() {
  678. if cfg.logger != nil {
  679. cfg.logger.Warn("ignoring peer auto TLS since certs given")
  680. } else {
  681. plog.Warningf("ignoring peer auto TLS since certs given")
  682. }
  683. return nil
  684. }
  685. phosts := make([]string, len(cfg.LPUrls))
  686. for i, u := range cfg.LPUrls {
  687. phosts[i] = u.Host
  688. }
  689. cfg.PeerTLSInfo, err = transport.SelfCert(cfg.logger, filepath.Join(cfg.Dir, "fixtures", "peer"), phosts)
  690. if err != nil {
  691. return err
  692. }
  693. return updateCipherSuites(&cfg.PeerTLSInfo, cfg.CipherSuites)
  694. }
  695. // UpdateDefaultClusterFromName updates cluster advertise URLs with, if available, default host,
  696. // if advertise URLs are default values(localhost:2379,2380) AND if listen URL is 0.0.0.0.
  697. // e.g. advertise peer URL localhost:2380 or listen peer URL 0.0.0.0:2380
  698. // then the advertise peer host would be updated with machine's default host,
  699. // while keeping the listen URL's port.
  700. // User can work around this by explicitly setting URL with 127.0.0.1.
  701. // It returns the default hostname, if used, and the error, if any, from getting the machine's default host.
  702. // TODO: check whether fields are set instead of whether fields have default value
  703. func (cfg *Config) UpdateDefaultClusterFromName(defaultInitialCluster string) (string, error) {
  704. if defaultHostname == "" || defaultHostStatus != nil {
  705. // update 'initial-cluster' when only the name is specified (e.g. 'etcd --name=abc')
  706. if cfg.Name != DefaultName && cfg.InitialCluster == defaultInitialCluster {
  707. cfg.InitialCluster = cfg.InitialClusterFromName(cfg.Name)
  708. }
  709. return "", defaultHostStatus
  710. }
  711. used := false
  712. pip, pport := cfg.LPUrls[0].Hostname(), cfg.LPUrls[0].Port()
  713. if cfg.defaultPeerHost() && pip == "0.0.0.0" {
  714. cfg.APUrls[0] = url.URL{Scheme: cfg.APUrls[0].Scheme, Host: fmt.Sprintf("%s:%s", defaultHostname, pport)}
  715. used = true
  716. }
  717. // update 'initial-cluster' when only the name is specified (e.g. 'etcd --name=abc')
  718. if cfg.Name != DefaultName && cfg.InitialCluster == defaultInitialCluster {
  719. cfg.InitialCluster = cfg.InitialClusterFromName(cfg.Name)
  720. }
  721. cip, cport := cfg.LCUrls[0].Hostname(), cfg.LCUrls[0].Port()
  722. if cfg.defaultClientHost() && cip == "0.0.0.0" {
  723. cfg.ACUrls[0] = url.URL{Scheme: cfg.ACUrls[0].Scheme, Host: fmt.Sprintf("%s:%s", defaultHostname, cport)}
  724. used = true
  725. }
  726. dhost := defaultHostname
  727. if !used {
  728. dhost = ""
  729. }
  730. return dhost, defaultHostStatus
  731. }
  732. // checkBindURLs returns an error if any URL uses a domain name.
  733. func checkBindURLs(urls []url.URL) error {
  734. for _, url := range urls {
  735. if url.Scheme == "unix" || url.Scheme == "unixs" {
  736. continue
  737. }
  738. host, _, err := net.SplitHostPort(url.Host)
  739. if err != nil {
  740. return err
  741. }
  742. if host == "localhost" {
  743. // special case for local address
  744. // TODO: support /etc/hosts ?
  745. continue
  746. }
  747. if net.ParseIP(host) == nil {
  748. return fmt.Errorf("expected IP in URL for binding (%s)", url.String())
  749. }
  750. }
  751. return nil
  752. }
  753. func checkHostURLs(urls []url.URL) error {
  754. for _, url := range urls {
  755. host, _, err := net.SplitHostPort(url.Host)
  756. if err != nil {
  757. return err
  758. }
  759. if host == "" {
  760. return fmt.Errorf("unexpected empty host (%s)", url.String())
  761. }
  762. }
  763. return nil
  764. }
  765. func (cfg *Config) getAPURLs() (ss []string) {
  766. ss = make([]string, len(cfg.APUrls))
  767. for i := range cfg.APUrls {
  768. ss[i] = cfg.APUrls[i].String()
  769. }
  770. return ss
  771. }
  772. func (cfg *Config) getLPURLs() (ss []string) {
  773. ss = make([]string, len(cfg.LPUrls))
  774. for i := range cfg.LPUrls {
  775. ss[i] = cfg.LPUrls[i].String()
  776. }
  777. return ss
  778. }
  779. func (cfg *Config) getACURLs() (ss []string) {
  780. ss = make([]string, len(cfg.ACUrls))
  781. for i := range cfg.ACUrls {
  782. ss[i] = cfg.ACUrls[i].String()
  783. }
  784. return ss
  785. }
  786. func (cfg *Config) getLCURLs() (ss []string) {
  787. ss = make([]string, len(cfg.LCUrls))
  788. for i := range cfg.LCUrls {
  789. ss[i] = cfg.LCUrls[i].String()
  790. }
  791. return ss
  792. }
  793. func (cfg *Config) getMetricsURLs() (ss []string) {
  794. ss = make([]string, len(cfg.ListenMetricsUrls))
  795. for i := range cfg.ListenMetricsUrls {
  796. ss[i] = cfg.ListenMetricsUrls[i].String()
  797. }
  798. return ss
  799. }
  800. func parseBackendFreelistType(freelistType string) bolt.FreelistType {
  801. if freelistType == freelistMapType {
  802. return bolt.FreelistMapType
  803. }
  804. return bolt.FreelistArrayType
  805. }