Fix nswering procedure: have to use source resource.

[anna.git] / example / diameter / launcher / ProgrammedAnswers.cpp

diff --git a/example/diameter/launcher/ProgrammedAnswers.cpp b/example/diameter/launcher/ProgrammedAnswers.cpp
index f617282..9356dac 100644 (file)
--- a/example/diameter/launcher/ProgrammedAnswers.cpp
+++ b/example/diameter/launcher/ProgrammedAnswers.cpp
@@ -14,7 +14,7 @@
 #include <anna/diameter/codec/Engine.hpp>
 
 // Process
-#include "ProgrammedAnswers.hpp"
+#include <ProgrammedAnswers.hpp>
 
 
 void ProgrammedAnswers::clear () throw() {
@@ -45,6 +45,9 @@ void ProgrammedAnswers::dump () throw() {
 }
 
 void ProgrammedAnswers::addMessage(int code, anna::diameter::codec::Message *message) throw() {
+  if (!message) return; // just in case
+  message->setEngine(a_codecEngine); // just in case
+
   reacting_answers_const_iterator it = a_deques.find(code);
   if (it != a_deques.end()) {
        it->second->push_back(message);