src/net/http/disk_cache_based_quic_server_info.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef NET_HTTP_DISK_CACHE_BASED_QUIC_SERVER_INFO_H_
   6 #define NET_HTTP_DISK_CACHE_BASED_QUIC_SERVER_INFO_H_
   7
   8 #include <string>
   9
  10 #include "base/memory/ref_counted.h"
  11 #include "base/memory/weak_ptr.h"
  12 #include "base/threading/non_thread_safe.h"
  13 #include "net/base/completion_callback.h"
  14 #include "net/disk_cache/disk_cache.h"
  15 #include "net/quic/crypto/quic_server_info.h"
  16
  17 namespace net {
  18
  19 class HttpCache;
  20 class IOBuffer;
  21 class QuicServerId;
  22
  23 // DiskCacheBasedQuicServerInfo fetches information about a QUIC server from
  24 // our standard disk cache. Since the information is defined to be
  25 // non-sensitive, it's ok for us to keep it on disk.
  26 class NET_EXPORT_PRIVATE DiskCacheBasedQuicServerInfo
  27     : public QuicServerInfo,
  28       public NON_EXPORTED_BASE(base::NonThreadSafe) {
  29  public:
  30   DiskCacheBasedQuicServerInfo(const QuicServerId& server_id,
  31                                HttpCache* http_cache);
  32
  33   // QuicServerInfo implementation.
  34   virtual void Start() OVERRIDE;
  35   virtual int WaitForDataReady(const CompletionCallback& callback) OVERRIDE;
  36   virtual bool IsDataReady() OVERRIDE;
  37   virtual bool IsReadyToPersist() OVERRIDE;
  38   virtual void Persist() OVERRIDE;
  39
  40  private:
  41   struct CacheOperationDataShim;
  42   enum State {
  43     GET_BACKEND,
  44     GET_BACKEND_COMPLETE,
  45     OPEN,
  46     OPEN_COMPLETE,
  47     READ,
  48     READ_COMPLETE,
  49     WAIT_FOR_DATA_READY_DONE,
  50     CREATE_OR_OPEN,
  51     CREATE_OR_OPEN_COMPLETE,
  52     WRITE,
  53     WRITE_COMPLETE,
  54     SET_DONE,
  55     NONE,
  56   };
  57
  58   virtual ~DiskCacheBasedQuicServerInfo();
  59
  60   std::string key() const;
  61
  62   // The |unused| parameter is a small hack so that we can have the
  63   // CacheOperationDataShim object owned by the Callback that is created for
  64   // this method.  See comment above CacheOperationDataShim for details.
  65   void OnIOComplete(CacheOperationDataShim* unused, int rv);
  66
  67   int DoLoop(int rv);
  68
  69   int DoGetBackendComplete(int rv);
  70   int DoOpenComplete(int rv);
  71   int DoReadComplete(int rv);
  72   int DoWriteComplete(int rv);
  73   int DoCreateOrOpenComplete(int rv);
  74
  75   int DoGetBackend();
  76   int DoOpen();
  77   int DoRead();
  78   int DoWrite();
  79   int DoCreateOrOpen();
  80
  81   // DoWaitForDataReadyDone is the terminal state of the read operation.
  82   int DoWaitForDataReadyDone();
  83
  84   // DoSetDone is the terminal state of the write operation.
  85   int DoSetDone();
  86
  87   CacheOperationDataShim* data_shim_;  // Owned by |io_callback_|.
  88   CompletionCallback io_callback_;
  89   State state_;
  90   bool ready_;
  91   bool found_entry_;  // Controls the behavior of DoCreateOrOpen.
  92   std::string new_data_;
  93   const QuicServerId server_id_;
  94   HttpCache* const http_cache_;
  95   disk_cache::Backend* backend_;
  96   disk_cache::Entry* entry_;
  97   CompletionCallback user_callback_;
  98   scoped_refptr<IOBuffer> read_buffer_;
  99   scoped_refptr<IOBuffer> write_buffer_;
 100   std::string data_;
 101
 102   base::WeakPtrFactory<DiskCacheBasedQuicServerInfo> weak_factory_;
 103 };
 104
 105 }  // namespace net
 106
 107 #endif  // NET_HTTP_DISK_CACHE_BASED_QUIC_SERVER_INFO_H_