SessionDescription.java

  1package eu.siacs.conversations.xmpp.jingle;
  2
  3import android.util.Log;
  4import android.util.Pair;
  5
  6import com.google.common.base.Joiner;
  7import com.google.common.base.Strings;
  8import com.google.common.collect.ArrayListMultimap;
  9import com.google.common.collect.ImmutableList;
 10
 11import java.util.List;
 12import java.util.Locale;
 13import java.util.Map;
 14
 15import eu.siacs.conversations.Config;
 16import eu.siacs.conversations.xmpp.jingle.stanzas.Group;
 17import eu.siacs.conversations.xmpp.jingle.stanzas.IceUdpTransportInfo;
 18import eu.siacs.conversations.xmpp.jingle.stanzas.RtpDescription;
 19
 20public class SessionDescription {
 21
 22    private final static String LINE_DIVIDER = "\r\n";
 23    private final static String HARDCODED_MEDIA_PROTOCOL = "UDP/TLS/RTP/SAVPF"; //probably only true for DTLS-SRTP aka when we have a fingerprint
 24    private final static int HARDCODED_MEDIA_PORT = 9;
 25    private final static String HARDCODED_ICE_OPTIONS = "trickle renomination";
 26    private final static String HARDCODED_CONNECTION = "IN IP4 0.0.0.0";
 27
 28    public final int version;
 29    public final String name;
 30    public final String connectionData;
 31    public final ArrayListMultimap<String, String> attributes;
 32    public final List<Media> media;
 33
 34
 35    public SessionDescription(int version, String name, String connectionData, ArrayListMultimap<String, String> attributes, List<Media> media) {
 36        this.version = version;
 37        this.name = name;
 38        this.connectionData = connectionData;
 39        this.attributes = attributes;
 40        this.media = media;
 41    }
 42
 43    private static void appendAttributes(StringBuilder s, ArrayListMultimap<String, String> attributes) {
 44        for (Map.Entry<String, String> attribute : attributes.entries()) {
 45            final String key = attribute.getKey();
 46            final String value = attribute.getValue();
 47            s.append("a=").append(key);
 48            if (!Strings.isNullOrEmpty(value)) {
 49                s.append(':').append(value);
 50            }
 51            s.append(LINE_DIVIDER);
 52        }
 53    }
 54
 55    public static SessionDescription parse(final Map<String, RtpContentMap.DescriptionTransport> contents) {
 56        final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
 57        return sessionDescriptionBuilder.createSessionDescription();
 58    }
 59
 60    public static SessionDescription parse(final String input) {
 61        final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
 62        MediaBuilder currentMediaBuilder = null;
 63        ArrayListMultimap<String, String> attributeMap = ArrayListMultimap.create();
 64        ImmutableList.Builder<Media> mediaBuilder = new ImmutableList.Builder<>();
 65        for (final String line : input.split(LINE_DIVIDER)) {
 66            final String[] pair = line.trim().split("=", 2);
 67            if (pair.length < 2 || pair[0].length() != 1) {
 68                Log.d(Config.LOGTAG, "skipping sdp parsing on line " + line);
 69                continue;
 70            }
 71            final char key = pair[0].charAt(0);
 72            final String value = pair[1];
 73            switch (key) {
 74                case 'v':
 75                    sessionDescriptionBuilder.setVersion(ignorantIntParser(value));
 76                    break;
 77                case 'c':
 78                    if (currentMediaBuilder != null) {
 79                        currentMediaBuilder.setConnectionData(value);
 80                    } else {
 81                        sessionDescriptionBuilder.setConnectionData(value);
 82                    }
 83                    break;
 84                case 's':
 85                    sessionDescriptionBuilder.setName(value);
 86                    break;
 87                case 'a':
 88                    final Pair<String, String> attribute = parseAttribute(value);
 89                    attributeMap.put(attribute.first, attribute.second);
 90                    break;
 91                case 'm':
 92                    if (currentMediaBuilder == null) {
 93                        sessionDescriptionBuilder.setAttributes(attributeMap);
 94                        ;
 95                    } else {
 96                        currentMediaBuilder.setAttributes(attributeMap);
 97                        mediaBuilder.add(currentMediaBuilder.createMedia());
 98                    }
 99                    attributeMap = ArrayListMultimap.create();
100                    currentMediaBuilder = new MediaBuilder();
101                    final String[] parts = value.split(" ");
102                    if (parts.length >= 3) {
103                        currentMediaBuilder.setMedia(parts[0]);
104                        currentMediaBuilder.setPort(ignorantIntParser(parts[1]));
105                        currentMediaBuilder.setProtocol(parts[2]);
106                        ImmutableList.Builder<Integer> formats = new ImmutableList.Builder<>();
107                        for (int i = 3; i < parts.length; ++i) {
108                            formats.add(ignorantIntParser(parts[i]));
109                        }
110                        currentMediaBuilder.setFormats(formats.build());
111                    } else {
112                        Log.d(Config.LOGTAG, "skipping media line " + line);
113                    }
114                    break;
115            }
116
117        }
118        if (currentMediaBuilder != null) {
119            currentMediaBuilder.setAttributes(attributeMap);
120            mediaBuilder.add(currentMediaBuilder.createMedia());
121        } else {
122            sessionDescriptionBuilder.setAttributes(attributeMap);
123        }
124        sessionDescriptionBuilder.setMedia(mediaBuilder.build());
125        return sessionDescriptionBuilder.createSessionDescription();
126    }
127
128    public static SessionDescription of(final RtpContentMap contentMap) {
129        final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
130        final ArrayListMultimap<String, String> attributeMap = ArrayListMultimap.create();
131        final ImmutableList.Builder<Media> mediaListBuilder = new ImmutableList.Builder<>();
132        final Group group = contentMap.group;
133        if (group != null) {
134            attributeMap.put("group", group.getSemantics() + " " + Joiner.on(' ').join(group.getIdentificationTags()));
135        }
136
137        attributeMap.put("msid-semantic", " WMS my-media-stream");
138
139        for (Map.Entry<String, RtpContentMap.DescriptionTransport> entry : contentMap.contents.entrySet()) {
140            final String name = entry.getKey();
141            RtpContentMap.DescriptionTransport descriptionTransport = entry.getValue();
142            RtpDescription description = descriptionTransport.description;
143            IceUdpTransportInfo transport = descriptionTransport.transport;
144            final ArrayListMultimap<String, String> mediaAttributes = ArrayListMultimap.create();
145            final String ufrag = transport.getAttribute("ufrag");
146            final String pwd = transport.getAttribute("pwd");
147            if (!Strings.isNullOrEmpty(ufrag)) {
148                mediaAttributes.put("ice-ufrag", ufrag);
149            }
150            if (!Strings.isNullOrEmpty(pwd)) {
151                mediaAttributes.put("ice-pwd", pwd);
152            }
153            mediaAttributes.put("ice-options", HARDCODED_ICE_OPTIONS);
154            final IceUdpTransportInfo.Fingerprint fingerprint = transport.getFingerprint();
155            if (fingerprint != null) {
156                mediaAttributes.put("fingerprint", fingerprint.getHash() + " " + fingerprint.getContent());
157                mediaAttributes.put("setup", fingerprint.getSetup());
158            }
159            final ImmutableList.Builder<Integer> formatBuilder = new ImmutableList.Builder<>();
160            for (RtpDescription.PayloadType payloadType : description.getPayloadTypes()) {
161                formatBuilder.add(payloadType.getIntId());
162                mediaAttributes.put("rtpmap", payloadType.toSdpAttribute());
163                List<RtpDescription.Parameter> parameters = payloadType.getParameters();
164                if (parameters.size() > 0) {
165                    mediaAttributes.put("fmtp", RtpDescription.Parameter.toSdpString(payloadType.getId(), parameters));
166                }
167                for (RtpDescription.FeedbackNegotiation feedbackNegotiation : payloadType.getFeedbackNegotiations()) {
168                    mediaAttributes.put("rtcp-fb", payloadType.getId() + " " + feedbackNegotiation.getType() + (Strings.isNullOrEmpty(feedbackNegotiation.getSubType()) ? "" : " " + feedbackNegotiation.getSubType()));
169                }
170                for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : payloadType.feedbackNegotiationTrrInts()) {
171                    mediaAttributes.put("rtcp-fb", payloadType.getId() + " trr-int " + feedbackNegotiationTrrInt.getValue());
172                }
173            }
174
175            for (RtpDescription.FeedbackNegotiation feedbackNegotiation : description.getFeedbackNegotiations()) {
176                mediaAttributes.put("rtcp-fb", "* " + feedbackNegotiation.getType() + (Strings.isNullOrEmpty(feedbackNegotiation.getSubType()) ? "" : " " + feedbackNegotiation.getSubType()));
177            }
178            for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : description.feedbackNegotiationTrrInts()) {
179                mediaAttributes.put("rtcp-fb", "* trr-int " + feedbackNegotiationTrrInt.getValue());
180            }
181            for (RtpDescription.RtpHeaderExtension extension : description.getHeaderExtensions()) {
182                mediaAttributes.put("extmap", extension.getId() + " " + extension.getUri());
183            }
184            for (RtpDescription.Source source : description.getSources()) {
185                for (RtpDescription.Source.Parameter parameter : source.getParameters()) {
186                    mediaAttributes.put("ssrc", source.getSsrcId() + " " + parameter.getParameterName() + ":" + parameter.getParameterValue());
187                }
188            }
189
190            mediaAttributes.put("mid", name);
191
192            //random additional attributes
193            mediaAttributes.put("rtcp", "9 IN IP4 0.0.0.0");
194            mediaAttributes.put("sendrecv", "");
195            mediaAttributes.put("rtcp-mux", "");
196
197            final MediaBuilder mediaBuilder = new MediaBuilder();
198            mediaBuilder.setMedia(description.getMedia().toString().toLowerCase(Locale.ROOT));
199            mediaBuilder.setConnectionData(HARDCODED_CONNECTION);
200            mediaBuilder.setPort(HARDCODED_MEDIA_PORT);
201            mediaBuilder.setProtocol(HARDCODED_MEDIA_PROTOCOL);
202            mediaBuilder.setAttributes(mediaAttributes);
203            mediaBuilder.setFormats(formatBuilder.build());
204            mediaListBuilder.add(mediaBuilder.createMedia());
205
206        }
207        sessionDescriptionBuilder.setVersion(0);
208        sessionDescriptionBuilder.setName("-");
209        sessionDescriptionBuilder.setMedia(mediaListBuilder.build());
210        sessionDescriptionBuilder.setAttributes(attributeMap);
211
212        return sessionDescriptionBuilder.createSessionDescription();
213    }
214
215    public static int ignorantIntParser(final String input) {
216        try {
217            return Integer.parseInt(input);
218        } catch (NumberFormatException e) {
219            return 0;
220        }
221    }
222
223    public static Pair<String, String> parseAttribute(final String input) {
224        final String[] pair = input.split(":", 2);
225        if (pair.length == 2) {
226            return new Pair<>(pair[0], pair[1]);
227        } else {
228            return new Pair<>(pair[0], "");
229        }
230    }
231
232    @Override
233    public String toString() {
234        final StringBuilder s = new StringBuilder()
235                .append("v=").append(version).append(LINE_DIVIDER)
236                .append("o=- 8770656990916039506 2 IN IP4 127.0.0.1").append(LINE_DIVIDER) //what ever that means
237                .append("s=").append(name).append(LINE_DIVIDER)
238                .append("t=0 0").append(LINE_DIVIDER);
239        appendAttributes(s, attributes);
240        for (Media media : this.media) {
241            s.append("m=").append(media.media).append(' ').append(media.port).append(' ').append(media.protocol).append(' ').append(Joiner.on(' ').join(media.formats)).append(LINE_DIVIDER);
242            s.append("c=").append(media.connectionData).append(LINE_DIVIDER);
243            appendAttributes(s, media.attributes);
244        }
245        return s.toString();
246    }
247
248    public static class Media {
249        public final String media;
250        public final int port;
251        public final String protocol;
252        public final List<Integer> formats;
253        public final String connectionData;
254        public final ArrayListMultimap<String, String> attributes;
255
256        public Media(String media, int port, String protocol, List<Integer> formats, String connectionData, ArrayListMultimap<String, String> attributes) {
257            this.media = media;
258            this.port = port;
259            this.protocol = protocol;
260            this.formats = formats;
261            this.connectionData = connectionData;
262            this.attributes = attributes;
263        }
264    }
265
266}