Image.java

Go to the documentation of this file.

/*
 * Sleuth Kit Data Model
 *
 * Copyright 2011-2018 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 *
 * 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 org.sleuthkit.datamodel;

import java.text.MessageFormat;
import java.util.ResourceBundle;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.io.File;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class Image extends AbstractContent implements DataSource {
        //data about image

        private final long type, ssize;
        private long size;
        private final String[] paths;
        private volatile long imageHandle = 0;
        private final String deviceId, timezone;
        private String md5, sha1, sha256;
        private static ResourceBundle bundle = ResourceBundle.getBundle("org.sleuthkit.datamodel.Bundle");

        private static final Logger LOGGER = Logger.getLogger(Image.class.getName());

        @Deprecated
        protected Image(SleuthkitCase db, long obj_id, long type, long ssize, String name, String[] paths, String timezone, String md5) throws TskCoreException {
                super(db, obj_id, name);
                this.deviceId = "";
                this.type = type;
                this.ssize = ssize;
                this.paths = paths;
                this.timezone = timezone;
                this.size = 0;
                this.md5 = md5;
                this.sha1 = "";
                this.sha256 = "";
        }

        Image(SleuthkitCase db, long obj_id, long type, String deviceId, long ssize, String name, String[] paths, String timezone, 
                        String md5, String sha1, String sha256, long size) throws TskCoreException {
                super(db, obj_id, name);
                this.deviceId = deviceId;
                this.type = type;
                this.ssize = ssize;
                this.paths = paths;
                this.timezone = timezone;
                this.size = size;
                this.md5 = md5;
                this.sha1 = sha1;
                this.sha256 = sha256;
        }

        public synchronized long getImageHandle() throws TskCoreException {
                if (paths.length == 0) {
                        throw new TskCoreException("Image has no associated paths");
                }
                
                if (imageHandle == 0) {
                        imageHandle = SleuthkitJNI.openImage(paths, (int)ssize, getSleuthkitCase());
                }

                return imageHandle;
        }

        @Override
        public Content getDataSource() {
                return this;
        }

        @Override
        public void close() {
                //frees nothing, as we are caching image handles
        }

        @Override
        public void finalize() throws Throwable {
                try {
                        if (imageHandle != 0) {
                                // SleuthkitJNI.closeImg(imageHandle); // closeImg is currently a no-op
                                imageHandle = 0;
                        }
                } finally {
                        super.finalize();
                }
        }

        @Override
        public int read(byte[] buf, long offset, long len) throws TskCoreException {
                // If there are no paths, don't attempt to read the image
                if (paths.length == 0) {
                        return 0;
                }
                
                // read from the image
                return SleuthkitJNI.readImg(getImageHandle(), buf, offset, len);
        }

        @Override
        public long getSize() {
                if (size == 0) {
                        try {
                                if (paths.length > 0) {
                                        //should always had at least one path 
                                        size = SleuthkitJNI.findDeviceSize(paths[0]);
                                }
                        } catch (TskCoreException ex) {
                                Logger.getLogger(Image.class.getName()).log(Level.SEVERE, "Could not find image size, image: " + this.getId(), ex); //NON-NLS
                        }
                }
                return size;
        }

        //Methods for retrieval of meta-data attributes
        public TskData.TSK_IMG_TYPE_ENUM getType() {
                return TskData.TSK_IMG_TYPE_ENUM.valueOf(type);
        }

        public long getSsize() {
                return ssize;
        }

        @Override
        public String getUniquePath() throws TskCoreException {
                return "/img_" + getName(); //NON-NLS
        }

        public String[] getPaths() {
                return paths;
        }

        public List<VolumeSystem> getVolumeSystems() throws TskCoreException {

                List<Content> children = getChildren();
                List<VolumeSystem> vs = new ArrayList<VolumeSystem>();
                for (Content child : children) {
                        if (child instanceof VolumeSystem) {
                                vs.add((VolumeSystem) child);
                        }
                }

                return vs;
        }

        public List<Volume> getVolumes() throws TskCoreException {

                List<Content> children = getChildren();
                List<Volume> volumes = new ArrayList<Volume>();
                for (Content child : children) {
                        if (child instanceof Volume) {
                                volumes.add((Volume) child);
                        }
                }

                return volumes;
        }

        public List<FileSystem> getFileSystems() throws TskCoreException {
                List<FileSystem> fs = new ArrayList<FileSystem>();
                fs.addAll(getSleuthkitCase().getFileSystems(this));
                return fs;
        }

        @Override
        public String getTimeZone() {
                return timezone;
        }

        @Override
        public <T> T accept(SleuthkitItemVisitor<T> v) {
                return v.visit(this);
        }

        @Override
        public <T> T accept(ContentVisitor<T> v) {
                return v.visit(this);
        }

        @Override
        public List<Content> getChildren() throws TskCoreException {
                return getSleuthkitCase().getImageChildren(this);
        }

        @Override
        public List<Long> getChildrenIds() throws TskCoreException {
                return getSleuthkitCase().getImageChildrenIds(this);
        }

        @Override
        public String toString(boolean preserveState) {
                return super.toString(preserveState) + "Image [\t" + "\t" + "paths " + Arrays.toString(paths) + "\t" + "size " + size + "\t" + "ssize " + ssize + "\t" + "timezone " + timezone + "\t" + "type " + type + "]\t"; //NON-NLS
        }

        public Boolean imageFileExists() {
                if (paths.length > 0) {
                        File imageFile = new File(paths[0]);
                        return imageFile.exists();
                }

                return false;
        }

        public String verifyImageSize() {
                Logger logger1 = Logger.getLogger("verifyImageSizes"); //NON-NLS
                String errorString = "";
                try {
                        List<VolumeSystem> volumeSystems = getVolumeSystems();
                        for (VolumeSystem vs : volumeSystems) {
                                List<Volume> volumes = vs.getVolumes();
                                for (Volume v : volumes) {
                                        byte[] buf = new byte[512];
                                        long endOffset = (v.getStart() + v.getLength()) * 512 - 512;
                                        try {
                                                int readBytes = read(buf, endOffset, 512);
                                                if (readBytes < 0) {
                                                        logger1.log(Level.WARNING, "Possible Incomplete Image: Error reading volume at offset {0}", endOffset); //NON-NLS
                                                        errorString = MessageFormat.format(bundle.getString("Image.verifyImageSize.errStr1.text"), endOffset);
                                                }
                                        } catch (TskCoreException ex) {
                                                logger1.log(Level.WARNING, "Possible Incomplete Image: Error reading volume at offset {0}: {1}", new Object[]{endOffset, ex.getLocalizedMessage()}); //NON-NLS
                                                errorString = MessageFormat.format(bundle.getString("Image.verifyImageSize.errStr2.text"), endOffset);
                                        }
                                }
                        }

                        List<FileSystem> fileSystems = getFileSystems();
                        for (FileSystem fs : fileSystems) {
                                long block_size = fs.getBlock_size();
                                long endOffset = fs.getImageOffset() + fs.getSize() - block_size;
                                try {
                                        byte[] buf = new byte[(int) block_size];
                                        int readBytes = read(buf, endOffset, block_size);
                                        if (readBytes < 0) {
                                                logger1.log(Level.WARNING, "Possible Incomplete Image: Error reading file system at offset {0}", endOffset); //NON-NLS
                                                errorString = MessageFormat.format(bundle.getString("Image.verifyImageSize.errStr3.text"), endOffset);
                                        }
                                } catch (TskCoreException ex) {
                                        logger1.log(Level.WARNING, "Possible Incomplete Image: Error reading file system at offset {0}: {1}", new Object[]{endOffset, ex.getLocalizedMessage()}); //NON-NLS
                                        errorString = MessageFormat.format(bundle.getString("Image.verifyImageSize.errStr4.text"), endOffset);
                                }
                        }
                } catch (TskException ex) {
                        // do nothing if we got an exception from trying to get file systems and volume systems
                }
                return errorString;
        }

        public String getMd5() throws TskCoreException {
                if (md5 == null || md5.isEmpty()) {
                        md5 = getSleuthkitCase().getMd5ImageHash(this);
                }
                return md5;
        }
        
        public String getSha1() throws TskCoreException {
                if (sha1 == null || sha1.isEmpty()) {
                        sha1 = getSleuthkitCase().getSha1ImageHash(this);
                }
                return sha1;
        }
        
        public String getSha256() throws TskCoreException {
                if (sha256 == null || sha256.isEmpty()) {
                        sha256 = getSleuthkitCase().getSha256ImageHash(this);
                }
                return sha256;
        }
        
        public void setMD5(String md5) throws TskCoreException, TskDataException {
                if (getMd5().isEmpty() == false) {
                        throw new TskDataException("MD5 value has already been set");
                }
                getSleuthkitCase().setMd5ImageHash(this, md5);
                this.md5 = md5;
        }
        
        public void setSha1(String sha1) throws TskCoreException, TskDataException {
                if (getSha1().isEmpty() == false) {
                        throw new TskDataException("SHA1 value has already been set");
                }
                getSleuthkitCase().setSha1ImageHash(this, sha1);
                this.sha1 = sha1;
        }
        
        public void setSha256(String sha256) throws TskCoreException, TskDataException {
                if (getSha256().isEmpty() == false) {
                        throw new TskDataException("SHA256 value has already been set");
                }
                getSleuthkitCase().setSha256ImageHash(this, sha256);
                this.sha256 = sha256;
        }

        @Override
        public String getDeviceId() {
                return deviceId;
        }

        @Override
        public void setDisplayName(String newName) throws TskCoreException {
                this.getSleuthkitCase().setImageName(newName, getId());
        }
        
        @Override
        public long getContentSize(SleuthkitCase sleuthkitCase) throws TskCoreException {
                SleuthkitCase.CaseDbConnection connection;
                Statement statement = null;
                ResultSet resultSet = null;
                long contentSize = 0;

                connection = sleuthkitCase.getConnection();

                try {
                        statement = connection.createStatement();
                        resultSet = connection.executeQuery(statement, "SELECT SUM (size) FROM tsk_image_info WHERE tsk_image_info.obj_id = " + getId());
                        if (resultSet.next()) {
                                contentSize = resultSet.getLong("sum");
                        }
                } catch (SQLException ex) {
                        throw new TskCoreException(String.format("There was a problem while querying the database for size data for object ID %d.", getId()), ex);
                } finally {
                        closeResultSet(resultSet);
                        closeStatement(statement);
                        connection.close();
                }

                return contentSize;
        }
        
        @Override
        public void setAcquisitionDetails(String details) throws TskCoreException {
                getSleuthkitCase().setAcquisitionDetails(this, details);
        }
        
        @Override
        public String getAcquisitionDetails() throws TskCoreException {
                return getSleuthkitCase().getAcquisitionDetails(this);
        }       

        private static void closeResultSet(ResultSet resultSet) {
                if (resultSet != null) {
                        try {
                                resultSet.close();
                        } catch (SQLException ex) {
                                LOGGER.log(Level.SEVERE, "Error closing ResultSet", ex); //NON-NLS
                        }
                }
        }

        private static void closeStatement(Statement statement) {
                if (statement != null) {
                        try {
                                statement.close();
                        } catch (SQLException ex) {
                                LOGGER.log(Level.SEVERE, "Error closing Statement", ex); //NON-NLS
                        }
                }
        }
}