LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge "Fix empty value problem in the TextEntry"

Browse Source
This commit is contained in:
Paul Hu
2022-12-06 05:31:39 +00:00
committed by Gerrit Code Review
parent a43bba1277 536cc4d102
commit ed3e683407
2 changed files with 28 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
service/mdns/com/android/server/connectivity/mdns/MdnsServiceInfo.java
View File

@@ -270,7 +270,8 @@ public class MdnsServiceInfo implements Parcelable {
public Map<String, String> getAttributes() {
Map<String, String> map = new HashMap<>(attributes.size());
for (Map.Entry<String, byte[]> kv : attributes.entrySet()) {
map.put(kv.getKey(), new String(kv.getValue(), UTF_8));
final byte[] value = kv.getValue();
map.put(kv.getKey(), value == null ? null : new String(value, UTF_8));
}
return Collections.unmodifiableMap(map);
}
@@ -342,7 +343,7 @@ public class MdnsServiceInfo implements Parcelable {
// 2. If there is no '=' in a DNS-SD TXT record string, then it is a
// boolean attribute, simply identified as being present, with no value.
if (delimitPos < 0) {
return new TextEntry(new String(textBytes, US_ASCII), "");
return new TextEntry(new String(textBytes, US_ASCII), (byte[]) null);
} else if (delimitPos == 0) {
return null;
}
@@ -353,13 +354,13 @@ public class MdnsServiceInfo implements Parcelable {
/** Creates a new {@link TextEntry} with given key and value of a UTF-8 string. */
public TextEntry(String key, String value) {
this(key, value.getBytes(UTF_8));
this(key, value == null ? null : value.getBytes(UTF_8));
}
/** Creates a new {@link TextEntry} with given key and value of a byte array. */
public TextEntry(String key, byte[] value) {
this.key = key;
this.value = value.clone();
this.value = value == null ? null : value.clone();
}
private TextEntry(Parcel in) {
@@ -372,17 +373,24 @@ public class MdnsServiceInfo implements Parcelable {
}
public byte[] getValue() {
return value.clone();
return value == null ? null : value.clone();
}
/** Converts this {@link TextEntry} instance to '=' separated byte array. */
public byte[] toBytes() {
return ByteUtils.concat(key.getBytes(US_ASCII), new byte[]{'='}, value);
final byte[] keyBytes = key.getBytes(US_ASCII);
if (value == null) {
return keyBytes;
}
return ByteUtils.concat(keyBytes, new byte[]{'='}, value);
}
/** Converts this {@link TextEntry} instance to '=' separated string. */
@Override
public String toString() {
if (value == null) {
return key;
}
return key + "=" + new String(value, UTF_8);
}

19
tests/unit/java/com/android/server/connectivity/mdns/MdnsServiceInfoTest.java
View File

@@ -164,7 +164,8 @@ public class MdnsServiceInfoTest {
List.of("vn=Alphabet Inc.", "mn=Google Nest Hub Max", "id=12345"),
List.of(
MdnsServiceInfo.TextEntry.fromString("vn=Google Inc."),
MdnsServiceInfo.TextEntry.fromString("mn=Google Nest Hub Max")));
MdnsServiceInfo.TextEntry.fromString("mn=Google Nest Hub Max"),
MdnsServiceInfo.TextEntry.fromString("test=")));
beforeParcel.writeToParcel(parcel, 0);
parcel.setDataPosition(0);
@@ -208,11 +209,11 @@ public class MdnsServiceInfoTest {
}
@Test
public void textEntry_fromStringWithoutAssignPunc_valueisEmpty() {
public void textEntry_fromStringWithoutAssignPunc_noValue() {
TextEntry entry = TextEntry.fromString("AA");
assertEquals("AA", entry.getKey());
assertArrayEquals(new byte[] {}, entry.getValue());
assertNull(entry.getValue());
}
@Test
@@ -241,11 +242,11 @@ public class MdnsServiceInfoTest {
}
@Test
public void textEntry_fromBytesWithoutAssignPunc_valueisEmpty() {
public void textEntry_fromBytesWithoutAssignPunc_noValue() {
TextEntry entry = TextEntry.fromBytes(new byte[] {'A', 'A'});
assertEquals("AA", entry.getKey());
assertArrayEquals(new byte[] {}, entry.getValue());
assertNull(entry.getValue());
}
@Test
@@ -270,4 +271,12 @@ public class MdnsServiceInfoTest {
assertEquals(new TextEntry("BB", "xxyyzz"), new TextEntry("BB", "xxyyzz"));
assertEquals(new TextEntry("AA", "XXYYZZ"), new TextEntry("AA", "XXYYZZ"));
}
@Test
public void textEntry_fromString_valueIsEmpty() {
TextEntry entry = TextEntry.fromString("AA=");
assertEquals("AA", entry.getKey());
assertArrayEquals(new byte[] {}, entry.getValue());
}
}