BUG-1199 : fixed incorrect parsing of Error message.
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful07 / Stateful07ErrorMessageParser.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.ietf.stateful07;
9
10 import com.google.common.base.Preconditions;
11 import io.netty.buffer.ByteBuf;
12 import io.netty.buffer.Unpooled;
13 import java.util.ArrayList;
14 import java.util.List;
15 import org.opendaylight.protocol.pcep.impl.message.PCEPErrorMessageParser;
16 import org.opendaylight.protocol.pcep.spi.MessageUtil;
17 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
18 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
19 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
20 import org.opendaylight.protocol.pcep.spi.UnknownObject;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.StatefulCase;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.StatefulCaseBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.stateful._case.StatefulBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.stateful._case.stateful.Srps;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.pcerr.pcerr.message.error.type.stateful._case.stateful.SrpsBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcerrMessage;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObject;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.RequestCase;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.RequestCaseBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionCase;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionCaseBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request._case.RequestBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request._case.request.Rps;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request._case.request.RpsBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.session._case.SessionBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
45
46 /**
47  * Parser for {@link PcerrMessage}
48  */
49 public final class Stateful07ErrorMessageParser extends PCEPErrorMessageParser {
50
51     public Stateful07ErrorMessageParser(final ObjectRegistry registry) {
52         super(registry);
53     }
54
55     @Override
56     public void serializeMessage(final Message message, final ByteBuf out) {
57         Preconditions.checkArgument(message instanceof PcerrMessage, "Wrong instance of Message. Passed instance of %s. Need ErrorMessage.", message.getClass());
58         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessage err = ((PcerrMessage) message).getPcerrMessage();
59
60         if (err.getErrors() == null || err.getErrors().isEmpty()) {
61             throw new IllegalArgumentException("Errors should not be empty.");
62         }
63         final ByteBuf buffer = Unpooled.buffer();
64
65         if (err.getErrorType() instanceof RequestCase) {
66             final List<Rps> rps = ((RequestCase) err.getErrorType()).getRequest().getRps();
67             for (final Rps r : rps) {
68                 serializeObject(r.getRp(), buffer);
69             }
70         }
71         if (err.getErrorType() instanceof StatefulCase) {
72             final List<Srps> srps = ((StatefulCase) err.getErrorType()).getStateful().getSrps();
73             for (final Srps s : srps) {
74                 serializeObject(s.getSrp(), buffer);
75             }
76         }
77         for (final Errors e : err.getErrors()) {
78             serializeObject(e.getErrorObject(), buffer);
79         }
80         if (err.getErrorType() instanceof SessionCase) {
81             serializeObject(((SessionCase) err.getErrorType()).getSession().getOpen(), buffer);
82         }
83         MessageUtil.formatMessage(TYPE, buffer, out);
84     }
85
86     @Override
87     protected PcerrMessage validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
88         if (objects == null) {
89             throw new IllegalArgumentException("Passed list can't be null.");
90         }
91         if (objects.isEmpty()) {
92             throw new PCEPDeserializerException("Error message is empty.");
93         }
94
95         final List<Rps> requestParameters = new ArrayList<>();
96         final List<Srps> srps = new ArrayList<>();
97         final List<Errors> errorObjects = new ArrayList<>();
98         final PcerrMessageBuilder b = new PcerrMessageBuilder();
99
100         Object obj;
101         State state = State.Init;
102         obj = objects.get(0);
103
104         if (obj instanceof ErrorObject) {
105             final ErrorObject o = (ErrorObject) obj;
106             errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
107             state = State.ErrorIn;
108             objects.remove(0);
109         } else if (obj instanceof Rp) {
110             final Rp o = (Rp) obj;
111             if (o.isProcessingRule()) {
112                 errors.add(createErrorMsg(PCEPErrors.P_FLAG_NOT_SET));
113                 return null;
114             }
115             requestParameters.add(new RpsBuilder().setRp(o).build());
116             state = State.RpIn;
117             objects.remove(0);
118         } else if (obj instanceof Srp) {
119             final Srp s = (Srp) obj;
120             srps.add(new SrpsBuilder().setSrp(s).build());
121             state = State.SrpIn;
122             objects.remove(0);
123         }
124
125         while (!objects.isEmpty()) {
126             obj = objects.get(0);
127
128             if (obj instanceof UnknownObject) {
129                 return new PcerrBuilder().setPcerrMessage(b.setErrors(((UnknownObject) obj).getErrors()).build()).build();
130             }
131
132             switch (state) {
133             case ErrorIn:
134                 state = State.Open;
135                 if (obj instanceof ErrorObject) {
136                     final ErrorObject o = (ErrorObject) obj;
137                     errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
138                     state = State.ErrorIn;
139                     break;
140                 }
141             case RpIn:
142                 state = State.Error;
143                 if (obj instanceof Rp) {
144                     final Rp o = ((Rp) obj);
145                     requestParameters.add(new RpsBuilder().setRp(o).build());
146                     state = State.RpIn;
147                     break;
148                 }
149             case SrpIn:
150                 state = State.Error;
151                 if (obj instanceof Srp) {
152                     final Srp o = ((Srp) obj);
153                     srps.add(new SrpsBuilder().setSrp(o).build());
154                     state = State.SrpIn;
155                     break;
156                 }
157             case Open:
158                 state = State.OpenIn;
159                 if (obj instanceof Open) {
160                     b.setErrorType(new SessionCaseBuilder().setSession(new SessionBuilder().setOpen((Open) obj).build()).build());
161                     break;
162                 }
163             case Error:
164                 state = State.OpenIn;
165                 if (obj instanceof ErrorObject) {
166                     final ErrorObject o = (ErrorObject) obj;
167                     errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
168                     state = State.Error;
169                     break;
170                 }
171             case OpenIn:
172                 state = State.End;
173                 break;
174             case End:
175                 break;
176             default:
177                 break;
178             }
179             if (!state.equals(State.End)) {
180                 objects.remove(0);
181             }
182         }
183
184         if (errorObjects.isEmpty()) {
185             throw new PCEPDeserializerException("At least one PCEPErrorObject is mandatory.");
186         }
187
188         if (!objects.isEmpty()) {
189             throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
190         }
191         if (!requestParameters.isEmpty()) {
192             b.setErrorType(new RequestCaseBuilder().setRequest(new RequestBuilder().setRps(requestParameters).build()).build());
193         }
194         if (!srps.isEmpty()) {
195             b.setErrorType(new StatefulCaseBuilder().setStateful(new StatefulBuilder().setSrps(srps).build()).build());
196         }
197
198         return new PcerrBuilder().setPcerrMessage(b.setErrors(errorObjects).build()).build();
199     }
200
201     private enum State {
202         Init, ErrorIn, RpIn, SrpIn, Open, Error, OpenIn, End
203     }
204 }