1/*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of 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,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.launcher3;
18
19import android.app.ActivityManager;
20import android.app.Service;
21import android.content.Context;
22import android.content.Intent;
23import android.os.*;
24import android.util.Log;
25import android.util.LongSparseArray;
26
27import java.util.ArrayList;
28import java.util.List;
29
30public class MemoryTracker extends Service {
31    public static final String TAG = MemoryTracker.class.getSimpleName();
32    public static final String ACTION_START_TRACKING = "com.android.launcher3.action.START_TRACKING";
33
34    private static final long UPDATE_RATE = 5000;
35
36    private static final int MSG_START = 1;
37    private static final int MSG_STOP = 2;
38    private static final int MSG_UPDATE = 3;
39
40    public static class ProcessMemInfo {
41        public int pid;
42        public String name;
43        public long startTime;
44        public long currentPss, currentUss;
45        public long[] pss = new long[256];
46        public long[] uss = new long[256];
47            //= new Meminfo[(int) (30 * 60 / (UPDATE_RATE / 1000))]; // 30 minutes
48        public long max = 1;
49        public int head = 0;
50        public ProcessMemInfo(int pid, String name, long start) {
51            this.pid = pid;
52            this.name = name;
53            this.startTime = start;
54        }
55        public long getUptime() {
56            return System.currentTimeMillis() - startTime;
57        }
58    };
59    public final LongSparseArray<ProcessMemInfo> mData = new LongSparseArray<ProcessMemInfo>();
60    public final ArrayList<Long> mPids = new ArrayList<Long>();
61    private int[] mPidsArray = new int[0];
62    private final Object mLock = new Object();
63
64    Handler mHandler = new Handler() {
65        @Override
66        public void handleMessage(Message m) {
67            switch (m.what) {
68                case MSG_START:
69                    mHandler.removeMessages(MSG_UPDATE);
70                    mHandler.sendEmptyMessage(MSG_UPDATE);
71                    break;
72                case MSG_STOP:
73                    mHandler.removeMessages(MSG_UPDATE);
74                    break;
75                case MSG_UPDATE:
76                    update();
77                    mHandler.removeMessages(MSG_UPDATE);
78                    mHandler.sendEmptyMessageDelayed(MSG_UPDATE, UPDATE_RATE);
79                    break;
80            }
81        }
82    };
83
84    ActivityManager mAm;
85
86    public static void startTrackingMe(Context context, String name) {
87        context.startService(new Intent(context, MemoryTracker.class)
88                .setAction(MemoryTracker.ACTION_START_TRACKING)
89                .putExtra("pid", android.os.Process.myPid())
90                .putExtra("name", name)
91        );
92    }
93
94    public ProcessMemInfo getMemInfo(int pid) {
95        return mData.get(pid);
96    }
97
98    public int[] getTrackedProcesses() {
99        return mPidsArray;
100    }
101
102    public void startTrackingProcess(int pid, String name, long start) {
103        synchronized (mLock) {
104            final Long lpid = new Long(pid);
105
106            if (mPids.contains(lpid)) return;
107
108            mPids.add(lpid);
109            updatePidsArrayL();
110
111            mData.put(pid, new ProcessMemInfo(pid, name, start));
112        }
113    }
114
115    void updatePidsArrayL() {
116        final int N = mPids.size();
117        mPidsArray = new int[N];
118        StringBuffer sb = new StringBuffer("Now tracking processes: ");
119        for (int i=0; i<N; i++) {
120            final int p = mPids.get(i).intValue();
121            mPidsArray[i] = p;
122            sb.append(p); sb.append(" ");
123        }
124        Log.v(TAG, sb.toString());
125    }
126
127    void update() {
128        synchronized (mLock) {
129            Debug.MemoryInfo[] dinfos = mAm.getProcessMemoryInfo(mPidsArray);
130            for (int i=0; i<dinfos.length; i++) {
131                Debug.MemoryInfo dinfo = dinfos[i];
132                if (i > mPids.size()) {
133                    Log.e(TAG, "update: unknown process info received: " + dinfo);
134                    break;
135                }
136                final long pid = mPids.get(i).intValue();
137                final ProcessMemInfo info = mData.get(pid);
138                info.head = (info.head+1) % info.pss.length;
139                info.pss[info.head] = info.currentPss = dinfo.getTotalPss();
140                info.uss[info.head] = info.currentUss = dinfo.getTotalPrivateDirty();
141                if (info.currentPss > info.max) info.max = info.currentPss;
142                if (info.currentUss > info.max) info.max = info.currentUss;
143                // Log.v(TAG, "update: pid " + pid + " pss=" + info.currentPss + " uss=" + info.currentUss);
144                if (info.currentPss == 0) {
145                    Log.v(TAG, "update: pid " + pid + " has pss=0, it probably died");
146                    mData.remove(pid);
147                }
148            }
149            for (int i=mPids.size()-1; i>=0; i--) {
150                final long pid = mPids.get(i).intValue();
151                if (mData.get(pid) == null) {
152                    mPids.remove(i);
153                    updatePidsArrayL();
154                }
155            }
156        }
157    }
158
159    @Override
160    public void onCreate() {
161        mAm = (ActivityManager) getSystemService(Context.ACTIVITY_SERVICE);
162
163        // catch up in case we crashed but other processes are still running
164        List<ActivityManager.RunningServiceInfo> svcs = mAm.getRunningServices(256);
165        for (ActivityManager.RunningServiceInfo svc : svcs) {
166            if (svc.service.getPackageName().equals(getPackageName())) {
167                Log.v(TAG, "discovered running service: " + svc.process + " (" + svc.pid + ")");
168                startTrackingProcess(svc.pid, svc.process,
169                        System.currentTimeMillis() - (SystemClock.elapsedRealtime() - svc.activeSince));
170            }
171        }
172
173        List<ActivityManager.RunningAppProcessInfo> procs = mAm.getRunningAppProcesses();
174        for (ActivityManager.RunningAppProcessInfo proc : procs) {
175            final String pname = proc.processName;
176            if (pname.startsWith(getPackageName())) {
177                Log.v(TAG, "discovered other running process: " + pname + " (" + proc.pid + ")");
178                startTrackingProcess(proc.pid, pname, System.currentTimeMillis());
179            }
180        }
181    }
182
183    @Override
184    public void onDestroy() {
185        mHandler.sendEmptyMessage(MSG_STOP);
186    }
187
188    @Override
189    public int onStartCommand(Intent intent, int flags, int startId) {
190        Log.v(TAG, "Received start id " + startId + ": " + intent);
191
192        if (intent != null) {
193            if (ACTION_START_TRACKING.equals(intent.getAction())) {
194                final int pid = intent.getIntExtra("pid", -1);
195                final String name = intent.getStringExtra("name");
196                final long start = intent.getLongExtra("start", System.currentTimeMillis());
197                startTrackingProcess(pid, name, start);
198            }
199        }
200
201        mHandler.sendEmptyMessage(MSG_START);
202
203        return START_STICKY;
204    }
205
206    public class MemoryTrackerInterface extends Binder {
207        MemoryTracker getService() {
208            return MemoryTracker.this;
209        }
210    }
211
212    private final IBinder mBinder = new MemoryTrackerInterface();
213
214    public IBinder onBind(Intent intent) {
215        mHandler.sendEmptyMessage(MSG_START);
216
217        return mBinder;
218    }
219}
220