// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package org.chromium.chrome.browser.cookies;
import org.jni_zero.CalledByNative;
import org.jni_zero.NativeMethods;
import org.chromium.base.ContextUtils;
import org.chromium.base.ImportantFileWriterAndroid;
import org.chromium.base.Log;
import org.chromium.base.ThreadUtils;
import org.chromium.base.task.AsyncTask;
import org.chromium.base.task.BackgroundOnlyAsyncTask;
import org.chromium.chrome.browser.crypto.CipherFactory;
import org.chromium.chrome.browser.profiles.ProfileManager;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import javax.crypto.Cipher;
import javax.crypto.CipherInputStream;
import javax.crypto.CipherOutputStream;
/**
* Responsible for fetching, (de)serializing, and restoring cookies between the CookieJar and an
* encrypted file storage.
*
* These methods are used for incognito only. Incognito cookies are kept separately from other
* profiles' cookies, and are normally not persisted (on most platforms). However, on Android we
* serialize and store them to avoid logging the user out of their accounts being used in incognito,
* if the app is killed e.g. while in the background.
*/
public class CookiesFetcher {
/** The default file name for the encrypted cookies storage. */
private static final String DEFAULT_COOKIE_FILE_NAME = "COOKIES.DAT";
/** Used for logging. */
private static final String TAG = "CookiesFetcher";
private CookiesFetcher() {}
/**
* Fetches the cookie file's path on demand to prevent IO on the main thread.
*
* @return Path to the cookie file.
*/
private static String fetchFileName() {
assert !ThreadUtils.runningOnUiThread();
return ContextUtils.getApplicationContext()
.getFileStreamPath(DEFAULT_COOKIE_FILE_NAME)
.getAbsolutePath();
}
/** Asynchronously fetches cookies from the incognito profile and saves them to a file. */
public static void persistCookies() {
CookiesFetcherJni.get().persistCookies();
}
/**
* If an incognito profile exists, synchronously fetch cookies from the file specified and
* populate the incognito profile with it. Otherwise deletes the file and does not restore the
* cookies.
*/
public static void restoreCookies() {
if (!ProfileManager.getLastUsedRegularProfile().hasPrimaryOTRProfile()) {
scheduleDeleteCookies();
return;
}
new AsyncTask<List<CanonicalCookie>>() {
@Override
protected List<CanonicalCookie> doInBackground() {
// Read cookies from disk on a background thread to avoid strict mode violations.
List<CanonicalCookie> cookies = new ArrayList<CanonicalCookie>();
DataInputStream in = null;
try {
Cipher cipher = CipherFactory.getInstance().getCipher(Cipher.DECRYPT_MODE);
if (cipher == null) {
// Something is wrong. Can't encrypt, don't restore cookies.
return cookies;
}
File fileIn = new File(fetchFileName());
if (!fileIn.exists()) return cookies; // Nothing to read
FileInputStream streamIn = new FileInputStream(fileIn);
in = new DataInputStream(new CipherInputStream(streamIn, cipher));
cookies = CanonicalCookie.readListFromStream(in);
// The Cookie File should not be restored again. It'll be overwritten
// on the next onPause.
scheduleDeleteCookies();
} catch (Throwable t) {
Log.w(TAG, "Error restoring cookies.", t);
} finally {
try {
if (in != null) in.close();
} catch (Throwable t) {
Log.w(TAG, "Error restoring cookies.", t);
}
}
return cookies;
}
@Override
protected void onPostExecute(List<CanonicalCookie> cookies) {
// We can only access cookies and profiles on the UI thread.
for (CanonicalCookie cookie : cookies) {
CookiesFetcherJni.get()
.restoreCookies(
cookie.getName(),
cookie.getValue(),
cookie.getDomain(),
cookie.getPath(),
cookie.getCreationDate(),
cookie.getExpirationDate(),
cookie.getLastAccessDate(),
cookie.getLastUpdateDate(),
cookie.isSecure(),
cookie.isHttpOnly(),
cookie.getSameSite(),
cookie.getPriority(),
cookie.getPartitionKey(),
cookie.sourceScheme(),
cookie.sourcePort(),
cookie.sourceType());
}
}
}.executeOnExecutor(AsyncTask.SERIAL_EXECUTOR);
}
/** Delete the cookies file. Called when we detect that all incognito tabs have been closed. */
public static void scheduleDeleteCookies() {
new BackgroundOnlyAsyncTask<Void>() {
@Override
protected Void doInBackground() {
File cookiesFile = new File(fetchFileName());
if (cookiesFile.exists()) {
if (!cookiesFile.delete()) {
Log.e(TAG, "Failed to delete " + cookiesFile.getName());
}
}
return null;
}
}.executeOnExecutor(AsyncTask.SERIAL_EXECUTOR);
}
@CalledByNative
private static CanonicalCookie createCookie(
String name,
String value,
String domain,
String path,
long creation,
long expiration,
long lastAccess,
long lastUpdate,
boolean secure,
boolean httpOnly,
int sameSite,
int priority,
String partitionKey,
int sourceScheme,
int sourcePort,
int sourceType) {
return new CanonicalCookie(
name,
value,
domain,
path,
creation,
expiration,
lastAccess,
lastUpdate,
secure,
httpOnly,
sameSite,
priority,
partitionKey,
sourceScheme,
sourcePort,
sourceType);
}
@CalledByNative
private static void onCookieFetchFinished(final CanonicalCookie[] cookies) {
// Cookies fetching requires operations with the profile and must be
// done in the main thread. Once that is done, do the save to disk
// part in {@link BackgroundOnlyAsyncTask} to avoid strict mode violations.
new BackgroundOnlyAsyncTask<Void>() {
@Override
protected Void doInBackground() {
saveFetchedCookiesToDisk(cookies);
return null;
}
}.executeOnExecutor(AsyncTask.SERIAL_EXECUTOR);
}
private static void saveFetchedCookiesToDisk(CanonicalCookie[] cookies) {
DataOutputStream out = null;
try {
Cipher cipher = CipherFactory.getInstance().getCipher(Cipher.ENCRYPT_MODE);
if (cipher == null) {
// Something is wrong. Can't encrypt, don't save cookies.
return;
}
ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
CipherOutputStream cipherOut = new CipherOutputStream(byteOut, cipher);
out = new DataOutputStream(cipherOut);
CanonicalCookie.saveListToStream(out, cookies);
out.close();
ImportantFileWriterAndroid.writeFileAtomically(fetchFileName(), byteOut.toByteArray());
out = null;
} catch (IOException e) {
Log.w(TAG, "IOException during Cookie Fetch");
} catch (Throwable t) {
Log.w(TAG, "Error storing cookies.", t);
} finally {
try {
if (out != null) out.close();
} catch (IOException e) {
Log.w(TAG, "IOException during Cookie Fetch");
}
}
}
@CalledByNative
private static CanonicalCookie[] createCookiesArray(int size) {
return new CanonicalCookie[size];
}
@NativeMethods
interface Natives {
void persistCookies();
void restoreCookies(
String name,
String value,
String domain,
String path,
long creation,
long expiration,
long lastAccess,
long lastUpdate,
boolean secure,
boolean httpOnly,
int sameSite,
int priority,
String partitionKey,
int sourceScheme,
int sourcePort,
int sourceType);
}
}