Merge "BUG-50 : added test for Error message."
[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / message / PCEPErrorMessageParser.java
1 /*
2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.protocol.pcep.impl.message;
9
10 import io.netty.buffer.ByteBuf;
11
12 import java.util.Arrays;
13 import java.util.List;
14
15 import org.opendaylight.protocol.pcep.PCEPDeserializerException;
16 import org.opendaylight.protocol.pcep.PCEPDocumentedException;
17 import org.opendaylight.protocol.pcep.PCEPErrorMapping;
18 import org.opendaylight.protocol.pcep.PCEPErrors;
19 import org.opendaylight.protocol.pcep.UnknownObject;
20 import org.opendaylight.protocol.pcep.impl.AbstractMessageParser;
21 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcerrMessage;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObject;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObjectBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.Request;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.RequestBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.Session;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request.Rps;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request.RpsBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
39
40 import com.google.common.collect.Lists;
41
42 /**
43  * Parser for {@link PcerrMessage}
44  */
45 public class PCEPErrorMessageParser extends AbstractMessageParser {
46
47         public static final int TYPE = 6;
48
49         public PCEPErrorMessageParser(final ObjectHandlerRegistry registry) {
50                 super(registry);
51         }
52
53         @Override
54         public void serializeMessage(final Message message, final ByteBuf buffer) {
55                 if (!(message instanceof PcerrMessage)) {
56                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance " + message.getClass()
57                                         + ". Nedded ErrorMessage.");
58                 }
59                 final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessage err = ((PcerrMessage) message).getPcerrMessage();
60
61                 if (err.getErrors() == null || err.getErrors().isEmpty()) {
62                         throw new IllegalArgumentException("Errors should not be empty.");
63                 }
64
65                 if (err.getErrorType() instanceof Request) {
66                         final List<Rps> rps = ((Request) err.getErrorType()).getRps();
67                         for (final Rps r : rps) {
68                                 buffer.writeBytes(serializeObject(r.getRp()));
69                         }
70                 }
71
72                 for (final Errors e : err.getErrors()) {
73                         buffer.writeBytes(serializeObject(e.getErrorObject()));
74                 }
75
76                 if (err.getErrorType() instanceof Session) {
77                         buffer.writeBytes(serializeObject(((Session) err.getErrorType()).getOpen()));
78                 }
79         }
80
81         @Override
82         public Message parseMessage(final byte[] buffer) throws PCEPDeserializerException, PCEPDocumentedException {
83                 if (buffer == null || buffer.length == 0) {
84                         throw new PCEPDeserializerException("Error message is empty.");
85                 }
86                 final List<Object> objs = parseObjects(buffer);
87                 PcerrMessage m = null;
88                 try {
89                         m = validate(objs);
90                 } catch (final PCEPDocumentedException e) {
91                         final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
92                         return new PcerrBuilder().setPcerrMessage(
93                                         new PcerrMessageBuilder().setErrors(
94                                                         Arrays.asList(new ErrorsBuilder().setErrorObject(
95                                                                         new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e.getError()).type).setValue(
96                                                                                         maping.getFromErrorsEnum(e.getError()).value).build()).build())).build()).build();
97                 }
98                 return m;
99         }
100
101         private PcerrMessage validate(final List<Object> objects) throws PCEPDeserializerException, PCEPDocumentedException {
102                 if (objects == null) {
103                         throw new IllegalArgumentException("Passed list can't be null.");
104                 }
105
106                 Open openObj = null;
107                 final List<Rps> requestParameters = Lists.newArrayList();
108                 final List<Errors> errorObjects = Lists.newArrayList();
109                 final PcerrMessageBuilder b = new PcerrMessageBuilder();
110
111                 Object obj;
112                 int state = 0;
113                 while (!objects.isEmpty()) {
114                         obj = objects.get(0);
115
116                         if (obj instanceof UnknownObject) {
117                                 return new PcerrBuilder().setPcerrMessage(b.setErrors(((UnknownObject) obj).getErrors()).build()).build();
118                         }
119
120                         if (state == 0) {
121                                 if (obj instanceof ErrorObject) {
122                                         final ErrorObject o = (ErrorObject) obj;
123                                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
124                                         state = 1;
125                                         objects.remove(0);
126                                         continue;
127                                 } else if (obj instanceof Rp) {
128                                         final Rp o = ((Rp) obj);
129                                         if (o.isProcessingRule()) {
130                                                 throw new PCEPDocumentedException("Invalid setting of P flag.", PCEPErrors.P_FLAG_NOT_SET);
131                                         }
132                                         requestParameters.add(new RpsBuilder().setRp(o).build());
133                                         state = 2;
134                                         objects.remove(0);
135                                         continue;
136                                 }
137                         }
138
139                         switch (state) {
140                         case 1:
141                                 if (obj instanceof ErrorObject) {
142                                         final ErrorObject o = (ErrorObject) obj;
143                                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
144                                         state = 1;
145                                         break;
146                                 }
147                                 state = 3;
148                         case 2:
149                                 if (obj instanceof Rp) {
150                                         final Rp o = ((Rp) obj);
151                                         if (o.isProcessingRule()) {
152                                                 throw new PCEPDocumentedException("Invalid setting of P flag.", PCEPErrors.P_FLAG_NOT_SET);
153                                         }
154                                         requestParameters.add(new RpsBuilder().setRp(o).build());
155                                         state = 2;
156                                         break;
157                                 }
158                                 state = 4;
159                         case 3:
160                                 if (obj instanceof Open) {
161                                         openObj = (Open) obj;
162                                         state = 5;
163                                         break;
164                                 }
165                         case 4:
166                                 if (obj instanceof ErrorObject) {
167                                         final ErrorObject o = (ErrorObject) obj;
168                                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
169                                         state = 4;
170                                         break;
171                                 }
172                                 state = 5;
173
174                                 if (state == 5) {
175                                         break;
176                                 }
177                         }
178
179                         objects.remove(0);
180                 }
181
182                 if (errorObjects.isEmpty() && errorObjects.isEmpty()) {
183                         throw new PCEPDeserializerException("At least one PCEPErrorObject is mandatory.");
184                 }
185
186                 if (!objects.isEmpty()) {
187                         throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
188                 }
189                 if (requestParameters != null) {
190                         b.setErrorType(new RequestBuilder().setRps(requestParameters).build());
191                 }
192                 if (openObj != null) {
193                         b.setErrorType(new SessionBuilder().setOpen(openObj).build());
194                 }
195
196                 return new PcerrBuilder().setPcerrMessage(b.setErrors(errorObjects).build()).build();
197         }
198
199         @Override
200         public int getMessageType() {
201                 return TYPE;
202         }
203 }