Merge 44735 - With the ChromeFrame moniker patch on, the data cache maintaine...

chrome_frame/urlmon_bind_status_callback.h

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_FRAME_URLMON_BIND_STATUS_CALLBACK_H_
 #define CHROME_FRAME_URLMON_BIND_STATUS_CALLBACK_H_
 
 #include <atlbase.h>
 #include <atlcom.h>
 
 #include "chrome_frame/bind_status_callback_impl.h"
 #include "chrome_frame/stream_impl.h"
 
 
 // A fake stream class to serve cached data to arbitrary
 // IBindStatusCallback
 class CacheStream : public CComObjectRoot, public StreamImpl {
  public:
   BEGIN_COM_MAP(CacheStream)
     COM_INTERFACE_ENTRY(IStream)
     COM_INTERFACE_ENTRY(ISequentialStream)
   END_COM_MAP()
 
-  CacheStream() : cache_(NULL), size_(0), position_(0) {
+  CacheStream() : cache_(NULL), size_(0), position_(0), eof_(false) {
   }
-  void Initialize(const char* cache, size_t size);
+  void Initialize(const char* cache, size_t size, bool eof);
   static HRESULT BSCBFeedData(IBindStatusCallback* bscb, const char* data,
                               size_t size, CLIPFORMAT clip_format,
-                              size_t flags);
+                              size_t flags, bool eof);
 
   // IStream overrides
   STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read);
 
  protected:
   const char* cache_;
   size_t size_;
   size_t position_;
+  bool eof_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(CacheStream);
 };
 
 // Utility class for data sniffing
 class SniffData {
  public:
-  SniffData() : renderer_type_(OTHER), size_(0) {}
+  SniffData() : renderer_type_(OTHER), size_(0), eof_(false) {}
 
   enum RendererType {
     UNDETERMINED,
     CHROME,
     OTHER
   };
 
   bool InitializeCache(const std::wstring& url);
   HRESULT ReadIntoCache(IStream* stream, bool force_determination);
   HRESULT DrainCache(IBindStatusCallback* bscb, DWORD bscf,
@@ skipping 18 matching lines @@
   bool is_cache_valid() {
     return (size_ != 0);
   }
 
   ScopedComPtr<IStream> cache_;
   std::wstring url_;
   RendererType renderer_type_;
   size_t size_;
 
   static const size_t kMaxSniffSize = 2 * 1024;
+  bool eof_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(SniffData);
 };
 
 // A wrapper for bind status callback in IMoniker::BindToStorage
 class BSCBStorageBind : public BSCBImpl {
  public:
   typedef BSCBImpl CallbackImpl;
   BSCBStorageBind() : clip_format_(CF_NULL) {}
@@ skipping 25 matching lines @@
   };
 
   std::vector<Progress> saved_progress_;
   CLIPFORMAT clip_format_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(BSCBStorageBind);
 };
 
 #endif  // CHROME_FRAME_URLMON_BIND_STATUS_CALLBACK_H_