| /* |
| * Copyright (C) 2019, The Android Open Source Project |
| * |
| * Licensed 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. |
| */ |
| |
| #include "java_writer.h" |
| #include "java_writer_q.h" |
| #include "utils.h" |
| |
| namespace android { |
| namespace stats_log_api_gen { |
| |
| static int write_java_q_logger_class( |
| FILE* out, |
| const map<vector<java_type_t>, FieldNumberToAnnotations>& signatureInfoMap, |
| const AtomDecl &attributionDecl) { |
| fprintf(out, "\n"); |
| fprintf(out, " // Write logging helper methods for statsd in Q and earlier.\n"); |
| fprintf(out, " private static class QLogger {\n"); |
| |
| write_java_q_logging_constants(out, " "); |
| |
| // Print Q write methods. |
| fprintf(out, "\n"); |
| fprintf(out, " // Write methods.\n"); |
| write_java_methods_q_schema( |
| out, signatureInfoMap, attributionDecl, " "); |
| |
| fprintf(out, " }\n"); |
| return 0; |
| } |
| |
| static void write_annotations( |
| FILE* out, int argIndex, |
| const FieldNumberToAnnotations& fieldNumberToAnnotations) { |
| auto it = fieldNumberToAnnotations.find(argIndex); |
| if (it == fieldNumberToAnnotations.end()) { |
| return; |
| } |
| const set<shared_ptr<Annotation>>& annotations = it->second; |
| for (auto& annotation: annotations) { |
| // TODO(b/151744250): Group annotations for same atoms. |
| // TODO(b/151786433): Write atom constant name instead of atom id literal. |
| fprintf(out, " if (code == %d) {\n", annotation->atomId); |
| switch(annotation->type) { |
| case ANNOTATION_TYPE_INT: |
| // TODO(b/151776731): Check for reset state annotation and only include reset state |
| // when field value == default state annotation value. |
| // TODO(b/151786433): Write annotation constant name instead of |
| // annotation id literal. |
| fprintf(out, " builder.addIntAnnotation((byte) %d, %d);\n", |
| annotation->annotationId, annotation->value.intValue); |
| break; |
| case ANNOTATION_TYPE_BOOL: |
| // TODO(b/151786433): Write annotation constant name instead of |
| // annotation id literal. |
| fprintf(out, " builder.addBooleanAnnotation((byte) %d, %s);\n", |
| annotation->annotationId, |
| annotation->value.boolValue ? "true" : "false"); |
| break; |
| default: |
| break; |
| } |
| fprintf(out, " }\n"); |
| } |
| } |
| |
| static int write_java_methods( |
| FILE* out, |
| const map<vector<java_type_t>, FieldNumberToAnnotations>& signatureInfoMap, |
| const AtomDecl &attributionDecl, |
| const bool supportQ |
| ) { |
| for (auto signatureInfoMapIt = signatureInfoMap.begin(); |
| signatureInfoMapIt != signatureInfoMap.end(); signatureInfoMapIt++) { |
| // Print method signature. |
| fprintf(out, " public static void write(int code"); |
| const vector<java_type_t>& signature = signatureInfoMapIt->first; |
| const FieldNumberToAnnotations& fieldNumberToAnnotations = signatureInfoMapIt->second; |
| int argIndex = 1; |
| for (vector<java_type_t>::const_iterator arg = signature.begin(); |
| arg != signature.end(); arg++) { |
| if (*arg == JAVA_TYPE_ATTRIBUTION_CHAIN) { |
| for (auto chainField : attributionDecl.fields) { |
| fprintf(out, ", %s[] %s", |
| java_type_name(chainField.javaType), chainField.name.c_str()); |
| } |
| } else if (*arg == JAVA_TYPE_KEY_VALUE_PAIR) { |
| fprintf(out, ", android.util.SparseArray<Object> valueMap"); |
| } else { |
| fprintf(out, ", %s arg%d", java_type_name(*arg), argIndex); |
| } |
| argIndex++; |
| } |
| fprintf(out, ") {\n"); |
| |
| // Print method body. |
| string indent(""); |
| if (supportQ) { |
| // TODO(b/146235828): Use just SDK_INT check once it is incremented from Q. |
| fprintf(out, " if (Build.VERSION.SDK_INT > Build.VERSION_CODES.Q\n"); |
| fprintf(out, " || (Build.VERSION.SDK_INT == Build.VERSION_CODES.Q\n"); |
| fprintf(out, " && Build.VERSION.PREVIEW_SDK_INT > 0)) {\n"); |
| indent = " "; |
| } |
| |
| // Start StatsEvent.Builder. |
| fprintf(out, "%s final StatsEvent.Builder builder = StatsEvent.newBuilder();\n", |
| indent.c_str()); |
| |
| // Write atom code. |
| fprintf(out, "%s builder.setAtomId(code);\n", indent.c_str()); |
| |
| // Write the args. |
| argIndex = 1; |
| for (vector<java_type_t>::const_iterator arg = signature.begin(); |
| arg != signature.end(); arg++) { |
| switch (*arg) { |
| case JAVA_TYPE_BOOLEAN: |
| fprintf(out, "%s builder.writeBoolean(arg%d);\n", indent.c_str(), argIndex); |
| break; |
| case JAVA_TYPE_INT: |
| case JAVA_TYPE_ENUM: |
| fprintf(out, "%s builder.writeInt(arg%d);\n", indent.c_str(), argIndex); |
| break; |
| case JAVA_TYPE_FLOAT: |
| fprintf(out, "%s builder.writeFloat(arg%d);\n", indent.c_str(), argIndex); |
| break; |
| case JAVA_TYPE_LONG: |
| fprintf(out, "%s builder.writeLong(arg%d);\n", indent.c_str(), argIndex); |
| break; |
| case JAVA_TYPE_STRING: |
| fprintf(out, "%s builder.writeString(arg%d);\n", indent.c_str(), argIndex); |
| break; |
| case JAVA_TYPE_BYTE_ARRAY: |
| fprintf(out, "%s builder.writeByteArray(null == arg%d ? new byte[0] : arg%d);\n", |
| indent.c_str(), argIndex, argIndex); |
| break; |
| case JAVA_TYPE_ATTRIBUTION_CHAIN: |
| { |
| const char* uidName = attributionDecl.fields.front().name.c_str(); |
| const char* tagName = attributionDecl.fields.back().name.c_str(); |
| |
| fprintf(out, "%s builder.writeAttributionChain(\n", indent.c_str()); |
| fprintf(out, "%s null == %s ? new int[0] : %s,\n", |
| indent.c_str(), uidName, uidName); |
| fprintf(out, "%s null == %s ? new String[0] : %s);\n", |
| indent.c_str(), tagName, tagName); |
| break; |
| } |
| case JAVA_TYPE_KEY_VALUE_PAIR: |
| fprintf(out, "\n"); |
| fprintf(out, |
| "%s // Write KeyValuePairs.\n", indent.c_str()); |
| fprintf(out, |
| "%s final int count = valueMap.size();\n", indent.c_str()); |
| fprintf(out, |
| "%s android.util.SparseIntArray intMap = null;\n", |
| indent.c_str()); |
| fprintf(out, |
| "%s android.util.SparseLongArray longMap = null;\n", |
| indent.c_str()); |
| fprintf(out, |
| "%s android.util.SparseArray<String> stringMap = null;\n", |
| indent.c_str()); |
| fprintf(out, |
| "%s android.util.SparseArray<Float> floatMap = null;\n", |
| indent.c_str()); |
| fprintf(out, |
| "%s for (int i = 0; i < count; i++) {\n", indent.c_str()); |
| fprintf(out, |
| "%s final int key = valueMap.keyAt(i);\n", indent.c_str()); |
| fprintf(out, |
| "%s final Object value = valueMap.valueAt(i);\n", |
| indent.c_str()); |
| fprintf(out, |
| "%s if (value instanceof Integer) {\n", indent.c_str()); |
| fprintf(out, |
| "%s if (null == intMap) {\n", indent.c_str()); |
| fprintf(out, |
| "%s intMap = new android.util.SparseIntArray();\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s intMap.put(key, (Integer) value);\n", indent.c_str()); |
| fprintf(out, |
| "%s } else if (value instanceof Long) {\n", indent.c_str()); |
| fprintf(out, |
| "%s if (null == longMap) {\n", indent.c_str()); |
| fprintf(out, |
| "%s longMap = new android.util.SparseLongArray();\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s longMap.put(key, (Long) value);\n", indent.c_str()); |
| fprintf(out, |
| "%s } else if (value instanceof String) {\n", indent.c_str()); |
| fprintf(out, |
| "%s if (null == stringMap) {\n", indent.c_str()); |
| fprintf(out, |
| "%s stringMap = new android.util.SparseArray<>();\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s stringMap.put(key, (String) value);\n", indent.c_str()); |
| fprintf(out, |
| "%s } else if (value instanceof Float) {\n", indent.c_str()); |
| fprintf(out, |
| "%s if (null == floatMap) {\n", indent.c_str()); |
| fprintf(out, |
| "%s floatMap = new android.util.SparseArray<>();\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s floatMap.put(key, (Float) value);\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s }\n", indent.c_str()); |
| fprintf(out, |
| "%s builder.writeKeyValuePairs(" |
| "intMap, longMap, stringMap, floatMap);\n", indent.c_str()); |
| break; |
| default: |
| // Unsupported types: OBJECT, DOUBLE. |
| fprintf(stderr, "Encountered unsupported type."); |
| return 1; |
| } |
| write_annotations(out, argIndex, fieldNumberToAnnotations); |
| argIndex++; |
| } |
| |
| fprintf(out, "\n"); |
| fprintf(out, "%s builder.usePooledBuffer();\n", indent.c_str()); |
| fprintf(out, "%s StatsLog.write(builder.build());\n", indent.c_str()); |
| |
| // Add support for writing using Q schema if this is not the default module. |
| if (supportQ) { |
| fprintf(out, " } else {\n"); |
| fprintf(out, " QLogger.write(code"); |
| argIndex = 1; |
| for (vector<java_type_t>::const_iterator arg = signature.begin(); |
| arg != signature.end(); arg++) { |
| if (*arg == JAVA_TYPE_ATTRIBUTION_CHAIN) { |
| const char* uidName = attributionDecl.fields.front().name.c_str(); |
| const char* tagName = attributionDecl.fields.back().name.c_str(); |
| fprintf(out, ", %s, %s", uidName, tagName); |
| } else if (*arg == JAVA_TYPE_KEY_VALUE_PAIR) { |
| // Module logging does not yet support key value pair. |
| fprintf(stderr, "Module logging does not yet support key value pair.\n"); |
| return 1; |
| } else { |
| fprintf(out, ", arg%d", argIndex); |
| } |
| argIndex++; |
| } |
| fprintf(out, ");\n"); |
| fprintf(out, " }\n"); // if |
| } |
| |
| fprintf(out, " }\n"); // method |
| fprintf(out, "\n"); |
| } |
| return 0; |
| |
| } |
| |
| int write_stats_log_java(FILE* out, const Atoms& atoms, const AtomDecl &attributionDecl, |
| const string& javaClass, |
| const string& javaPackage, const bool supportQ, |
| const bool supportWorkSource) { |
| // Print prelude |
| fprintf(out, "// This file is autogenerated\n"); |
| fprintf(out, "\n"); |
| fprintf(out, "package %s;\n", javaPackage.c_str()); |
| fprintf(out, "\n"); |
| fprintf(out, "\n"); |
| if (supportQ) { |
| fprintf(out, "import android.os.Build;\n"); |
| fprintf(out, "import android.os.SystemClock;\n"); |
| } |
| |
| fprintf(out, "import android.util.StatsEvent;\n"); |
| fprintf(out, "import android.util.StatsLog;\n"); |
| |
| fprintf(out, "\n"); |
| fprintf(out, "\n"); |
| fprintf(out, "/**\n"); |
| fprintf(out, " * Utility class for logging statistics events.\n"); |
| fprintf(out, " */\n"); |
| fprintf(out, "public class %s {\n", javaClass.c_str()); |
| |
| write_java_atom_codes(out, atoms); |
| write_java_enum_values(out, atoms); |
| |
| int errors = 0; |
| |
| // Print write methods. |
| fprintf(out, " // Write methods\n"); |
| errors += write_java_methods( |
| out, atoms.signatureInfoMap, attributionDecl, supportQ); |
| errors += write_java_non_chained_methods( |
| out, atoms.nonChainedSignatureInfoMap); |
| if (supportWorkSource) { |
| errors += write_java_work_source_methods(out, atoms.signatureInfoMap); |
| } |
| |
| if (supportQ) { |
| errors += write_java_q_logger_class( |
| out, atoms.signatureInfoMap, attributionDecl); |
| } |
| |
| fprintf(out, "}\n"); |
| |
| return errors; |
| } |
| |
| } // namespace stats_log_api_gen |
| } // namespace android |