Updated serializer for Pcreq 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.List;
13
14 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
15 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
16 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
17 import org.opendaylight.protocol.pcep.spi.UnknownObject;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcerrMessage;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObject;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.Request;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.RequestBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.Session;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request.Rps;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.request.RpsBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
34
35 import com.google.common.collect.Lists;
36
37 /**
38  * Parser for {@link PcerrMessage}
39  */
40 public class PCEPErrorMessageParser extends AbstractMessageParser {
41
42         public static final int TYPE = 6;
43
44         public PCEPErrorMessageParser(final ObjectHandlerRegistry registry) {
45                 super(registry);
46         }
47
48         @Override
49         public void serializeMessage(final Message message, final ByteBuf buffer) {
50                 if (!(message instanceof PcerrMessage)) {
51                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance " + message.getClass()
52                                         + ". Nedded ErrorMessage.");
53                 }
54                 final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessage err = ((PcerrMessage) message).getPcerrMessage();
55
56                 if (err.getErrors() == null || err.getErrors().isEmpty()) {
57                         throw new IllegalArgumentException("Errors should not be empty.");
58                 }
59
60                 if (err.getErrorType() instanceof Request) {
61                         final List<Rps> rps = ((Request) err.getErrorType()).getRps();
62                         for (final Rps r : rps) {
63                                 buffer.writeBytes(serializeObject(r.getRp()));
64                         }
65                 }
66
67                 for (final Errors e : err.getErrors()) {
68                         buffer.writeBytes(serializeObject(e.getErrorObject()));
69                 }
70
71                 if (err.getErrorType() instanceof Session) {
72                         buffer.writeBytes(serializeObject(((Session) err.getErrorType()).getOpen()));
73                 }
74         }
75
76         @Override
77         protected PcerrMessage validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
78                 if (objects == null) {
79                         throw new IllegalArgumentException("Passed list can't be null.");
80                 }
81
82                 if (objects.isEmpty()) {
83                         throw new PCEPDeserializerException("Error message is empty.");
84                 }
85
86                 final List<Rps> requestParameters = Lists.newArrayList();
87                 final List<Errors> errorObjects = Lists.newArrayList();
88                 final PcerrMessageBuilder b = new PcerrMessageBuilder();
89
90                 Object obj;
91                 State state = State.Init;
92                 obj = objects.get(0);
93
94                 if (obj instanceof ErrorObject) {
95                         final ErrorObject o = (ErrorObject) obj;
96                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
97                         state = State.ErrorIn;
98                         objects.remove(0);
99                 } else if (obj instanceof Rp) {
100                         final Rp o = ((Rp) obj);
101                         if (o.isProcessingRule()) {
102                                 errors.add(createErrorMsg(PCEPErrors.P_FLAG_NOT_SET));
103                                 return null;
104                         }
105                         requestParameters.add(new RpsBuilder().setRp(o).build());
106                         state = State.RpIn;
107                         objects.remove(0);
108                 }
109
110                 while (!objects.isEmpty()) {
111                         obj = objects.get(0);
112
113                         if (obj instanceof UnknownObject) {
114                                 return new PcerrBuilder().setPcerrMessage(b.setErrors(((UnknownObject) obj).getErrors()).build()).build();
115                         }
116
117                         switch (state) {
118                         case ErrorIn:
119                                 state = State.Open;
120                                 if (obj instanceof ErrorObject) {
121                                         final ErrorObject o = (ErrorObject) obj;
122                                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
123                                         state = State.ErrorIn;
124                                         break;
125                                 }
126                         case RpIn:
127                                 state = State.Error;
128                                 if (obj instanceof Rp) {
129                                         final Rp o = ((Rp) obj);
130                                         requestParameters.add(new RpsBuilder().setRp(o).build());
131                                         state = State.RpIn;
132                                         break;
133                                 }
134                         case Open:
135                                 state = State.OpenIn;
136                                 if (obj instanceof Open) {
137                                         b.setErrorType(new SessionBuilder().setOpen((Open) obj).build());
138                                         break;
139                                 }
140                         case Error:
141                                 state = State.OpenIn;
142                                 if (obj instanceof ErrorObject) {
143                                         final ErrorObject o = (ErrorObject) obj;
144                                         errorObjects.add(new ErrorsBuilder().setErrorObject(o).build());
145                                         state = State.Error;
146                                         break;
147                                 }
148                         case OpenIn:
149                                 state = State.End;
150                                 break;
151                         case End:
152                                 break;
153                         default:
154                                 break;
155                         }
156                         if (!state.equals(State.End)) {
157                                 objects.remove(0);
158                         }
159                 }
160
161                 if (errorObjects.isEmpty() && errorObjects.isEmpty()) {
162                         throw new PCEPDeserializerException("At least one PCEPErrorObject is mandatory.");
163                 }
164
165                 if (!objects.isEmpty()) {
166                         throw new PCEPDeserializerException("Unprocessed Objects: " + objects);
167                 }
168                 if (requestParameters != null && !requestParameters.isEmpty()) {
169                         b.setErrorType(new RequestBuilder().setRps(requestParameters).build());
170                 }
171
172                 return new PcerrBuilder().setPcerrMessage(b.setErrors(errorObjects).build()).build();
173         }
174
175         private enum State {
176                 Init, ErrorIn, RpIn, Open, Error, OpenIn, End
177         }
178
179         @Override
180         public int getMessageType() {
181                 return TYPE;
182         }
183 }