From 4048de35d02e0ed3cf8fbeaacf425bc27432df9b Mon Sep 17 00:00:00 2001
From: Zaid Abdulla <zkam83@gmail.com>
Date: Fri, 2 May 2014 19:38:47 +0300
Subject: [PATCH] Implemented GetPRCMethods ACS method

---
 lib/cwmp.coffee |  9 ++++++
 lib/soap.coffee | 83 +++++++++++++++++++++++++++++--------------------
 2 files changed, 59 insertions(+), 33 deletions(-)

diff --git a/lib/cwmp.coffee b/lib/cwmp.coffee
index 0707d4a..259c866 100644
--- a/lib/cwmp.coffee
+++ b/lib/cwmp.coffee
@@ -432,6 +432,15 @@ listener = (httpRequest, httpResponse) ->
     if cwmpRequest.methodRequest?
       if cwmpRequest.methodRequest.type is 'Inform'
         inform(currentRequest, cwmpRequest)
+      else if cwmpRequest.methodRequest.type is 'GetRPCMethods'
+        util.log("#{currentRequest.deviceId}: GetRPCMethods")
+        res = soap.response({
+          id : cwmpRequest.id,
+          methodResponse : {type : 'GetRPCMethodsResponse', methodList : ['Inform', 'GetRPCMethods', 'TransferComplete', 'RequestDownload']},
+          cwmpVersion : currentRequest.cwmpVersion,
+          cookies : cookies
+        })
+        writeResponse(currentRequest, res)
       else if cwmpRequest.methodRequest.type is 'TransferComplete'
         # do nothing
         util.log("#{currentRequest.deviceId}: Transfer complete")
diff --git a/lib/soap.coffee b/lib/soap.coffee
index e8d7091..2203ab6 100644
--- a/lib/soap.coffee
+++ b/lib/soap.coffee
@@ -151,34 +151,6 @@ cpeDeleteObjectResponse = (xml) ->
   {status : parseInt(xml.get('Status').text())}
 
 
-traverseXml = (xml) ->
-  obj = {}
-  for n in xml.childNodes()
-    if n.type() == 'element'
-      obj[n.name()] = traverseXml(n)
-
-  if Object.keys(obj).length == 0
-    xml.text()
-  else
-    obj
-
-fault = (xml) ->
-  return traverseXml(xml)
-
-
-acsInform = (xml) ->
-  {
-    parameterList : parameterValueList(xml.get('ParameterList')),
-    deviceId : traverseXml(xml.get('DeviceId')),
-    event : event(xml.get('Event')),
-    retryCount : JSON.parse(xml.get('RetryCount').text())
-  }
-
-
-acsInformResponse = (xml) ->
-  xml.node('cwmp:InformResponse').node('MaxEnvelopes').text(1)
-
-
 cpeReboot = (xml, methodRequest) ->
   el = xml.node('cwmp:Reboot')
   el.node('CommandKey').text(methodRequest.commandKey ? '')
@@ -214,6 +186,30 @@ cpeDownloadResponse = (xml) ->
   {status : JSON.parse(xml.get('Status').text())}
 
 
+acsInform = (xml) ->
+  {
+    parameterList : parameterValueList(xml.get('ParameterList')),
+    deviceId : traverseXml(xml.get('DeviceId')),
+    event : event(xml.get('Event')),
+    retryCount : JSON.parse(xml.get('RetryCount').text())
+  }
+
+
+acsInformResponse = (xml, methodResponse) ->
+  xml.node('cwmp:InformResponse').node('MaxEnvelopes').text(1)
+
+
+acsGetRPCMethods = (xml) ->
+  {}
+
+
+acsGetRPCMethodsResponse = (xml, methodResponse) ->
+  el = xml.node('cwmp:GetRPCMethodsResponse').node('MethodList')
+  el.attr({'soap-enc:arrayType' : "xsd:string[#{methodResponse.methodList.length}]"})
+  for m in methodResponse.methodList
+    el.node('xsd:string').text(m)
+
+
 acsTransferComplete = (xml) ->
   {
     commandKey : xml.get('CommandKey').text(),
@@ -223,7 +219,7 @@ acsTransferComplete = (xml) ->
   }
 
 
-acsTransferCompleteResponse = (xml, methodRequest) ->
+acsTransferCompleteResponse = (xml, methodResponse) ->
   xml.node('cwmp:TransferCompleteResponse').text('')
 
 
@@ -234,10 +230,26 @@ acsRequestDownload = (xml) ->
   }
 
 
-acsRequestDownloadResponse = (xml, methodRequest) ->
+acsRequestDownloadResponse = (xml, methodResponse) ->
   xml.node('cwmp:RequestDownloadResponse').text('')
 
 
+traverseXml = (xml) ->
+  obj = {}
+  for n in xml.childNodes()
+    if n.type() == 'element'
+      obj[n.name()] = traverseXml(n)
+
+  if Object.keys(obj).length == 0
+    xml.text()
+  else
+    obj
+
+
+fault = (xml) ->
+  return traverseXml(xml)
+
+
 exports.request = (httpRequest) ->
   cwmpRequest = {cookies: {}}
   cwmpRequest.cookies = cookiesToObj(httpRequest.headers.cookie) if httpRequest.headers.cookie
@@ -282,6 +294,9 @@ exports.request = (httpRequest) ->
         when 'Inform'
           cwmpRequest.methodRequest = acsInform(methodElement)
           cwmpRequest.methodRequest.type = 'Inform'
+        when 'GetRPCMethods'
+          cwmpRequest.methodRequest = acsGetRPCMethods(methodElement)
+          cwmpRequest.methodRequest.type = 'GetRPCMethods'
         when 'TransferComplete'
           cwmpRequest.methodRequest = acsTransferComplete(methodElement)
           cwmpRequest.methodRequest.type = 'TransferComplete'
@@ -345,11 +360,13 @@ exports.response = (cwmpResponse) ->
       when 'InformResponse'
         cwmpResponse.cookies ?= {}
         cwmpResponse.cookies.cwmpVersion = cwmpResponse.cwmpVersion
-        acsInformResponse(body)
+        acsInformResponse(body, cwmpResponse.methodResponse)
+      when 'GetRPCMethodsResponse'
+        acsGetRPCMethodsResponse(body, cwmpResponse.methodResponse)
       when 'TransferCompleteResponse'
-        acsTransferCompleteResponse(body)
+        acsTransferCompleteResponse(body, cwmpResponse.methodResponse)
       when 'RequestDownloadResponse'
-        acsRequestDownloadResponse(body)
+        acsRequestDownloadResponse(body, cwmpResponse.methodResponse)
       else
         throw Error("Unknown method response type #{cwmpResponse.methodResponse.type}")
   else if cwmpResponse?.methodRequest?