1 files changed, 87 insertions, 0 deletions
diff --git a/pumpkin/TFTPPacket.h b/pumpkin/TFTPPacket.h
new file mode 100644
index 0000000..a46d5bc
--- a/dev/null
+++ b/pumpkin/TFTPPacket.h
@@ -0,0 +1,87 @@
+
+#import <Cocoa/Cocoa.h>
+#include <stdint.h>
+
+enum TFTPOp {
+	tftpOpRRQ=1, tftpOpWRQ=2,
+	tftpOpDATA=3,
+	tftpOpACK=4,
+	tftpOpERROR=5,
+	tftpOpOACK=6
+};
+
+enum TFTPError {
+	tftpErrUndefined=0,
+	tftpErrNotFound=1,
+	tftpErrAccessViolation=2,
+	tftpErrDiskFull=3,
+	tftpErrIllegalOp=4,
+	tftpErrUnknownTID=5,
+	tftpErrFileExists=6,
+	tftpErrNoUser=7,
+	tftpErrOption=8
+};
+
+#pragma pack(push,1)
+struct AnyTFTPPacket {
+    uint16_t op;
+    union {
+	struct {
+	    char data[1];
+	} any;
+	struct {
+	    char data[1];
+	} rq;
+	struct {
+	    char data[1];
+	} rrq;
+	struct {
+	    char data[1];
+	} wrq;
+	struct {
+	    uint16_t block;
+	    char data[1];
+	} data;
+	struct {
+	    uint16_t block;
+	} ack;
+	struct {
+	    uint16_t code;
+	    char data[1];
+	} err;
+	struct {
+	    char data[1];
+	} oack;
+    };
+};
+#pragma pack(pop)
+
+@interface TFTPPacket : NSObject {
+    NSData *data;
+    struct AnyTFTPPacket *packet;
+}
+
+@property (readonly) enum TFTPOp op;
+@property (readonly) NSString* rqFilename;
+@property (readonly) NSString* rqType;
+@property (readonly) NSDictionary* rqOptions;
+@property (readonly) NSData *data;
+@property (readonly) uint16_t block;
+@property (readonly) NSData *rqData;
+@property (readonly) uint16_t rqCode;
+@property (readonly) NSString* rqMessage;
+
+-(TFTPPacket*)initWithData:(NSData*)d;
+
++(TFTPPacket*)packetWithData:(NSData*)d;
++(TFTPPacket*)packetWithBytesNoCopy:(void*)b andLength:(size_t)l;
+
++(TFTPPacket*)packetErrorWithCode:(enum TFTPError)c andMessage:(NSString*)m;
++(TFTPPacket*)packetErrorWithErrno:(int)en andFallback:(NSString*)fb;
++(TFTPPacket*)packetOACKWithOptions:(NSDictionary*)o;
++(TFTPPacket*)packetDataWithBlock:(uint16_t)b andData:(NSData*)d;
++(TFTPPacket*)packetACKWithBlock:(uint16_t)b;
++(TFTPPacket*)packetRRQWithFile:(NSString*)f xferType:(NSString*)t andOptions:(NSDictionary*)o;
++(TFTPPacket*)packetWRQWithFile:(NSString*)f xferType:(NSString*)t andOptions:(NSDictionary*)o;
+
+@end