From 7f32ff342da2c7161971c0b22f6734b8c0b3121a Mon Sep 17 00:00:00 2001
From: Sergey Abramchuk <personal@ss-abramchuk.me>
Date: Wed, 10 Jun 2020 08:45:01 +0300
Subject: [PATCH] Add assertion to check packetFlow property

---
 Sources/OpenVPNAdapter/library/OpenVPNPacketFlowBridge.mm | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Sources/OpenVPNAdapter/library/OpenVPNPacketFlowBridge.mm b/Sources/OpenVPNAdapter/library/OpenVPNPacketFlowBridge.mm
index c464901..72e73c0 100644
--- a/Sources/OpenVPNAdapter/library/OpenVPNPacketFlowBridge.mm
+++ b/Sources/OpenVPNAdapter/library/OpenVPNPacketFlowBridge.mm
@@ -135,6 +135,8 @@ static void SocketCallback(CFSocketRef socket, CFSocketCallBackType type, CFData
 }
 
 - (void)startReading {
+    NSAssert(self.packetFlow != nil, @"packetFlow property shouldn't be nil, set it before start reading packets.");
+    
     __weak typeof(self) weakSelf = self;
     
     [self.packetFlow readPacketsWithCompletionHandler:^(NSArray<NSData *> *packets, NSArray<NSNumber *> *protocols) {
@@ -161,6 +163,8 @@ static void SocketCallback(CFSocketRef socket, CFSocketCallBackType type, CFData
 #pragma mark - VPN -> TUN
 
 - (void)writePackets:(NSArray<OpenVPNPacket *> *)packets toPacketFlow:(id<OpenVPNAdapterPacketFlow>)packetFlow {
+    NSAssert(packetFlow != nil, @"packetFlow shouldn't be nil, check provided parameter before start writing packets.");
+    
     NSMutableArray<NSData *> *flowPackets = [[NSMutableArray alloc] init];
     NSMutableArray<NSNumber *> *protocols = [[NSMutableArray alloc] init];