v3_server.go 22 KB

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