smpp_transmitter.go

// GoSMPP - An SMPP library for Go
// Copyright 2010 Phil Bayfield
// This software is licensed under a Creative Commons Attribution-Share Alike 2.0 UK: England & Wales License
// Further information on this license can be found here: http://creativecommons.org/licenses/by-sa/2.0/uk/
package smpp

import (
	"errors"
	"fmt"
	"reflect"
)

// Transmitter type
type Transmitter struct {
	smpp
}

// Submit SM
func (tx *Transmitter) SubmitSM(dest, msg string, params Params, optional ...OptParams) (sequence uint32, msgId string, err error) {
	// Check connected and bound
	if !tx.connected || !tx.bound {
		err = errors.New("SubmitSM: A bound connection is required to submit a message")
		return
	}
	// Check destination number and message
	if dest == "" {
		err = errors.New("SubmitSM: A destination number is required and should not be null")
		return
	}
	// Merge params with defaults
	allParams := mergeParams(params, defaultsSubmitSM)
	// Increment sequence number
	tx.sequence++
	// PDU header
	hdr := new(PDUHeader)
	hdr.CmdLength = 34
	hdr.CmdId = CMD_SUBMIT_SM
	hdr.CmdStatus = STATUS_ESME_ROK
	hdr.Sequence = tx.sequence
	// Mising params cause panic, this provides a clean error/exit
	paramOK := false
	defer func() {
		if !paramOK && recover() != nil {
			err = errors.New("SubmitSM: Panic, invalid params")
			return
		}
	}()
	// Create new PDU
	pdu := new(PDUSubmitSM)
	// Populate params
	pdu.ServiceType = allParams["serviceType"].(string)
	pdu.SourceAddrTon = allParams["sourceAddrTon"].(SMPPTypeOfNumber)
	pdu.SourceAddrNpi = allParams["sourceAddrNpi"].(SMPPNumericPlanIndicator)
	pdu.SourceAddr = allParams["sourceAddr"].(string)
	pdu.DestAddrTon = allParams["destAddrTon"].(SMPPTypeOfNumber)
	pdu.DestAddrNpi = allParams["destAddrNpi"].(SMPPNumericPlanIndicator)
	pdu.DestAddr = dest
	pdu.EsmClass = allParams["esmClass"].(SMPPEsmClassESME)
	pdu.ProtocolId = allParams["protocolId"].(uint8)
	pdu.PriorityFlag = allParams["priorityFlag"].(SMPPPriority)
	pdu.SchedDelTime = allParams["schedDelTime"].(string)
	pdu.ValidityPeriod = allParams["validityPeriod"].(string)
	pdu.RegDelivery = allParams["regDelivery"].(SMPPDelivery)
	pdu.ReplaceFlag = allParams["replaceFlag"].(uint8)
	pdu.DataCoding = allParams["dataCoding"].(SMPPDataCoding)
	pdu.SmDefaultMsgId = allParams["smDefaultMsgId"].(uint8)
	pdu.SmLength = uint8(len(msg))
	pdu.ShortMessage = msg
	// Add length of strings to pdu length
	hdr.CmdLength += uint32(len(pdu.ServiceType))
	hdr.CmdLength += uint32(len(pdu.SourceAddr))
	hdr.CmdLength += uint32(len(pdu.DestAddr))
	hdr.CmdLength += uint32(len(pdu.SchedDelTime))
	hdr.CmdLength += uint32(len(pdu.ValidityPeriod))
	hdr.CmdLength += uint32(len(pdu.ShortMessage))
	// Calculate size of optional params
	if len(optional) > 0 && len(optional[0]) > 0 {
		pdu.Optional = optional[0]
		for _, val := range optional[0] {
			v := reflect.ValueOf(val)
			switch t := v; t.Kind() {
			default:
				err = errors.New("SubmitSM: Invalid optional param format")
				return
			case reflect.String:
				hdr.CmdLength += uint32(len(val.(string)))
				pdu.OptionalLen += uint32(len(val.(string)))
			case reflect.Uint8:
				hdr.CmdLength++
				pdu.OptionalLen++
			case reflect.Uint16:
				hdr.CmdLength += 2
				pdu.OptionalLen += 2
			case reflect.Uint32:
				hdr.CmdLength += 4
				pdu.OptionalLen += 4
			}
			// Add 4 bytes for optional param header
			hdr.CmdLength += 4
			pdu.OptionalLen += 4
		}
	}
	// Params were fine 'disable' the recover
	paramOK = true
	// Send PDU
	pdu.setHeader(hdr)
	err = pdu.write(tx.writer)
	if err != nil {
		return
	}
	// If not async get the response
	if tx.async {
		sequence = tx.sequence
	} else {
		rpdu, e := tx.GetResp(CMD_SUBMIT_SM_RESP, tx.sequence)
		if e != nil {
			err = e
			return
		}
		s := rpdu.GetStruct()
		msgId = s.(PDUSubmitSMResp).MessageId
	}
	return
}

