/*

  *  Copyright (C) 2014-2019 Cisco Systems, Inc. and/or its affiliates. All rights reserved.

  *
  *  Authors: Kevin Lin <klin@sourcefire.com>
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License version 2 as
  *  published by the Free Software Foundation.
  *
  *  This program is distributed in the hope that it will be useful,
  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  *  GNU General Public License for more details.
  *
  *  You should have received a copy of the GNU General Public License
  *  along with this program; if not, write to the Free Software
  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  *  MA 02110-1301, USA.
  */
 
 #ifndef __APM_H
 #define __APM_H
 
 #if HAVE_CONFIG_H
 #include "clamav-config.h"
 #endif
 

 #include "clamav-types.h"

 #include "others.h"
 
 #define APM_FALLBACK_SECTOR_SIZE 512
 
 #define APM_PTABLE_BLOCK 1
 #define APM_STRUCT_SIZE 512
 
 #define DDM_SIGNATURE 0x4552 /* driver description signature ('ER') */
 #define APM_SIGNATURE 0x504D /* partition map signature ('PM') */
 
 /* partition flags */
 #define VALID          0x00000001
 #define ALLOCATED      0x00000002
 #define IN_USE         0x00000004
 #define BOOTABLE       0x00000008
 #define READABLE       0x00000010
 #define WRITEABLE      0x00000020
 #define POSINDEPENDENT 0x00000040
 /* end of partition flags */
 
 #ifndef HAVE_ATTRIB_PACKED
 #define __attribute__(x)
 #endif
 
 #ifdef HAVE_PRAGMA_PACK
 #pragma pack(1)
 #endif
 
 #ifdef HAVE_PRAGMA_PACK_HPPA
 #pragma pack 1
 #endif
 
 /* 8-byte driver description entry for ddmap, big endian */
 struct apm_driver_desc_entry {
     uint32_t block  __attribute__ ((packed));
     uint16_t size  __attribute__ ((packed));
     uint16_t type  __attribute__ ((packed));
 }; //NOTE may need to be renamed
 
 /* 512(82)-byte driver descriptor map (Block0), big endian */
 struct apm_driver_desc_map {
     uint16_t signature  __attribute__ ((packed));
     uint16_t blockSize  __attribute__ ((packed));
     uint32_t blockCount  __attribute__ ((packed));
     uint16_t deviceType  __attribute__ ((packed));
     uint16_t deviceID  __attribute__ ((packed));
     uint32_t driverData  __attribute__ ((packed));
     uint16_t driverCount  __attribute__ ((packed));
     struct apm_driver_desc_entry driverTable[8];
     /* zeroes fill remainder of sector (430 bytes in 512 sector size) */
 };
 
 /* 512(136)-byte partition info, big endian;
  * both the partition table and the individual partitions use this 
  * struct to describe their details
  */
 struct apm_partition_info {
     uint16_t signature  __attribute__ ((packed));
     uint16_t reserved  __attribute__ ((packed));
     uint32_t numPartitions  __attribute__ ((packed));
     uint32_t pBlockStart  __attribute__ ((packed));
     uint32_t pBlockCount  __attribute__ ((packed));
     uint8_t name[32];
     uint8_t type[32];
     uint32_t lBlockStart  __attribute__ ((packed));
     uint32_t lBlockCount  __attribute__ ((packed));
     uint32_t flags  __attribute__ ((packed));
     uint32_t bootBlockStart  __attribute__ ((packed));
     uint32_t bootSize  __attribute__ ((packed));
     uint32_t bootAddr  __attribute__ ((packed));
     uint32_t bootAddr2  __attribute__ ((packed));
     uint32_t bootEntry  __attribute__ ((packed));
     uint32_t bootEntry2  __attribute__ ((packed));
     uint32_t bootChecksum  __attribute__ ((packed));
     uint8_t processor[16];
     /* zeroes fill remainder of sector (376 bytes in 512 sector size) */
 };
 
 #ifdef HAVE_PRAGMA_PACK
 #pragma pack()
 #endif
 
 #ifdef HAVE_PRAGMA_PACK_HPPA
 #pragma pack
 #endif
 
 int cli_scanapm(cli_ctx *ctx);
 
 #endif