From dd21fa83ac063f3786fa17ddb23ff3f2d6536b56 Mon Sep 17 00:00:00 2001
From: Kaarle Ritvanen <kaarle.ritvanen@datakunkku.fi>
Date: Thu, 24 Jan 2013 14:08:14 +0000
Subject: move mark attribute checking to MarkRule.init

---
 awall/modules/mark.lua | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/awall/modules/mark.lua b/awall/modules/mark.lua
index ebdb169..787ea36 100644
--- a/awall/modules/mark.lua
+++ b/awall/modules/mark.lua
@@ -16,13 +16,15 @@ local model = awall.model
 
 local MarkRule = model.class(model.Rule)
 
-function MarkRule:table() return 'mangle' end
-
-function MarkRule:target()
+function MarkRule:init(...)
+   model.Rule.init(self, unpack(arg))
    if not self.mark then self:error('Mark not specified') end
-   return 'MARK --set-mark '..self.mark
 end
 
+function MarkRule:table() return 'mangle' end
+
+function MarkRule:target() return 'MARK --set-mark '..self.mark end
+
 
 local RouteTrackRule = model.class(MarkRule)
 
-- 
cgit v1.2.3