1/*
2 * Copyright (C) 2009 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
5 * use this file except in compliance with the License. You may obtain a copy of
6 * the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 * License for the specific language governing permissions and limitations under
14 * the License.
15 */
16
17package com.android.lowstoragetest;
18
19import android.app.Activity;
20import android.content.Context;
21
22import android.os.Bundle;
23import android.os.Environment;
24import android.os.StatFs;
25import android.util.Log;
26import android.view.KeyEvent;
27import android.view.View;
28import android.view.View.OnClickListener;
29import java.io.File;
30import java.io.FileOutputStream;
31import android.widget.TextView;
32import android.widget.Button;
33
34public class LowStorageTest extends Activity {
35    static final String TAG = "DiskFullTest";
36    static final long WAIT_FOR_FINISH = 5 * 60 * 60;
37    static final int NO_OF_BLOCKS_TO_FILL = 1000;
38    static final int BYTE_SIZE = 1024;
39    static final int WAIT_FOR_SYSTEM_UPDATE = 10000;
40
41    private int mBlockSize = 0;
42    private final Object fillUpDone = new Object();
43
44    /** Called when the activity is first created. */
45    @Override
46    public void onCreate(Bundle icicle) {
47        super.onCreate(icicle);
48        setContentView(R.layout.main);
49
50        // Update the current data info
51        File path = Environment.getDataDirectory();
52        StatFs stat = new StatFs(path.getPath());
53        int totalBlocks = stat.getBlockCount();
54        mBlockSize = (int) (stat.getBlockSize());
55        TextView startSizeTextView = (TextView) findViewById(R.id.totalsize);
56        startSizeTextView.setText(Long.toString((totalBlocks * mBlockSize) / BYTE_SIZE));
57        Button button = (Button) findViewById(R.id.button_run);
58        button.setOnClickListener(mStartListener);
59    }
60
61    View.OnClickListener mStartListener = new OnClickListener() {
62        public void onClick(View v) {
63            fillDataAndUpdateInfo();
64        }
65    };
66
67    public void fillDataAndUpdateInfo() {
68        updateInfo(this);
69    }
70
71    // Fill up 100% of the data partition
72    public void fillupdisk(Context context) {
73        final Context contextfill = context;
74        new Thread() {
75            @Override
76            public void run() {
77                try {
78                    // Fill up all the memory
79                    File path = Environment.getDataDirectory();
80                    StatFs stat = new StatFs(path.getPath());
81                    int totalBlocks = stat.getBlockCount();
82                    int noOfBlockToFill = stat.getAvailableBlocks();
83                    FileOutputStream fs =
84                            contextfill.openFileOutput("testdata", Context.MODE_APPEND);
85                    for (int i = 0; i < (noOfBlockToFill / NO_OF_BLOCKS_TO_FILL); i++) {
86                        byte buf[] = new byte[mBlockSize * NO_OF_BLOCKS_TO_FILL];
87                        fs.write(buf);
88                        fs.flush();
89                    }
90
91                    // Fill up the last few block
92                    byte buf[] = new byte[(noOfBlockToFill % NO_OF_BLOCKS_TO_FILL) * mBlockSize];
93                    fs.write(buf);
94                    fs.flush();
95                    fs.close();
96
97                    // Finished, update the info
98                    synchronized (fillUpDone) {
99                        fillUpDone.notify();
100                    }
101                } catch (Exception e) {
102                    Log.v(TAG, e.toString());
103                }
104            }
105        }.start();
106    }
107
108    public void updateInfo(Context context) {
109        fillupdisk(this);
110        synchronized (fillUpDone) {
111            try {
112                fillUpDone.wait(WAIT_FOR_FINISH);
113            } catch (Exception e) {
114                Log.v(TAG, "wait was interrupted.");
115            }
116        }
117        try {
118            // The stat didn't relect the correct data right away
119            // put some extra time to make sure if get the right size.
120            Thread.sleep(WAIT_FOR_SYSTEM_UPDATE);
121            File path = Environment.getDataDirectory();
122            StatFs stat = new StatFs(path.getPath());
123            long availableBlocks = stat.getAvailableBlocks();
124            TextView freeSizeTextView = (TextView) findViewById(R.id.freesize);
125            freeSizeTextView.setText(Long.toString((availableBlocks * mBlockSize) / BYTE_SIZE));
126            TextView statusTextView = (TextView) findViewById(R.id.status);
127            statusTextView.setText("Finished. You can start the test now.");
128        } catch (Exception e) {
129            Log.v(TAG, e.toString());
130        }
131    }
132}
133