// Submit Multi
func (tx *Transmitter) SubmitMulti(destNum, destList []string, msg string, params Params, optional ...OptParams) (sequence uint32, msgId string, unsuccess []string, err error) {
	// Check connected and bound
	if !tx.connected || !tx.bound {
		err = errors.New("SubmitMulti: A bound connection is required to submit a message")
		return
	}
	// Check destination number and message
	if len(destNum) == 0 && len(destList) == 0 {
		err = errors.New("SubmitMulti: At least 1 destination number or list is required")
		return
	}
	// Merge params with defaults
	allParams := mergeParams(params, defaultsSubmitMulti)
	// Increment sequence number
	tx.sequence++
	// PDU header
	hdr := new(PDUHeader)
	hdr.CmdLength = 32
	hdr.CmdId = CMD_SUBMIT_MULTI
	hdr.CmdStatus = STATUS_ESME_ROK
	hdr.Sequence = tx.sequence
	// Mising params cause panic, this provides a clean error/exit
	paramOK := false
	defer func() {
		if !paramOK && recover() != nil {
			err = errors.New("SubmitMulti: Panic, invalid params")
			return
		}
	}()
	// Create new PDU
	pdu := new(PDUSubmitMulti)
	// Populate params
	pdu.ServiceType = allParams["serviceType"].(string)
	pdu.SourceAddrTon = allParams["sourceAddrTon"].(SMPPTypeOfNumber)
	pdu.SourceAddrNpi = allParams["sourceAddrNpi"].(SMPPNumericPlanIndicator)
	pdu.SourceAddr = allParams["sourceAddr"].(string)
	pdu.NumOfDests = uint8(len(destNum) + len(destList))
	pdu.DestAddrTon = allParams["destAddrTon"].(SMPPTypeOfNumber)
	pdu.DestAddrNpi = allParams["destAddrNpi"].(SMPPNumericPlanIndicator)
	pdu.DestAddrs = destNum
	pdu.DestLists = destList
	pdu.EsmClass = allParams["esmClass"].(SMPPEsmClassESME)
	pdu.ProtocolId = allParams["protocolId"].(uint8)
	pdu.PriorityFlag = allParams["priorityFlag"].(SMPPPriority)
	pdu.SchedDelTime = allParams["schedDelTime"].(string)
	pdu.ValidityPeriod = allParams["validityPeriod"].(string)
	pdu.RegDelivery = allParams["regDelivery"].(SMPPDelivery)
	pdu.ReplaceFlag = allParams["replaceFlag"].(uint8)
	pdu.DataCoding = allParams["dataCoding"].(SMPPDataCoding)
	pdu.SmDefaultMsgId = allParams["smDefaultMsgId"].(uint8)
	pdu.SmLength = uint8(len(msg))
	pdu.ShortMessage = msg
	// Add length of strings to pdu length
	hdr.CmdLength += uint32(len(pdu.ServiceType))
	hdr.CmdLength += uint32(len(pdu.SourceAddr))
	hdr.CmdLength += uint32(len(pdu.SchedDelTime))
	hdr.CmdLength += uint32(len(pdu.ValidityPeriod))
	hdr.CmdLength += uint32(len(pdu.ShortMessage))
	// Add length of any numbers to the pdu length
	if len(destNum) > 0 {
		for _, num := range destNum {
			hdr.CmdLength += uint32(len(num)) + 4
		}
	}
	// Add length of any list names to the pdu length
	if len(destList) > 0 {
		for _, list := range destList {
			hdr.CmdLength += uint32(len(list)) + 2
		}
	}
	// Calculate size of optional params
	if len(optional) > 0 && len(optional[0]) > 0 {
		pdu.Optional = optional[0]
		for _, val := range optional[0] {
			v := reflect.ValueOf(val)
			switch t := v; t.Kind() {
			default:
				err = errors.New("SubmitMulti: Invalid optional param format")
				return
			case reflect.String:
				hdr.CmdLength += uint32(len(val.(string)))
				pdu.OptionalLen += uint32(len(val.(string)))
			case reflect.Uint8:
				hdr.CmdLength++
				pdu.OptionalLen++
			case reflect.Uint16:
				hdr.CmdLength += 2
				pdu.OptionalLen += 2
			case reflect.Uint32:
				hdr.CmdLength += 4
				pdu.OptionalLen += 4
			}
			// Add 4 bytes for optional param header
			hdr.CmdLength += 4
			pdu.OptionalLen += 4
		}
	}
	// Params were fine 'disable' the recover
	paramOK = true
	// Send PDU
	pdu.setHeader(hdr)
	err = pdu.write(tx.writer)
	if err != nil {
		return
	}
	// If not async get the response
	if tx.async {
		sequence = tx.sequence
	} else {
		rpdu, e := tx.GetResp(CMD_SUBMIT_MULTI_RESP, tx.sequence)
		if e != nil {
			fmt.Printf("Result: %#v, %#v\n", rpdu, e)
			err = e
			return
		}
		s := rpdu.GetStruct()
		msgId = s.(PDUSubmitMultiResp).MessageId
		unsuccess = s.(PDUSubmitMultiResp).Unsuccess
	}
	return
}