#ifndef _IOPOLLEDINTERFACE_H_
#define _IOPOLLEDINTERFACE_H_
+enum
+{
+ kIOPolledPreflightState = 1,
+ kIOPolledBeforeSleepState = 2,
+ kIOPolledAfterSleepState = 3,
+ kIOPolledPostflightState = 4,
+
+ kIOPolledPreflightCoreDumpState = 5,
+ kIOPolledPostflightCoreDumpState = 6,
+
+ kIOPolledBeforeSleepStateAborted = 7,
+};
+
+#if defined(__cplusplus)
+
#include <libkern/c++/OSObject.h>
#include <IOKit/IOMemoryDescriptor.h>
#define kIOPolledInterfaceSupportKey "IOPolledInterface"
#define kIOPolledInterfaceActiveKey "IOPolledInterfaceActive"
-
-enum
-{
- kIOPolledPreflightState = 1,
- kIOPolledBeforeSleepState = 2,
- kIOPolledAfterSleepState = 3,
- kIOPolledPostflightState = 4
-};
+#define kIOPolledInterfaceStackKey "IOPolledInterfaceStack"
enum
{
kIOPolledWrite = 1,
- kIOPolledRead = 2
+ kIOPolledRead = 2,
+ kIOPolledFlush = 3
};
typedef void (*IOPolledCompletionAction)( void * target,
virtual IOReturn checkForWork(void) = 0;
- static IOReturn checkAllForWork(void);
+ virtual IOReturn setEncryptionKey(const uint8_t * key, size_t keySize);
- OSMetaClassDeclareReservedUnused(IOPolledInterface, 0);
+ OSMetaClassDeclareReservedUsed(IOPolledInterface, 0);
OSMetaClassDeclareReservedUnused(IOPolledInterface, 1);
OSMetaClassDeclareReservedUnused(IOPolledInterface, 2);
OSMetaClassDeclareReservedUnused(IOPolledInterface, 3);
OSMetaClassDeclareReservedUnused(IOPolledInterface, 15);
};
+#endif /* defined(__cplusplus) */
+
+#ifdef XNU_KERNEL_PRIVATE
+
+#include <libkern/crypto/aes.h>
+#include <IOKit/IOTypes.h>
+#include <IOKit/IOHibernatePrivate.h>
+
+enum
+{
+ kIOPolledFileSSD = 0x00000001
+};
+
+#if !defined(__cplusplus)
+typedef struct IORegistryEntry IORegistryEntry;
+typedef struct OSData OSData;
+typedef struct OSArray OSArray;
+typedef struct IOMemoryDescriptor IOMemoryDescriptor;
+typedef struct IOPolledFilePollers IOPolledFilePollers;
+#else
+class IOPolledFilePollers;
+#endif
+
+struct IOPolledFileIOVars
+{
+ IOPolledFilePollers * pollers;
+ struct kern_direct_file_io_ref_t * fileRef;
+ OSData * fileExtents;
+ uint64_t block0;
+ IOByteCount blockSize;
+ uint64_t maxiobytes;
+ IOByteCount bufferLimit;
+ uint8_t * buffer;
+ IOByteCount bufferSize;
+ IOByteCount bufferOffset;
+ IOByteCount bufferHalf;
+ IOByteCount extentRemaining;
+ IOByteCount lastRead;
+ IOByteCount readEnd;
+ uint32_t flags;
+ uint64_t fileSize;
+ uint64_t position;
+ uint64_t extentPosition;
+ uint64_t encryptStart;
+ uint64_t encryptEnd;
+ uint64_t cryptBytes;
+ AbsoluteTime cryptTime;
+ IOPolledFileExtent * extentMap;
+ IOPolledFileExtent * currentExtent;
+ bool allocated;
+};
+
+typedef struct IOPolledFileIOVars IOPolledFileIOVars;
+
+struct IOPolledFileCryptVars
+{
+ uint8_t aes_iv[AES_BLOCK_SIZE];
+ aes_ctx ctx;
+};
+typedef struct IOPolledFileCryptVars IOPolledFileCryptVars;
+
+#if defined(__cplusplus)
+
+IOReturn IOPolledFileOpen(const char * filename,
+ uint64_t setFileSize, uint64_t fsFreeSize,
+ void * write_file_addr, size_t write_file_len,
+ IOPolledFileIOVars ** fileVars,
+ OSData ** imagePath,
+ uint8_t * volumeCryptKey, size_t * keySize);
+
+IOReturn IOPolledFileClose(IOPolledFileIOVars ** pVars,
+ off_t write_offset, void * addr, size_t write_length,
+ off_t discard_offset, off_t discard_end);
+
+IOReturn IOPolledFilePollersSetup(IOPolledFileIOVars * vars, uint32_t openState);
+
+IOMemoryDescriptor * IOPolledFileGetIOBuffer(IOPolledFileIOVars * vars);
+
+#endif /* defined(__cplusplus) */
+
+#if defined(__cplusplus)
+#define __C "C"
+#else
+#define __C
+#endif
+
+extern __C IOReturn IOPolledFileSeek(IOPolledFileIOVars * vars, uint64_t position);
+
+extern __C IOReturn IOPolledFileWrite(IOPolledFileIOVars * vars,
+ const uint8_t * bytes, IOByteCount size,
+ IOPolledFileCryptVars * cryptvars);
+extern __C IOReturn IOPolledFileRead(IOPolledFileIOVars * vars,
+ uint8_t * bytes, IOByteCount size,
+ IOPolledFileCryptVars * cryptvars);
+
+extern __C IOReturn IOPolledFileFlush(IOPolledFileIOVars * vars);
+
+extern __C IOReturn IOPolledFilePollersOpen(IOPolledFileIOVars * vars, uint32_t state, bool abortable);
+
+extern __C IOReturn IOPolledFilePollersClose(IOPolledFileIOVars * vars, uint32_t state);
+
+extern __C IOReturn IOPolledFilePollersSetEncryptionKey(IOPolledFileIOVars * vars,
+ const uint8_t * key, size_t keySize);
+
+extern __C IOPolledFileIOVars * gCoreFileVars;
+
+#ifdef _SYS_CONF_H_
+
+__BEGIN_DECLS
+
+typedef void (*kern_get_file_extents_callback_t)(void * ref, uint64_t start, uint64_t size);
+
+struct kern_direct_file_io_ref_t *
+kern_open_file_for_direct_io(const char * name, boolean_t create_file,
+ kern_get_file_extents_callback_t callback,
+ void * callback_ref,
+ off_t set_file_size,
+ off_t fs_free_size,
+ off_t write_file_offset,
+ void * write_file_addr,
+ size_t write_file_len,
+ dev_t * partition_device_result,
+ dev_t * image_device_result,
+ uint64_t * partitionbase_result,
+ uint64_t * maxiocount_result,
+ uint32_t * oflags);
+void
+kern_close_file_for_direct_io(struct kern_direct_file_io_ref_t * ref,
+ off_t write_offset, void * addr, size_t write_length,
+ off_t discard_offset, off_t discard_end);
+int
+kern_write_file(struct kern_direct_file_io_ref_t * ref, off_t offset, void * addr, size_t len, int ioflag);
+int
+kern_read_file(struct kern_direct_file_io_ref_t * ref, off_t offset, void * addr, size_t len, int ioflag);
+
+struct mount *
+kern_file_mount(struct kern_direct_file_io_ref_t * ref);
+
+enum
+{
+ kIOPolledFileMountChangeMount = 0x00000101,
+ kIOPolledFileMountChangeUnmount = 0x00000102,
+ kIOPolledFileMountChangeWillResize = 0x00000201,
+ kIOPolledFileMountChangeDidResize = 0x00000202,
+};
+extern void IOPolledFileMountChange(struct mount * mp, uint32_t op);
+
+__END_DECLS
+
+#endif /* _SYS_CONF_H_ */
+
+#endif /* XNU_KERNEL_PRIVATE */
+
#endif /* _IOPOLLEDINTERFACE_H_ */