Add a "session only" policy to the content settings.

chrome/browser/cookie_modal_dialog.cc

 // 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.
 
 #include "chrome/browser/cookie_modal_dialog.h"
 
 #include "app/message_box_flags.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/host_content_settings_map.h"
 #include "chrome/browser/pref_service.h"
@@ skipping 68 matching lines @@
 
 CookiePromptModalDialog::~CookiePromptModalDialog() {
 }
 
 bool CookiePromptModalDialog::IsValid() {
   ContentSetting content_setting =
       host_content_settings_map_->GetContentSetting(
           origin_, CONTENT_SETTINGS_TYPE_COOKIES);
   if (content_setting != CONTENT_SETTING_ASK) {
     if (content_setting == CONTENT_SETTING_ALLOW) {
-      // If it's remembered as allow, then we assume session_expire is false.
       AllowSiteData(false, false);
+    } else if (content_setting == CONTENT_SETTING_SESSION_ONLY) {
+      AllowSiteData(false, true);
     } else {
       DCHECK(content_setting == CONTENT_SETTING_BLOCK);
       BlockSiteData(false);
     }
     return false;
   }
   return !skip_this_dialog_;
 }
 
 void CookiePromptModalDialog::AllowSiteData(bool remember,
                                             bool session_expire) {
   DCHECK(!remember || DecisionPersistable());
   if (remember && DecisionPersistable()) {
     // Make sure there is no entry that would override the pattern we are about
     // to insert for exactly this URL.
     host_content_settings_map_->SetContentSetting(
         HostContentSettingsMap::Pattern::FromURLNoWildcard(origin_),
         CONTENT_SETTINGS_TYPE_COOKIES, CONTENT_SETTING_DEFAULT);
     host_content_settings_map_->SetContentSetting(
         HostContentSettingsMap::Pattern::FromURL(origin_),
-        CONTENT_SETTINGS_TYPE_COOKIES, CONTENT_SETTING_ALLOW);
+        CONTENT_SETTINGS_TYPE_COOKIES,
+        session_expire ? CONTENT_SETTING_SESSION_ONLY : CONTENT_SETTING_ALLOW);
   }
 
   if (delegate_) {
     delegate_->AllowSiteData(session_expire);
     delegate_ = NULL;  // It can be deleted at any point now.
   }
 }
 
 void CookiePromptModalDialog::BlockSiteData(bool remember) {
   DCHECK(!remember || DecisionPersistable());
@@ skipping 21 matching lines @@
 
 int CookiePromptModalDialog::GetDialogButtons() {
   // Enable the automation interface to accept/dismiss this dialog.
   return MessageBoxFlags::DIALOGBUTTON_OK |
          MessageBoxFlags::DIALOGBUTTON_CANCEL;
 }
 
 bool CookiePromptModalDialog::DecisionPersistable() {
   return !host_content_settings_map_->IsOffTheRecord();
 }