QString oscapItemGetReadableTitle(struct xccdf_item* item, struct xccdf_policy* policy, const QString& lang) { struct oscap_text_iterator* title_it = xccdf_item_get_title(item); char* unresolved = oscap_textlist_get_preferred_plaintext(title_it, lang.isEmpty() ? NULL : lang.toUtf8().constData()); oscap_text_iterator_free(title_it); if (!unresolved) return ""; char* resolved = xccdf_policy_substitute(Qt::escape(QString::fromUtf8(unresolved)).toUtf8().constData(), policy); free(unresolved); const QString ret = QString::fromUtf8(resolved); free(resolved); return ret; }
void TailoringWindow::synchronizeTreeItem(QTreeWidgetItem* treeItem, struct xccdf_item* xccdfItem, bool recursive) { ++mSynchronizeItemLock; const QString title = oscapTextIteratorGetPreferred(xccdf_item_get_title(xccdfItem)); treeItem->setText(0, title); QString searchable = QString("%1 %2").arg(title, QString::fromUtf8(xccdf_item_get_id(xccdfItem))); switch (xccdf_item_get_type(xccdfItem)) { case XCCDF_BENCHMARK: treeItem->setIcon(0, getShareIcon("benchmark.png")); break; case XCCDF_GROUP: treeItem->setIcon(0, getShareIcon("group.png")); break; case XCCDF_RULE: treeItem->setIcon(0, getShareIcon("rule.png")); { struct xccdf_ident_iterator* idents = xccdf_rule_get_idents(xccdf_item_to_rule(xccdfItem)); while (xccdf_ident_iterator_has_more(idents)) { struct xccdf_ident* ident = xccdf_ident_iterator_next(idents); searchable += " " + QString::fromUtf8(xccdf_ident_get_id(ident)); } xccdf_ident_iterator_free(idents); } break; case XCCDF_VALUE: treeItem->setIcon(0, getShareIcon("value.png")); break; default: treeItem->setIcon(0, QIcon()); break; } treeItem->setText(1, searchable); treeItem->setData(0, Qt::UserRole, QVariant::fromValue(reinterpret_cast<void*>(xccdfItem))); xccdf_type_t xccdfItemType = xccdf_item_get_type(xccdfItem); switch (xccdfItemType) { case XCCDF_RULE: case XCCDF_GROUP: { treeItem->setFlags(treeItem->flags() | Qt::ItemIsUserCheckable); treeItem->setCheckState(0, getXccdfItemInternalSelected(mPolicy, xccdfItem) ? Qt::Checked : Qt::Unchecked); _syncXCCDFItemChildrenDisabledState(treeItem, treeItem->checkState(0)); break; } case XCCDF_VALUE: treeItem->setFlags(treeItem->flags() & ~Qt::ItemIsUserCheckable); default: break; } if (recursive) { typedef std::vector<struct xccdf_item*> XCCDFItemVector; typedef std::map<struct xccdf_item*, QTreeWidgetItem*> XCCDFToQtItemMap; XCCDFItemVector itemsToAdd; XCCDFToQtItemMap existingItemsMap; // valuesIt contains Values struct xccdf_value_iterator* valuesIt = NULL; // itemsIt contains Rules and Groups struct xccdf_item_iterator* itemsIt = NULL; switch (xccdfItemType) { case XCCDF_GROUP: valuesIt = xccdf_group_get_values(xccdf_item_to_group(xccdfItem)); itemsIt = xccdf_group_get_content(xccdf_item_to_group(xccdfItem)); break; case XCCDF_BENCHMARK: valuesIt = xccdf_benchmark_get_values(xccdf_item_to_benchmark(xccdfItem)); itemsIt = xccdf_benchmark_get_content(xccdf_item_to_benchmark(xccdfItem)); break; default: break; } if (valuesIt != NULL) { while (xccdf_value_iterator_has_more(valuesIt)) { struct xccdf_value* childItem = xccdf_value_iterator_next(valuesIt); itemsToAdd.push_back(xccdf_value_to_item(childItem)); } xccdf_value_iterator_free(valuesIt); } if (itemsIt != NULL) { while (xccdf_item_iterator_has_more(itemsIt)) { struct xccdf_item* childItem = xccdf_item_iterator_next(itemsIt); itemsToAdd.push_back(childItem); } xccdf_item_iterator_free(itemsIt); } for (int i = 0; i < treeItem->childCount(); ++i) { QTreeWidgetItem* childTreeItem = treeItem->child(i); struct xccdf_item* childXccdfItem = getXccdfItemFromTreeItem(childTreeItem); if (std::find(itemsToAdd.begin(), itemsToAdd.end(), childXccdfItem) == itemsToAdd.end()) { // this will remove it from the tree as well, see ~QTreeWidgetItem() delete childTreeItem; } else { existingItemsMap[childXccdfItem] = childTreeItem; } } unsigned int idx = 0; for (XCCDFItemVector::const_iterator it = itemsToAdd.begin(); it != itemsToAdd.end(); ++it, ++idx) { struct xccdf_item* childXccdfItem = *it; QTreeWidgetItem* childTreeItem = 0; XCCDFToQtItemMap::iterator mapIt = existingItemsMap.find(childXccdfItem); if (mapIt == existingItemsMap.end()) { childTreeItem = new QTreeWidgetItem(); childTreeItem->setFlags( Qt::ItemIsSelectable | Qt::ItemIsEnabled); treeItem->insertChild(idx, childTreeItem); } else { childTreeItem = mapIt->second; } synchronizeTreeItem(childTreeItem, childXccdfItem, true); } } --mSynchronizeItemLock; }