CodeCow/PCAPdroid
1
0
Fork 0
mirror of https://github.com/emanuele-f/PCAPdroid.git synced 2026-06-16 21:10:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use memory-mapped file for geolocation lookups

Browse Source 
This saves ~14 MB of RAM usage due to the loading of mmdb files
This commit is contained in:
emanuele-f 2021-12-13 19:41:17 +01:00
parent 29612b389c
commit e270ee8670
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
app/src/main/java/com/emanuelef/remote_capture/Geolocation.java
View File

@ -25,8 +25,12 @@ import android.util.Log;
import com.emanuelef.remote_capture.model.Geomodel;
import com.maxmind.db.Reader;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.util.zip.GZIPInputStream;
@ -58,14 +62,14 @@ public class Geolocation {
private void openDb() {
try {
InputStream is = mContext.getResources().openRawResource(R.raw.dbip_country_lite_2021_11_mmdb_gz);
GZIPInputStream gis = new GZIPInputStream(is);
mCountryReader = new Reader(gis);
File countryFile = new File(mContext.getCacheDir() + "/dbip_country_lite.mmdb");
ungzip(R.raw.dbip_country_lite_2021_11_mmdb_gz, countryFile);
mCountryReader = new Reader(countryFile);
Log.d(TAG, "Country DB loaded: " + mCountryReader.getMetadata());
is = mContext.getResources().openRawResource(R.raw.dbip_asn_lite_2021_11_mmdb_gz);
gis = new GZIPInputStream(is);
mAsnReader = new Reader(gis);
File asnFile = new File(mContext.getCacheDir() + "/dbip_asn_lite.mmdb");
ungzip(R.raw.dbip_asn_lite_2021_11_mmdb_gz, asnFile);
mAsnReader = new Reader(asnFile);
Log.d(TAG, "ASN DB loaded: " + mAsnReader.getMetadata());
} catch (IOException e) {
e.printStackTrace();
@ -73,6 +77,18 @@ public class Geolocation {
}
}
private void ungzip(int resid, File dst) throws IOException {
try(InputStream is = new GZIPInputStream(mContext.getResources().openRawResource(resid))) {
try(BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(dst))) {
byte[] bytesIn = new byte[4096];
int read = 0;
while ((read = is.read(bytesIn)) != -1)
bos.write(bytesIn, 0, read);
}
}
}
public String getCountryCode(InetAddress addr) {
try {
Geomodel.CountryResult res = mCountryReader.get(addr, Geomodel.CountryResult.class);