]> git.saurik.com Git - apple/xnu.git/blobdiff - iokit/IOKit/IOPolledInterface.h
xnu-4570.51.1.tar.gz
[apple/xnu.git] / iokit / IOKit / IOPolledInterface.h
index 740ab11c57f1efbfd019931ec656ab2b5322e2fb..84d27043b03cc749c6142e06843c239bd499d9cf 100644 (file)
 #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,
@@ -82,9 +91,9 @@ public:
 
     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);
@@ -102,4 +111,157 @@ public:
     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_ */