73 lines
2.2 KiB
Java
73 lines
2.2 KiB
Java
/* MySendCallback.java
|
|
*
|
|
* Copyright (C) 2006-2022 wolfSSL Inc.
|
|
*
|
|
* This file is part of wolfSSL.
|
|
*
|
|
* wolfSSL is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* wolfSSL is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
import java.io.*;
|
|
import java.net.*;
|
|
import java.nio.*;
|
|
import com.wolfssl.*;
|
|
|
|
class MySendCallback implements WolfSSLIOSendCallback
|
|
{
|
|
public int sendCallback(WolfSSLSession ssl, byte[] buf, int sz,
|
|
Object ctx) {
|
|
|
|
MyIOCtx ioctx = (MyIOCtx) ctx;
|
|
int doDTLS = ioctx.isDTLS();
|
|
|
|
if (doDTLS == 1) {
|
|
|
|
DatagramSocket dsock = ioctx.getDatagramSocket();
|
|
InetAddress hostAddr = ioctx.getHostAddress();
|
|
int port = ioctx.getPort();
|
|
DatagramPacket dp = new DatagramPacket(buf, sz, hostAddr, port);
|
|
|
|
try {
|
|
dsock.send(dp);
|
|
|
|
} catch (IOException ioe) {
|
|
ioe.printStackTrace();
|
|
return WolfSSL.WOLFSSL_CBIO_ERR_GENERAL;
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return WolfSSL.WOLFSSL_CBIO_ERR_GENERAL;
|
|
}
|
|
|
|
return dp.getLength();
|
|
} else {
|
|
DataOutputStream os = ioctx.getOutputStream();
|
|
if (os == null) {
|
|
System.out.println("DataOutputStream is null in sendCallback!");
|
|
System.exit(1);
|
|
}
|
|
|
|
try {
|
|
os.write(buf, 0, sz);
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
return WolfSSL.WOLFSSL_CBIO_ERR_GENERAL;
|
|
}
|
|
}
|
|
|
|
return sz;
|
|
}
|
|
}
|
|
|