2015-03-16 07:46:14 +01:00
|
|
|
/*
|
|
|
|
* Copyright 2015 The Netty Project
|
|
|
|
*
|
|
|
|
* The Netty Project licenses this file to you under the Apache License,
|
|
|
|
* version 2.0 (the "License"); you may not use this file except in compliance
|
|
|
|
* with the License. You may obtain a copy of the License at:
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
|
|
* License for the specific language governing permissions and limitations
|
|
|
|
* under the License.
|
|
|
|
*/
|
|
|
|
package io.netty.handler.codec.dns;
|
|
|
|
|
|
|
|
import io.netty.buffer.ByteBuf;
|
|
|
|
import io.netty.buffer.ByteBufUtil;
|
|
|
|
import io.netty.handler.codec.UnsupportedMessageTypeException;
|
|
|
|
import io.netty.util.internal.StringUtil;
|
2016-04-12 14:22:41 +02:00
|
|
|
import io.netty.util.internal.UnstableApi;
|
2015-03-16 07:46:14 +01:00
|
|
|
|
2016-04-01 13:47:15 +02:00
|
|
|
import static io.netty.handler.codec.dns.DefaultDnsRecordDecoder.ROOT;
|
|
|
|
|
2015-03-16 07:46:14 +01:00
|
|
|
/**
|
|
|
|
* The default {@link DnsRecordEncoder} implementation.
|
|
|
|
*
|
|
|
|
* @see DefaultDnsRecordDecoder
|
|
|
|
*/
|
2016-04-12 14:22:41 +02:00
|
|
|
@UnstableApi
|
2015-03-16 07:46:14 +01:00
|
|
|
public class DefaultDnsRecordEncoder implements DnsRecordEncoder {
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Creates a new instance.
|
|
|
|
*/
|
|
|
|
protected DefaultDnsRecordEncoder() { }
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public final void encodeQuestion(DnsQuestion question, ByteBuf out) throws Exception {
|
|
|
|
encodeName(question.name(), out);
|
|
|
|
out.writeShort(question.type().intValue());
|
|
|
|
out.writeShort(question.dnsClass());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void encodeRecord(DnsRecord record, ByteBuf out) throws Exception {
|
|
|
|
if (record instanceof DnsQuestion) {
|
|
|
|
encodeQuestion((DnsQuestion) record, out);
|
2016-02-28 20:33:16 +01:00
|
|
|
} else if (record instanceof DnsPtrRecord) {
|
|
|
|
encodePtrRecord((DnsPtrRecord) record, out);
|
2015-03-16 07:46:14 +01:00
|
|
|
} else if (record instanceof DnsRawRecord) {
|
|
|
|
encodeRawRecord((DnsRawRecord) record, out);
|
|
|
|
} else {
|
|
|
|
throw new UnsupportedMessageTypeException(StringUtil.simpleClassName(record));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-02-28 20:33:16 +01:00
|
|
|
private void encodePtrRecord(DnsPtrRecord record, ByteBuf out) throws Exception {
|
|
|
|
encodeName(record.name(), out);
|
|
|
|
|
|
|
|
out.writeShort(record.type().intValue());
|
|
|
|
out.writeShort(record.dnsClass());
|
|
|
|
out.writeInt((int) record.timeToLive());
|
|
|
|
|
|
|
|
encodeName(record.hostname(), out);
|
|
|
|
}
|
|
|
|
|
2015-03-16 07:46:14 +01:00
|
|
|
private void encodeRawRecord(DnsRawRecord record, ByteBuf out) throws Exception {
|
|
|
|
encodeName(record.name(), out);
|
|
|
|
|
|
|
|
out.writeShort(record.type().intValue());
|
|
|
|
out.writeShort(record.dnsClass());
|
|
|
|
out.writeInt((int) record.timeToLive());
|
|
|
|
|
|
|
|
ByteBuf content = record.content();
|
|
|
|
int contentLen = content.readableBytes();
|
|
|
|
|
|
|
|
out.writeShort(contentLen);
|
|
|
|
out.writeBytes(content, content.readerIndex(), contentLen);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected void encodeName(String name, ByteBuf buf) throws Exception {
|
2016-04-01 13:47:15 +02:00
|
|
|
if (ROOT.equals(name)) {
|
|
|
|
// Root domain
|
|
|
|
buf.writeByte(0);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2016-08-01 11:17:05 +02:00
|
|
|
final String[] labels = name.split("\\.");
|
2016-04-01 13:47:15 +02:00
|
|
|
for (String label : labels) {
|
|
|
|
final int labelLen = label.length();
|
|
|
|
if (labelLen == 0) {
|
|
|
|
// zero-length label means the end of the name.
|
|
|
|
break;
|
2015-03-16 07:46:14 +01:00
|
|
|
}
|
2016-04-01 13:47:15 +02:00
|
|
|
|
|
|
|
buf.writeByte(labelLen);
|
|
|
|
ByteBufUtil.writeAscii(buf, label);
|
2015-03-16 07:46:14 +01:00
|
|
|
}
|
2016-04-01 13:47:15 +02:00
|
|
|
|
2015-03-16 07:46:14 +01:00
|
|
|
buf.writeByte(0); // marks end of name field
|
|
|
|
}
|
|
|
|
}
|