v3_server.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693
  1. // Copyright 2015 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 etcdserver
  15. import (
  16. "bytes"
  17. "encoding/binary"
  18. "time"
  19. "github.com/gogo/protobuf/proto"
  20. "github.com/coreos/etcd/auth"
  21. pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
  22. "github.com/coreos/etcd/etcdserver/membership"
  23. "github.com/coreos/etcd/lease"
  24. "github.com/coreos/etcd/lease/leasehttp"
  25. "github.com/coreos/etcd/mvcc"
  26. "github.com/coreos/etcd/raft"
  27. "golang.org/x/net/context"
  28. )
  29. const (
  30. // In the health case, there might be a small gap (10s of entries) between
  31. // the applied index and committed index.
  32. // However, if the committed entries are very heavy to apply, the gap might grow.
  33. // We should stop accepting new proposals if the gap growing to a certain point.
  34. maxGapBetweenApplyAndCommitIndex = 5000
  35. )
  36. type RaftKV interface {
  37. Range(ctx context.Context, r *pb.RangeRequest) (*pb.RangeResponse, error)
  38. Put(ctx context.Context, r *pb.PutRequest) (*pb.PutResponse, error)
  39. DeleteRange(ctx context.Context, r *pb.DeleteRangeRequest) (*pb.DeleteRangeResponse, error)
  40. Txn(ctx context.Context, r *pb.TxnRequest) (*pb.TxnResponse, error)
  41. Compact(ctx context.Context, r *pb.CompactionRequest) (*pb.CompactionResponse, error)
  42. }
  43. type Lessor interface {
  44. // LeaseGrant sends LeaseGrant request to raft and apply it after committed.
  45. LeaseGrant(ctx context.Context, r *pb.LeaseGrantRequest) (*pb.LeaseGrantResponse, error)
  46. // LeaseRevoke sends LeaseRevoke request to raft and apply it after committed.
  47. LeaseRevoke(ctx context.Context, r *pb.LeaseRevokeRequest) (*pb.LeaseRevokeResponse, error)
  48. // LeaseRenew renews the lease with given ID. The renewed TTL is returned. Or an error
  49. // is returned.
  50. LeaseRenew(ctx context.Context, id lease.LeaseID) (int64, error)
  51. // LeaseTimeToLive retrieves lease information.
  52. LeaseTimeToLive(ctx context.Context, r *pb.LeaseTimeToLiveRequest) (*pb.LeaseTimeToLiveResponse, error)
  53. }
  54. type Authenticator interface {
  55. AuthEnable(ctx context.Context, r *pb.AuthEnableRequest) (*pb.AuthEnableResponse, error)
  56. AuthDisable(ctx context.Context, r *pb.AuthDisableRequest) (*pb.AuthDisableResponse, error)
  57. Authenticate(ctx context.Context, r *pb.AuthenticateRequest) (*pb.AuthenticateResponse, error)
  58. UserAdd(ctx context.Context, r *pb.AuthUserAddRequest) (*pb.AuthUserAddResponse, error)
  59. UserDelete(ctx context.Context, r *pb.AuthUserDeleteRequest) (*pb.AuthUserDeleteResponse, error)
  60. UserChangePassword(ctx context.Context, r *pb.AuthUserChangePasswordRequest) (*pb.AuthUserChangePasswordResponse, error)
  61. UserGrantRole(ctx context.Context, r *pb.AuthUserGrantRoleRequest) (*pb.AuthUserGrantRoleResponse, error)
  62. UserGet(ctx context.Context, r *pb.AuthUserGetRequest) (*pb.AuthUserGetResponse, error)
  63. UserRevokeRole(ctx context.Context, r *pb.AuthUserRevokeRoleRequest) (*pb.AuthUserRevokeRoleResponse, error)
  64. RoleAdd(ctx context.Context, r *pb.AuthRoleAddRequest) (*pb.AuthRoleAddResponse, error)
  65. RoleGrantPermission(ctx context.Context, r *pb.AuthRoleGrantPermissionRequest) (*pb.AuthRoleGrantPermissionResponse, error)
  66. RoleGet(ctx context.Context, r *pb.AuthRoleGetRequest) (*pb.AuthRoleGetResponse, error)
  67. RoleRevokePermission(ctx context.Context, r *pb.AuthRoleRevokePermissionRequest) (*pb.AuthRoleRevokePermissionResponse, error)
  68. RoleDelete(ctx context.Context, r *pb.AuthRoleDeleteRequest) (*pb.AuthRoleDeleteResponse, error)
  69. UserList(ctx context.Context, r *pb.AuthUserListRequest) (*pb.AuthUserListResponse, error)
  70. RoleList(ctx context.Context, r *pb.AuthRoleListRequest) (*pb.AuthRoleListResponse, error)
  71. }
  72. func (s *EtcdServer) Range(ctx context.Context, r *pb.RangeRequest) (*pb.RangeResponse, error) {
  73. defer warnOfExpensiveReadOnlyRangeRequest(time.Now(), r)
  74. if !r.Serializable {
  75. err := s.linearizableReadNotify(ctx)
  76. if err != nil {
  77. return nil, err
  78. }
  79. }
  80. var resp *pb.RangeResponse
  81. var err error
  82. chk := func(ai *auth.AuthInfo) error {
  83. return s.authStore.IsRangePermitted(ai, r.Key, r.RangeEnd)
  84. }
  85. get := func() { resp, err = s.applyV3Base.Range(nil, r) }
  86. if serr := s.doSerialize(ctx, chk, get); serr != nil {
  87. return nil, serr
  88. }
  89. return resp, err
  90. }
  91. func (s *EtcdServer) Put(ctx context.Context, r *pb.PutRequest) (*pb.PutResponse, error) {
  92. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{Put: r})
  93. if err != nil {
  94. return nil, err
  95. }
  96. return resp.(*pb.PutResponse), nil
  97. }
  98. func (s *EtcdServer) DeleteRange(ctx context.Context, r *pb.DeleteRangeRequest) (*pb.DeleteRangeResponse, error) {
  99. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{DeleteRange: r})
  100. if err != nil {
  101. return nil, err
  102. }
  103. return resp.(*pb.DeleteRangeResponse), nil
  104. }
  105. func (s *EtcdServer) Txn(ctx context.Context, r *pb.TxnRequest) (*pb.TxnResponse, error) {
  106. if isTxnReadonly(r) {
  107. if !isTxnSerializable(r) {
  108. err := s.linearizableReadNotify(ctx)
  109. if err != nil {
  110. return nil, err
  111. }
  112. }
  113. var resp *pb.TxnResponse
  114. var err error
  115. chk := func(ai *auth.AuthInfo) error {
  116. return checkTxnAuth(s.authStore, ai, r)
  117. }
  118. defer warnOfExpensiveReadOnlyRangeRequest(time.Now(), r)
  119. get := func() { resp, err = s.applyV3Base.Txn(r) }
  120. if serr := s.doSerialize(ctx, chk, get); serr != nil {
  121. return nil, serr
  122. }
  123. return resp, err
  124. }
  125. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{Txn: r})
  126. if err != nil {
  127. return nil, err
  128. }
  129. return resp.(*pb.TxnResponse), nil
  130. }
  131. func isTxnSerializable(r *pb.TxnRequest) bool {
  132. for _, u := range r.Success {
  133. if r := u.GetRequestRange(); r == nil || !r.Serializable {
  134. return false
  135. }
  136. }
  137. for _, u := range r.Failure {
  138. if r := u.GetRequestRange(); r == nil || !r.Serializable {
  139. return false
  140. }
  141. }
  142. return true
  143. }
  144. func isTxnReadonly(r *pb.TxnRequest) bool {
  145. for _, u := range r.Success {
  146. if r := u.GetRequestRange(); r == nil {
  147. return false
  148. }
  149. }
  150. for _, u := range r.Failure {
  151. if r := u.GetRequestRange(); r == nil {
  152. return false
  153. }
  154. }
  155. return true
  156. }
  157. func (s *EtcdServer) Compact(ctx context.Context, r *pb.CompactionRequest) (*pb.CompactionResponse, error) {
  158. result, err := s.processInternalRaftRequestOnce(ctx, pb.InternalRaftRequest{Compaction: r})
  159. if r.Physical && result != nil && result.physc != nil {
  160. <-result.physc
  161. // The compaction is done deleting keys; the hash is now settled
  162. // but the data is not necessarily committed. If there's a crash,
  163. // the hash may revert to a hash prior to compaction completing
  164. // if the compaction resumes. Force the finished compaction to
  165. // commit so it won't resume following a crash.
  166. s.be.ForceCommit()
  167. }
  168. if err != nil {
  169. return nil, err
  170. }
  171. if result.err != nil {
  172. return nil, result.err
  173. }
  174. resp := result.resp.(*pb.CompactionResponse)
  175. if resp == nil {
  176. resp = &pb.CompactionResponse{}
  177. }
  178. if resp.Header == nil {
  179. resp.Header = &pb.ResponseHeader{}
  180. }
  181. resp.Header.Revision = s.kv.Rev()
  182. return resp, nil
  183. }
  184. func (s *EtcdServer) LeaseGrant(ctx context.Context, r *pb.LeaseGrantRequest) (*pb.LeaseGrantResponse, error) {
  185. // no id given? choose one
  186. for r.ID == int64(lease.NoLease) {
  187. // only use positive int64 id's
  188. r.ID = int64(s.reqIDGen.Next() & ((1 << 63) - 1))
  189. }
  190. resp, err := s.raftRequestOnce(ctx, pb.InternalRaftRequest{LeaseGrant: r})
  191. if err != nil {
  192. return nil, err
  193. }
  194. return resp.(*pb.LeaseGrantResponse), nil
  195. }
  196. func (s *EtcdServer) LeaseRevoke(ctx context.Context, r *pb.LeaseRevokeRequest) (*pb.LeaseRevokeResponse, error) {
  197. resp, err := s.raftRequestOnce(ctx, pb.InternalRaftRequest{LeaseRevoke: r})
  198. if err != nil {
  199. return nil, err
  200. }
  201. return resp.(*pb.LeaseRevokeResponse), nil
  202. }
  203. func (s *EtcdServer) LeaseRenew(ctx context.Context, id lease.LeaseID) (int64, error) {
  204. ttl, err := s.lessor.Renew(id)
  205. if err == nil { // already requested to primary lessor(leader)
  206. return ttl, nil
  207. }
  208. if err != lease.ErrNotPrimary {
  209. return -1, err
  210. }
  211. cctx, cancel := context.WithTimeout(ctx, s.Cfg.ReqTimeout())
  212. defer cancel()
  213. // renewals don't go through raft; forward to leader manually
  214. for cctx.Err() == nil && err != nil {
  215. leader, lerr := s.waitLeader(cctx)
  216. if lerr != nil {
  217. return -1, lerr
  218. }
  219. for _, url := range leader.PeerURLs {
  220. lurl := url + leasehttp.LeasePrefix
  221. ttl, err = leasehttp.RenewHTTP(cctx, id, lurl, s.peerRt)
  222. if err == nil || err == lease.ErrLeaseNotFound {
  223. return ttl, err
  224. }
  225. }
  226. }
  227. return -1, ErrTimeout
  228. }
  229. func (s *EtcdServer) LeaseTimeToLive(ctx context.Context, r *pb.LeaseTimeToLiveRequest) (*pb.LeaseTimeToLiveResponse, error) {
  230. if s.Leader() == s.ID() {
  231. // primary; timetolive directly from leader
  232. le := s.lessor.Lookup(lease.LeaseID(r.ID))
  233. if le == nil {
  234. return nil, lease.ErrLeaseNotFound
  235. }
  236. // TODO: fill out ResponseHeader
  237. resp := &pb.LeaseTimeToLiveResponse{Header: &pb.ResponseHeader{}, ID: r.ID, TTL: int64(le.Remaining().Seconds()), GrantedTTL: le.TTL()}
  238. if r.Keys {
  239. ks := le.Keys()
  240. kbs := make([][]byte, len(ks))
  241. for i := range ks {
  242. kbs[i] = []byte(ks[i])
  243. }
  244. resp.Keys = kbs
  245. }
  246. return resp, nil
  247. }
  248. cctx, cancel := context.WithTimeout(ctx, s.Cfg.ReqTimeout())
  249. defer cancel()
  250. // forward to leader
  251. for cctx.Err() == nil {
  252. leader, err := s.waitLeader(cctx)
  253. if err != nil {
  254. return nil, err
  255. }
  256. for _, url := range leader.PeerURLs {
  257. lurl := url + leasehttp.LeaseInternalPrefix
  258. resp, err := leasehttp.TimeToLiveHTTP(cctx, lease.LeaseID(r.ID), r.Keys, lurl, s.peerRt)
  259. if err == nil {
  260. return resp.LeaseTimeToLiveResponse, nil
  261. }
  262. if err == lease.ErrLeaseNotFound {
  263. return nil, err
  264. }
  265. }
  266. }
  267. return nil, ErrTimeout
  268. }
  269. func (s *EtcdServer) waitLeader(ctx context.Context) (*membership.Member, error) {
  270. leader := s.cluster.Member(s.Leader())
  271. for leader == nil {
  272. // wait an election
  273. dur := time.Duration(s.Cfg.ElectionTicks) * time.Duration(s.Cfg.TickMs) * time.Millisecond
  274. select {
  275. case <-time.After(dur):
  276. leader = s.cluster.Member(s.Leader())
  277. case <-s.stopping:
  278. return nil, ErrStopped
  279. case <-ctx.Done():
  280. return nil, ErrNoLeader
  281. }
  282. }
  283. if leader == nil || len(leader.PeerURLs) == 0 {
  284. return nil, ErrNoLeader
  285. }
  286. return leader, nil
  287. }
  288. func (s *EtcdServer) Alarm(ctx context.Context, r *pb.AlarmRequest) (*pb.AlarmResponse, error) {
  289. resp, err := s.raftRequestOnce(ctx, pb.InternalRaftRequest{Alarm: r})
  290. if err != nil {
  291. return nil, err
  292. }
  293. return resp.(*pb.AlarmResponse), nil
  294. }
  295. func (s *EtcdServer) AuthEnable(ctx context.Context, r *pb.AuthEnableRequest) (*pb.AuthEnableResponse, error) {
  296. resp, err := s.raftRequestOnce(ctx, pb.InternalRaftRequest{AuthEnable: r})
  297. if err != nil {
  298. return nil, err
  299. }
  300. return resp.(*pb.AuthEnableResponse), nil
  301. }
  302. func (s *EtcdServer) AuthDisable(ctx context.Context, r *pb.AuthDisableRequest) (*pb.AuthDisableResponse, error) {
  303. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthDisable: r})
  304. if err != nil {
  305. return nil, err
  306. }
  307. return resp.(*pb.AuthDisableResponse), nil
  308. }
  309. func (s *EtcdServer) Authenticate(ctx context.Context, r *pb.AuthenticateRequest) (*pb.AuthenticateResponse, error) {
  310. if err := s.linearizableReadNotify(ctx); err != nil {
  311. return nil, err
  312. }
  313. var resp proto.Message
  314. for {
  315. checkedRevision, err := s.AuthStore().CheckPassword(r.Name, r.Password)
  316. if err != nil {
  317. if err != auth.ErrAuthNotEnabled {
  318. plog.Errorf("invalid authentication request to user %s was issued", r.Name)
  319. }
  320. return nil, err
  321. }
  322. st, err := s.AuthStore().GenTokenPrefix()
  323. if err != nil {
  324. return nil, err
  325. }
  326. internalReq := &pb.InternalAuthenticateRequest{
  327. Name: r.Name,
  328. Password: r.Password,
  329. SimpleToken: st,
  330. }
  331. resp, err = s.raftRequestOnce(ctx, pb.InternalRaftRequest{Authenticate: internalReq})
  332. if err != nil {
  333. return nil, err
  334. }
  335. if checkedRevision == s.AuthStore().Revision() {
  336. break
  337. }
  338. plog.Infof("revision when password checked is obsolete, retrying")
  339. }
  340. return resp.(*pb.AuthenticateResponse), nil
  341. }
  342. func (s *EtcdServer) UserAdd(ctx context.Context, r *pb.AuthUserAddRequest) (*pb.AuthUserAddResponse, error) {
  343. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserAdd: r})
  344. if err != nil {
  345. return nil, err
  346. }
  347. return resp.(*pb.AuthUserAddResponse), nil
  348. }
  349. func (s *EtcdServer) UserDelete(ctx context.Context, r *pb.AuthUserDeleteRequest) (*pb.AuthUserDeleteResponse, error) {
  350. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserDelete: r})
  351. if err != nil {
  352. return nil, err
  353. }
  354. return resp.(*pb.AuthUserDeleteResponse), nil
  355. }
  356. func (s *EtcdServer) UserChangePassword(ctx context.Context, r *pb.AuthUserChangePasswordRequest) (*pb.AuthUserChangePasswordResponse, error) {
  357. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserChangePassword: r})
  358. if err != nil {
  359. return nil, err
  360. }
  361. return resp.(*pb.AuthUserChangePasswordResponse), nil
  362. }
  363. func (s *EtcdServer) UserGrantRole(ctx context.Context, r *pb.AuthUserGrantRoleRequest) (*pb.AuthUserGrantRoleResponse, error) {
  364. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserGrantRole: r})
  365. if err != nil {
  366. return nil, err
  367. }
  368. return resp.(*pb.AuthUserGrantRoleResponse), nil
  369. }
  370. func (s *EtcdServer) UserGet(ctx context.Context, r *pb.AuthUserGetRequest) (*pb.AuthUserGetResponse, error) {
  371. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserGet: r})
  372. if err != nil {
  373. return nil, err
  374. }
  375. return resp.(*pb.AuthUserGetResponse), nil
  376. }
  377. func (s *EtcdServer) UserList(ctx context.Context, r *pb.AuthUserListRequest) (*pb.AuthUserListResponse, error) {
  378. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserList: r})
  379. if err != nil {
  380. return nil, err
  381. }
  382. return resp.(*pb.AuthUserListResponse), nil
  383. }
  384. func (s *EtcdServer) UserRevokeRole(ctx context.Context, r *pb.AuthUserRevokeRoleRequest) (*pb.AuthUserRevokeRoleResponse, error) {
  385. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthUserRevokeRole: r})
  386. if err != nil {
  387. return nil, err
  388. }
  389. return resp.(*pb.AuthUserRevokeRoleResponse), nil
  390. }
  391. func (s *EtcdServer) RoleAdd(ctx context.Context, r *pb.AuthRoleAddRequest) (*pb.AuthRoleAddResponse, error) {
  392. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleAdd: r})
  393. if err != nil {
  394. return nil, err
  395. }
  396. return resp.(*pb.AuthRoleAddResponse), nil
  397. }
  398. func (s *EtcdServer) RoleGrantPermission(ctx context.Context, r *pb.AuthRoleGrantPermissionRequest) (*pb.AuthRoleGrantPermissionResponse, error) {
  399. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleGrantPermission: r})
  400. if err != nil {
  401. return nil, err
  402. }
  403. return resp.(*pb.AuthRoleGrantPermissionResponse), nil
  404. }
  405. func (s *EtcdServer) RoleGet(ctx context.Context, r *pb.AuthRoleGetRequest) (*pb.AuthRoleGetResponse, error) {
  406. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleGet: r})
  407. if err != nil {
  408. return nil, err
  409. }
  410. return resp.(*pb.AuthRoleGetResponse), nil
  411. }
  412. func (s *EtcdServer) RoleList(ctx context.Context, r *pb.AuthRoleListRequest) (*pb.AuthRoleListResponse, error) {
  413. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleList: r})
  414. if err != nil {
  415. return nil, err
  416. }
  417. return resp.(*pb.AuthRoleListResponse), nil
  418. }
  419. func (s *EtcdServer) RoleRevokePermission(ctx context.Context, r *pb.AuthRoleRevokePermissionRequest) (*pb.AuthRoleRevokePermissionResponse, error) {
  420. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleRevokePermission: r})
  421. if err != nil {
  422. return nil, err
  423. }
  424. return resp.(*pb.AuthRoleRevokePermissionResponse), nil
  425. }
  426. func (s *EtcdServer) RoleDelete(ctx context.Context, r *pb.AuthRoleDeleteRequest) (*pb.AuthRoleDeleteResponse, error) {
  427. resp, err := s.raftRequest(ctx, pb.InternalRaftRequest{AuthRoleDelete: r})
  428. if err != nil {
  429. return nil, err
  430. }
  431. return resp.(*pb.AuthRoleDeleteResponse), nil
  432. }
  433. func (s *EtcdServer) raftRequestOnce(ctx context.Context, r pb.InternalRaftRequest) (proto.Message, error) {
  434. result, err := s.processInternalRaftRequestOnce(ctx, r)
  435. if err != nil {
  436. return nil, err
  437. }
  438. if result.err != nil {
  439. return nil, result.err
  440. }
  441. return result.resp, nil
  442. }
  443. func (s *EtcdServer) raftRequest(ctx context.Context, r pb.InternalRaftRequest) (proto.Message, error) {
  444. for {
  445. resp, err := s.raftRequestOnce(ctx, r)
  446. if err != auth.ErrAuthOldRevision {
  447. return resp, err
  448. }
  449. }
  450. }
  451. // doSerialize handles the auth logic, with permissions checked by "chk", for a serialized request "get". Returns a non-nil error on authentication failure.
  452. func (s *EtcdServer) doSerialize(ctx context.Context, chk func(*auth.AuthInfo) error, get func()) error {
  453. for {
  454. ai, err := s.AuthInfoFromCtx(ctx)
  455. if err != nil {
  456. return err
  457. }
  458. if ai == nil {
  459. // chk expects non-nil AuthInfo; use empty credentials
  460. ai = &auth.AuthInfo{}
  461. }
  462. if err = chk(ai); err != nil {
  463. if err == auth.ErrAuthOldRevision {
  464. continue
  465. }
  466. return err
  467. }
  468. // fetch response for serialized request
  469. get()
  470. // empty credentials or current auth info means no need to retry
  471. if ai.Revision == 0 || ai.Revision == s.authStore.Revision() {
  472. return nil
  473. }
  474. // avoid TOCTOU error, retry of the request is required.
  475. }
  476. }
  477. func (s *EtcdServer) processInternalRaftRequestOnce(ctx context.Context, r pb.InternalRaftRequest) (*applyResult, error) {
  478. ai := s.getAppliedIndex()
  479. ci := s.getCommittedIndex()
  480. if ci > ai+maxGapBetweenApplyAndCommitIndex {
  481. return nil, ErrTooManyRequests
  482. }
  483. r.Header = &pb.RequestHeader{
  484. ID: s.reqIDGen.Next(),
  485. }
  486. authInfo, err := s.AuthInfoFromCtx(ctx)
  487. if err != nil {
  488. return nil, err
  489. }
  490. if authInfo != nil {
  491. r.Header.Username = authInfo.Username
  492. r.Header.AuthRevision = authInfo.Revision
  493. }
  494. data, err := r.Marshal()
  495. if err != nil {
  496. return nil, err
  497. }
  498. if len(data) > int(s.Cfg.MaxRequestBytes) {
  499. return nil, ErrRequestTooLarge
  500. }
  501. id := r.ID
  502. if id == 0 {
  503. id = r.Header.ID
  504. }
  505. ch := s.w.Register(id)
  506. cctx, cancel := context.WithTimeout(ctx, s.Cfg.ReqTimeout())
  507. defer cancel()
  508. start := time.Now()
  509. s.r.Propose(cctx, data)
  510. proposalsPending.Inc()
  511. defer proposalsPending.Dec()
  512. select {
  513. case x := <-ch:
  514. return x.(*applyResult), nil
  515. case <-cctx.Done():
  516. proposalsFailed.Inc()
  517. s.w.Trigger(id, nil) // GC wait
  518. return nil, s.parseProposeCtxErr(cctx.Err(), start)
  519. case <-s.done:
  520. return nil, ErrStopped
  521. }
  522. }
  523. // Watchable returns a watchable interface attached to the etcdserver.
  524. func (s *EtcdServer) Watchable() mvcc.WatchableKV { return s.KV() }
  525. func (s *EtcdServer) linearizableReadLoop() {
  526. var rs raft.ReadState
  527. for {
  528. ctx := make([]byte, 8)
  529. binary.BigEndian.PutUint64(ctx, s.reqIDGen.Next())
  530. select {
  531. case <-s.readwaitc:
  532. case <-s.stopping:
  533. return
  534. }
  535. nextnr := newNotifier()
  536. s.readMu.Lock()
  537. nr := s.readNotifier
  538. s.readNotifier = nextnr
  539. s.readMu.Unlock()
  540. cctx, cancel := context.WithTimeout(context.Background(), s.Cfg.ReqTimeout())
  541. if err := s.r.ReadIndex(cctx, ctx); err != nil {
  542. cancel()
  543. if err == raft.ErrStopped {
  544. return
  545. }
  546. plog.Errorf("failed to get read index from raft: %v", err)
  547. nr.notify(err)
  548. continue
  549. }
  550. cancel()
  551. var (
  552. timeout bool
  553. done bool
  554. )
  555. for !timeout && !done {
  556. select {
  557. case rs = <-s.r.readStateC:
  558. done = bytes.Equal(rs.RequestCtx, ctx)
  559. if !done {
  560. // a previous request might time out. now we should ignore the response of it and
  561. // continue waiting for the response of the current requests.
  562. plog.Warningf("ignored out-of-date read index response (want %v, got %v)", rs.RequestCtx, ctx)
  563. }
  564. case <-time.After(s.Cfg.ReqTimeout()):
  565. plog.Warningf("timed out waiting for read index response")
  566. nr.notify(ErrTimeout)
  567. timeout = true
  568. case <-s.stopping:
  569. return
  570. }
  571. }
  572. if !done {
  573. continue
  574. }
  575. if ai := s.getAppliedIndex(); ai < rs.Index {
  576. select {
  577. case <-s.applyWait.Wait(rs.Index):
  578. case <-s.stopping:
  579. return
  580. }
  581. }
  582. // unblock all l-reads requested at indices before rs.Index
  583. nr.notify(nil)
  584. }
  585. }
  586. func (s *EtcdServer) linearizableReadNotify(ctx context.Context) error {
  587. s.readMu.RLock()
  588. nc := s.readNotifier
  589. s.readMu.RUnlock()
  590. // signal linearizable loop for current notify if it hasn't been already
  591. select {
  592. case s.readwaitc <- struct{}{}:
  593. default:
  594. }
  595. // wait for read state notification
  596. select {
  597. case <-nc.c:
  598. return nc.err
  599. case <-ctx.Done():
  600. return ctx.Err()
  601. case <-s.done:
  602. return ErrStopped
  603. }
  604. }
  605. func (s *EtcdServer) AuthInfoFromCtx(ctx context.Context) (*auth.AuthInfo, error) {
  606. if s.Cfg.ClientCertAuthEnabled {
  607. authInfo := s.AuthStore().AuthInfoFromTLS(ctx)
  608. if authInfo != nil {
  609. return authInfo, nil
  610. }
  611. }
  612. return s.AuthStore().AuthInfoFromCtx(ctx)
  613. }