v3_server.go 23 KB

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