Home | History | Annotate | Download | only in util
      1 /*
      2  * Copyright (C) 2011 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 
     17 package android.cts.util;
     18 
     19 import java.io.ByteArrayOutputStream;
     20 import java.io.File;
     21 import java.io.FileOutputStream;
     22 import java.io.IOException;
     23 import java.io.InputStream;
     24 
     25 /** Bits and pieces copied from hidden API of android.os.FileUtils. */
     26 public class FileUtils {
     27 
     28     public static final int S_IFMT  = 0170000;
     29     public static final int S_IFSOCK = 0140000;
     30     public static final int S_IFLNK = 0120000;
     31     public static final int S_IFREG = 0100000;
     32     public static final int S_IFBLK = 0060000;
     33     public static final int S_IFDIR = 0040000;
     34     public static final int S_IFCHR = 0020000;
     35     public static final int S_IFIFO = 0010000;
     36 
     37     public static final int S_ISUID = 0004000;
     38     public static final int S_ISGID = 0002000;
     39     public static final int S_ISVTX = 0001000;
     40 
     41     public static final int S_IRWXU = 00700;
     42     public static final int S_IRUSR = 00400;
     43     public static final int S_IWUSR = 00200;
     44     public static final int S_IXUSR = 00100;
     45 
     46     public static final int S_IRWXG = 00070;
     47     public static final int S_IRGRP = 00040;
     48     public static final int S_IWGRP = 00020;
     49     public static final int S_IXGRP = 00010;
     50 
     51     public static final int S_IRWXO = 00007;
     52     public static final int S_IROTH = 00004;
     53     public static final int S_IWOTH = 00002;
     54     public static final int S_IXOTH = 00001;
     55 
     56     static {
     57         System.loadLibrary("cts_jni");
     58     }
     59 
     60     public static class FileStatus {
     61 
     62         public int dev;
     63         public int ino;
     64         public int mode;
     65         public int nlink;
     66         public int uid;
     67         public int gid;
     68         public int rdev;
     69         public long size;
     70         public int blksize;
     71         public long blocks;
     72         public long atime;
     73         public long mtime;
     74         public long ctime;
     75 
     76         public boolean hasModeFlag(int flag) {
     77             if (((S_IRWXU | S_IRWXG | S_IRWXO) & flag) != flag) {
     78                 throw new IllegalArgumentException("Inappropriate flag " + flag);
     79             }
     80             return (mode & flag) == flag;
     81         }
     82 
     83         public boolean isOfType(int type) {
     84             if ((type & S_IFMT) != type) {
     85                 throw new IllegalArgumentException("Unknown type " + type);
     86             }
     87             return (mode & S_IFMT) == type;
     88         }
     89     }
     90 
     91     /**
     92      * @param path of the file to stat
     93      * @param status object to set the fields on
     94      * @param statLinks or don't stat links (lstat vs stat)
     95      * @return whether or not we were able to stat the file
     96      */
     97     public native static boolean getFileStatus(String path, FileStatus status, boolean statLinks);
     98 
     99     public native static String getUserName(int uid);
    100 
    101     public native static String getGroupName(int gid);
    102 
    103     public native static int setPermissions(String file, int mode);
    104 
    105     /**
    106      * Copy data from a source stream to destFile.
    107      * Return true if succeed, return false if failed.
    108      */
    109     public static boolean copyToFile(InputStream inputStream, File destFile) {
    110         try {
    111             if (destFile.exists()) {
    112                 destFile.delete();
    113             }
    114             FileOutputStream out = new FileOutputStream(destFile);
    115             try {
    116                 byte[] buffer = new byte[4096];
    117                 int bytesRead;
    118                 while ((bytesRead = inputStream.read(buffer)) >= 0) {
    119                     out.write(buffer, 0, bytesRead);
    120                 }
    121             } finally {
    122                 out.flush();
    123                 try {
    124                     out.getFD().sync();
    125                 } catch (IOException e) {
    126                 }
    127                 out.close();
    128             }
    129             return true;
    130         } catch (IOException e) {
    131             return false;
    132         }
    133     }
    134 
    135     public static void createFile(File file, int numBytes) throws IOException {
    136         File parentFile = file.getParentFile();
    137         if (parentFile != null) {
    138             parentFile.mkdirs();
    139         }
    140         byte[] buffer = new byte[numBytes];
    141         FileOutputStream output = new FileOutputStream(file);
    142         try {
    143             output.write(buffer);
    144         } finally {
    145             output.close();
    146         }
    147     }
    148 
    149     public static byte[] readInputStreamFully(InputStream is) {
    150         ByteArrayOutputStream os = new ByteArrayOutputStream();
    151         byte[] buffer = new byte[32768];
    152         int count;
    153         try {
    154             while ((count = is.read(buffer)) != -1) {
    155                 os.write(buffer, 0, count);
    156             }
    157             is.close();
    158         } catch (IOException e) {
    159             throw new RuntimeException(e);
    160         }
    161         return os.toByteArray();
    162     }
    163 }
    164