/protocols/smpp/src/main/java/org/mobicents/protocols/smpp/gsm/ConcatenatedSms.java

http://mobicents.googlecode.com/ · Java · 92 lines · 57 code · 10 blank · 25 comment · 7 complexity · 0e8351c49ec17646fb65540c9538ffcc MD5 · raw file

  1. /*
  2. * JBoss, Home of Professional Open Source
  3. * Copyright 2011, Red Hat, Inc. and individual contributors
  4. * by the @authors tag. See the copyright.txt in the distribution for a
  5. * full listing of individual contributors.
  6. *
  7. * This is free software; you can redistribute it and/or modify it
  8. * under the terms of the GNU Lesser General Public License as
  9. * published by the Free Software Foundation; either version 2.1 of
  10. * the License, or (at your option) any later version.
  11. *
  12. * This software is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with this software; if not, write to the Free
  19. * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  20. * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
  21. */
  22. package org.mobicents.protocols.smpp.gsm;
  23. import java.nio.ByteBuffer;
  24. import java.security.SecureRandom;
  25. import java.util.HashMap;
  26. import java.util.List;
  27. import java.util.Map;
  28. import java.util.Random;
  29. public class ConcatenatedSms extends RecurringHeaderElement {
  30. private static final Random RANDOM = new SecureRandom();
  31. private int referenceNumber = 1;
  32. private boolean use16bitRef;
  33. /**
  34. * Store the location in each buffer where we need to go back
  35. * end update the correct number of segments being sent.
  36. */
  37. private Map<ByteBuffer, Integer> totalMsgs =
  38. new HashMap<ByteBuffer, Integer>();
  39. public ConcatenatedSms(boolean use16bitRef) {
  40. super(true);
  41. this.use16bitRef = use16bitRef;
  42. reset();
  43. }
  44. public boolean doWrite(int segmentNum, ByteBuffer buffer) {
  45. if (use16bitRef) {
  46. buffer.put((byte) 8);
  47. buffer.put((byte) 4);
  48. buffer.putShort((short) referenceNumber);
  49. } else {
  50. buffer.put((byte) 0);
  51. buffer.put((byte) 3);
  52. buffer.put((byte) referenceNumber);
  53. }
  54. totalMsgs.put(buffer, buffer.position());
  55. buffer.put((byte) 1);
  56. buffer.put((byte) segmentNum);
  57. return true;
  58. }
  59. public int getLength() {
  60. if (use16bitRef) {
  61. return 4;
  62. } else {
  63. return 3;
  64. }
  65. }
  66. @Override
  67. public void reset() {
  68. super.reset();
  69. int max = use16bitRef ? 65535 : 255;
  70. referenceNumber = RANDOM.nextInt(max);
  71. }
  72. @Override
  73. public void postProcess(List<ByteBuffer> segments) {
  74. int numSegments = segments.size();
  75. if (numSegments != totalMsgs.size()) {
  76. throw new IllegalStateException("Cannot update all segments");
  77. }
  78. for (Map.Entry<ByteBuffer, Integer> entry : totalMsgs.entrySet()) {
  79. entry.getKey().put(entry.getValue(), (byte) numSegments);
  80. }
  81. }
  82. }