| /* |
| * Copyright (C) 2020 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.keychain; |
| |
| import android.annotation.Nullable; |
| import android.security.CredentialManagementApp; |
| import android.util.AtomicFile; |
| import android.util.Log; |
| import android.util.Xml; |
| |
| import com.android.internal.util.FastXmlSerializer; |
| |
| import org.xmlpull.v1.XmlPullParser; |
| import org.xmlpull.v1.XmlPullParserException; |
| import org.xmlpull.v1.XmlSerializer; |
| |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.nio.charset.StandardCharsets; |
| |
| public class KeyChainStateStorage { |
| |
| private static final String TAG = "KeyChain"; |
| private static final String TAG_CREDENTIAL_MANAGEMENT_APP = "credential-management-app"; |
| |
| private final File mDirectory; |
| |
| public KeyChainStateStorage(File directory) { |
| mDirectory = directory; |
| } |
| |
| @Nullable |
| public CredentialManagementApp loadCredentialManagementApp() { |
| CredentialManagementApp credentialManagementApp = null; |
| AtomicFile file = getCredentialManagementFile(); |
| FileInputStream stream = null; |
| try { |
| stream = file.openRead(); |
| XmlPullParser parser = Xml.newPullParser(); |
| parser.setInput(stream, StandardCharsets.UTF_8.name()); |
| int type; |
| while ((type = parser.next()) != XmlPullParser.START_TAG |
| && type != XmlPullParser.END_DOCUMENT) { |
| } |
| String tag = parser.getName(); |
| if (TAG_CREDENTIAL_MANAGEMENT_APP.equals(tag)) { |
| credentialManagementApp = CredentialManagementApp.readFromXml(parser); |
| } |
| } catch (XmlPullParserException | IOException e) { |
| Log.e(TAG, "Failed to load state", e); |
| } finally { |
| try { |
| if (stream != null) { |
| stream.close(); |
| } |
| } catch (IOException e) { |
| } |
| } |
| return credentialManagementApp; |
| } |
| |
| public void saveCredentialManagementApp( |
| @Nullable CredentialManagementApp credentialManagementApp) { |
| AtomicFile file = getCredentialManagementFile(); |
| FileOutputStream stream; |
| try { |
| stream = file.startWrite(); |
| } catch (IOException e) { |
| Log.e(TAG, "Failed to write state " + e); |
| return; |
| } |
| try { |
| XmlSerializer out = new FastXmlSerializer(); |
| out.setOutput(stream, StandardCharsets.UTF_8.name()); |
| out.startDocument(null, true); |
| |
| if (credentialManagementApp != null) { |
| out.startTag(null, TAG_CREDENTIAL_MANAGEMENT_APP); |
| credentialManagementApp.writeToXml(out); |
| out.endTag(null, TAG_CREDENTIAL_MANAGEMENT_APP); |
| } |
| out.endDocument(); |
| file.finishWrite(stream); |
| stream.close(); |
| } catch (IOException e) { |
| Log.e(TAG, "Failed to store state"); |
| file.failWrite(stream); |
| } |
| } |
| |
| private AtomicFile getCredentialManagementFile() { |
| File file = new File(mDirectory, "credential-management-app.xml"); |
| if (!file.exists()) { |
| try { |
| file.createNewFile(); |
| } catch (IOException e) { |
| } |
| } |
| return new AtomicFile(file); |
| } |
| } |